Use actual growthbook instance for testing

Author: madhuchavva

dev-packages/node-integration-tests/package.json

@@ -26,6 +26,7 @@
     "@anthropic-ai/sdk": "0.63.0",
     "@aws-sdk/client-s3": "^3.552.0",
     "@google/genai": "^1.20.0",
+    "@growthbook/growthbook": "^1.6.1",
     "@hapi/hapi": "^21.3.10",
     "@hono/node-server": "^1.19.4",
     "@nestjs/common": "11.1.3",

dev-packages/node-integration-tests/suites/featureFlags/growthbook/onError/basic/scenario.ts

@@ -1,59 +1,89 @@
+import type { ClientOptions, UserContext } from '@growthbook/growthbook';
+import { GrowthBookClient } from '@growthbook/growthbook';
 import { _INTERNAL_FLAG_BUFFER_SIZE as FLAG_BUFFER_SIZE } from '@sentry/core';
 import * as Sentry from '@sentry/node';
 import { loggingTransport } from '@sentry-internal/node-integration-tests';
 
-// Minimal GrowthBook-like class that matches the real API for testing
-// This is necessary since we don't want to add @growthbook/growthbook as a dependency
-// just for integration tests, but we want to test the actual integration behavior
-class GrowthBookLike {
-  private _features: Record<string, { value: unknown }> = {};
+// Wrapper class to instantiate GrowthBookClient
+class GrowthBookWrapper {
+  private _gbClient: GrowthBookClient;
+  private _userContext: UserContext = { attributes: { id: 'test-user-123' } };
 
-  public isOn(featureKey: string): boolean {
-    const feature = this._features[featureKey];
-    return feature ? !!feature.value : false;
+  public constructor(..._args: unknown[]) {
+    // Create GrowthBookClient with proper configuration
+    const clientOptions: ClientOptions = {
+      apiHost: 'https://cdn.growthbook.io',
+      clientKey: 'sdk-abc123'
+    };
+    this._gbClient = new GrowthBookClient(clientOptions);
+
+    // Create features for testing
+    const features = this._createTestFeatures();
+
+    this._gbClient.initSync({
+      payload: { features }
+    });
   }
 
-  public getFeatureValue(featureKey: string, defaultValue: unknown): unknown {
-    const feature = this._features[featureKey];
-    return feature ? feature.value : defaultValue;
+  public isOn(featureKey: string, ..._rest: unknown[]): boolean {
+    return this._gbClient.isOn(featureKey, this._userContext);
   }
 
-  // Helper method to set feature values for testing
-  public setFeature(featureKey: string, value: unknown): void {
-    this._features[featureKey] = { value };
+  public getFeatureValue(featureKey: string, defaultValue: unknown, ..._rest: unknown[]): unknown {
+    return this._gbClient.getFeatureValue(featureKey, defaultValue as boolean | string | number, this._userContext);
+  }
+
+  private _createTestFeatures(): Record<string, { defaultValue: unknown }> {
+    const features: Record<string, { defaultValue: unknown }> = {};
+
+    // Fill buffer with flags 1-100 (all false by default)
+    for (let i = 1; i <= FLAG_BUFFER_SIZE; i++) {
+      features[`feat${i}`] = { defaultValue: false };
+    }
+
+    // Add feat101 (true), which should evict feat1
+    features[`feat${FLAG_BUFFER_SIZE + 1}`] = { defaultValue: true };
+
+    // Update feat3 to true, which should move it to the end
+    features['feat3'] = { defaultValue: true };
+
+    // Test features with boolean values (should be captured)
+    features['bool-feat'] = { defaultValue: true };
+
+    // Test features with non-boolean values (should NOT be captured)
+    features['string-feat'] = { defaultValue: 'hello' };
+    features['number-feat'] = { defaultValue: 42 };
+
+    return features;
   }
 }
 
 Sentry.init({
   dsn: 'https://[email protected]/1337',
   sampleRate: 1.0,
   transport: loggingTransport,
-  integrations: [Sentry.growthbookIntegration({ growthbookClass: GrowthBookLike })],
+  integrations: [Sentry.growthbookIntegration({ growthbookClass: GrowthBookWrapper })],
 });
 
-const gb = new GrowthBookLike();
+// Create GrowthBookWrapper instance
+const gb = new GrowthBookWrapper();
 
 // Fill buffer with flags 1-100 (all false by default)
 for (let i = 1; i <= FLAG_BUFFER_SIZE; i++) {
   gb.isOn(`feat${i}`);
 }
 
 // Add feat101 (true), which should evict feat1
-gb.setFeature(`feat${FLAG_BUFFER_SIZE + 1}`, true);
 gb.isOn(`feat${FLAG_BUFFER_SIZE + 1}`);
 
 // Update feat3 to true, which should move it to the end
-gb.setFeature('feat3', true);
 gb.isOn('feat3');
 
 // Test getFeatureValue with boolean values (should be captured)
-gb.setFeature('bool-feat', true);
 gb.getFeatureValue('bool-feat', false);
 
 // Test getFeatureValue with non-boolean values (should NOT be captured)
-gb.setFeature('string-feat', 'hello');
 gb.getFeatureValue('string-feat', 'default');
-gb.setFeature('number-feat', 42);
 gb.getFeatureValue('number-feat', 0);
 
 throw new Error('Test error');

dev-packages/node-integration-tests/suites/featureFlags/growthbook/onSpan/scenario.ts

@@ -1,41 +1,56 @@
+import type { ClientOptions, InitSyncOptions, UserContext } from '@growthbook/growthbook';
+import { GrowthBookClient } from '@growthbook/growthbook';
 import * as Sentry from '@sentry/node';
 import { loggingTransport } from '@sentry-internal/node-integration-tests';
 
-// Minimal GrowthBook-like class that matches the real API for testing
-class GrowthBookLike {
-  private _features: Record<string, { value: unknown }> = {};
+// Wrapper class to instantiate GrowthBookClient
+class GrowthBookWrapper {
+  private _gbClient: GrowthBookClient;
+  private _userContext: UserContext = { attributes: { id: 'test-user-123' } };
 
-  public isOn(featureKey: string): boolean {
-    const feature = this._features[featureKey];
-    return feature ? !!feature.value : false;
+  public constructor(..._args: unknown[]) {
+    // Create GrowthBookClient and initialize it synchronously with payload
+    const clientOptions: ClientOptions = {
+      apiHost: 'https://cdn.growthbook.io',
+      clientKey: 'sdk-abc123'
+    };
+    this._gbClient = new GrowthBookClient(clientOptions);
+
+    // Create test features
+    const features = {
+      'feat1': { defaultValue: true },
+      'feat2': { defaultValue: false },
+      'bool-feat': { defaultValue: true },
+      'string-feat': { defaultValue: 'hello' }
+    };
+
+    // Initialize synchronously with payload
+    const initOptions: InitSyncOptions = {
+      payload: { features }
+    };
+
+    this._gbClient.initSync(initOptions);
   }
 
-  public getFeatureValue(featureKey: string, defaultValue: unknown): unknown {
-    const feature = this._features[featureKey];
-    return feature ? feature.value : defaultValue;
+  public isOn(featureKey: string, ..._rest: unknown[]): boolean {
+    return this._gbClient.isOn(featureKey, this._userContext);
   }
 
-  // Helper method to set feature values for testing
-  public setFeature(featureKey: string, value: unknown): void {
-    this._features[featureKey] = { value };
+  public getFeatureValue(featureKey: string, defaultValue: unknown, ..._rest: unknown[]): unknown {
+    return this._gbClient.getFeatureValue(featureKey, defaultValue as boolean | string | number, this._userContext);
   }
 }
 
+const gb = new GrowthBookWrapper();
+
 Sentry.init({
   dsn: 'https://[email protected]/1337',
   sampleRate: 1.0,
   tracesSampleRate: 1.0,
   transport: loggingTransport,
-  integrations: [Sentry.growthbookIntegration({ growthbookClass: GrowthBookLike })],
+  integrations: [Sentry.growthbookIntegration({ growthbookClass: GrowthBookWrapper })],
 });
 
-const gb = new GrowthBookLike();
-
-// Set up feature flags
-gb.setFeature('feat1', true);
-gb.setFeature('feat2', false);
-gb.setFeature('bool-feat', true);
-
 Sentry.startSpan({ name: 'test-span', op: 'function' }, () => {
   // Evaluate feature flags during the span
   gb.isOn('feat1');
@@ -45,6 +60,5 @@ Sentry.startSpan({ name: 'test-span', op: 'function' }, () => {
   gb.getFeatureValue('bool-feat', false);
 
   // Test getFeatureValue with non-boolean values (should NOT be captured)
-  gb.setFeature('string-feat', 'hello');
   gb.getFeatureValue('string-feat', 'default');
 });