WIP

Author: jcobis

packages/compass-collection/src/stores/collection-tab.spec.ts

@@ -10,6 +10,7 @@ import type { experimentationServiceLocator } from '@mongodb-js/compass-telemetr
 import type { connectionInfoRefLocator } from '@mongodb-js/compass-connections/provider';
 import { createNoopLogger } from '@mongodb-js/compass-logging/provider';
 import { ReadOnlyPreferenceAccess } from 'compass-preferences-model/provider';
+import { TestName } from '../../../compass-telemetry/src/growth-experiments';
 
 const defaultMetadata = {
   namespace: 'test.foo',
@@ -143,9 +144,9 @@ describe('Collection Tab Content store', function () {
 
       await waitFor(() => {
         expect(assignExperiment).to.have.been.calledOnceWith(
-          'mock-data-generator',
+          TestName.mockDataGenerator,
           {
-            team: 'data-explorer',
+            team: 'Atlas Growth',
           }
         );
       });

packages/compass-collection/src/stores/collection-tab.ts

@@ -16,6 +16,7 @@ import {
   isAIFeatureEnabled,
   type PreferencesAccess,
 } from 'compass-preferences-model/provider';
+import { TestName } from '../../../compass-telemetry/src/growth-experiments';
 
 export type CollectionTabOptions = {
   /**
@@ -116,14 +117,14 @@ export function activatePlugin(
       isAIFeatureEnabled(preferences.getPreferences()) // Ensures org-level AI features setting is enabled
     ) {
       void experimentationServices
-        .assignExperiment('mock-data-generator', {
+        .assignExperiment(TestName.mockDataGenerator, {
           team: 'Atlas Growth',
         })
         .catch((error) => {
           logger.debug('Mock Data Generator experiment assignment failed', {
-            experiment: 'MOCK_DATA_GENERATOR',
+            experiment: TestName.mockDataGenerator,
             namespace: namespace,
-            error: error.message,
+            error: error instanceof Error ? error.message : String(error),
           });
         });
     }

packages/compass-telemetry/src/experimentation-provider.tsx

@@ -1,15 +1,16 @@
 import React, { createContext, useContext, useRef } from 'react';
 import type { types } from '@mongodb-js/mdb-experiment-js';
 import type { typesReact } from '@mongodb-js/mdb-experiment-js/react';
+import type { TestName } from './growth-experiments';
 
 type UseAssignmentHook = (
-  experimentName: string,
+  experimentName: TestName,
   trackIsInSample: boolean,
   options?: typesReact.UseAssignmentOptions<types.TypeData>
 ) => typesReact.UseAssignmentResponse<types.TypeData>;
 
 type AssignExperimentFn = (
-  experimentName: string,
+  experimentName: TestName,
   options?: types.AssignOptions<string>
 ) => Promise<types.AsyncStatus | null>;
 

packages/compass-telemetry/src/growth-experiments.ts

@@ -1,3 +1,4 @@
 export enum TestName {
   earlyJourneyIndexesGuidance = 'EARLY_JOURNEY_INDEXES_GUIDANCE_20250328',
+  mockDataGenerator = 'MOCK_DATA_GENERATOR_20251001',
 }

packages/compass-telemetry/src/provider.tsx

@@ -53,7 +53,7 @@ export function useTelemetry(): TrackFunction {
 export const experimentationServiceLocator = createServiceLocator(
   function useExperimentationServices(): {
     assignExperiment: (
-      experimentName: string,
+      experimentName: TestName,
       options?: types.AssignOptions<string>
     ) => Promise<types.AsyncStatus | null>;
   } {