fix domain context pre set provider issue

Signed-off-by: Michael Beemer <[email protected]>

Author: beeme1mr

Diff for: packages/client/src/open-feature.ts

@@ -28,7 +28,6 @@ export class OpenFeatureAPI extends OpenFeatureCommonAPI<Provider, Hook> impleme
   protected _events: GenericEventEmitter<ProviderEvents> = new OpenFeatureEventEmitter();
   protected _defaultProvider: Provider = NOOP_PROVIDER;
   protected _createEventEmitter = () => new OpenFeatureEventEmitter();
-  protected _domainScopedContext: Map<string, EvaluationContext> = new Map();
 
   private constructor() {
     super('client');

Diff for: packages/client/test/open-feature.spec.ts

@@ -235,5 +235,30 @@ describe('OpenFeature', () => {
       await OpenFeature.close();
       expect(provider3.onClose).toHaveBeenCalled();
     });
+
+    describe('context during initialization', () => {
+      it('should use the context set in the domain', async () => {
+        const domain = 'test';
+        await OpenFeature.setContext(domain, { user: 'mike' });
+
+        const provider = mockProvider();
+        const spy = jest.spyOn(provider, 'initialize');
+        OpenFeature.setProvider(domain, provider);
+
+        expect(spy).toHaveBeenCalledWith({ user: 'mike' });
+      });
+
+      it('should use the default context', async () => {
+        const domain = 'test';
+        await OpenFeature.setContext(domain, { user: 'mike' });
+        await OpenFeature.setContext({ name: 'todd' });
+
+        const provider = mockProvider();
+        const spy = jest.spyOn(provider, 'initialize');
+        OpenFeature.setProvider(provider);
+
+        expect(spy).toHaveBeenCalledWith({ name: 'todd' });
+      });
+    });
   });
 });

Diff for: packages/shared/src/open-feature.ts

@@ -29,6 +29,7 @@ export abstract class OpenFeatureCommonAPI<P extends CommonProvider = CommonProv
 
   private readonly _clientEventHandlers: Map<string | undefined, [AnyProviderEvent, EventHandler][]> = new Map();
   protected _domainScopedProviders: Map<string, P> = new Map();
+  protected _domainScopedContext: Map<string, EvaluationContext> = new Map();
   protected _clientEvents: Map<string | undefined, GenericEventEmitter<AnyProviderEvent>> = new Map();
   protected _runsOn: Paradigm;
 
@@ -215,7 +216,7 @@ export abstract class OpenFeatureCommonAPI<P extends CommonProvider = CommonProv
 
     if (provider?.status === ProviderStatus.NOT_READY && typeof provider.initialize === 'function') {
       initializationPromise = provider
-        .initialize?.(this._context)
+        .initialize?.(domain ? this._domainScopedContext.get(domain) ?? this._context : this._context)
         ?.then(() => {
           // fetch the most recent event emitters, some may have been added during init
           this.getAssociatedEventEmitters(domain).forEach((emitter) => {
@@ -284,16 +285,15 @@ export abstract class OpenFeatureCommonAPI<P extends CommonProvider = CommonProv
     this._clientEvents.set(domain, newEmitter);
 
     const clientProvider = this.getProviderForClient(domain);
-    Object.values<AllProviderEvents>(AllProviderEvents).forEach(
-      (eventType) =>
-        clientProvider.events?.addHandler(eventType, async (details) => {
-          newEmitter.emit(eventType, {
-            ...details,
-            clientName: domain,
-            domain,
-            providerName: clientProvider.metadata.name,
-          });
-        }),
+    Object.values<AllProviderEvents>(AllProviderEvents).forEach((eventType) =>
+      clientProvider.events?.addHandler(eventType, async (details) => {
+        newEmitter.emit(eventType, {
+          ...details,
+          clientName: domain,
+          domain,
+          providerName: clientProvider.metadata.name,
+        });
+      }),
     );
 
     return newEmitter;