fix(flagsmith): Fetch identity flags when context changes (WIP)

Signed-off-by: Rodrigo López Dato <[email protected]>

Author: rolodato

libs/providers/flagsmith-client/src/lib/flagsmith-client-provider.spec.ts

@@ -353,6 +353,30 @@ describe('FlagsmithProvider', () => {
       );
       expect(config.fetch).toHaveBeenCalledTimes(3);
     });
+
+    it('should set new traits when the context is updated', async () => {
+      const config = defaultConfig();
+      const provider = new FlagsmithClientProvider({
+        logger,
+        ...config,
+      });
+      await OpenFeature.setProviderAndWait(provider);
+      await OpenFeature.setContext({ targetingKey: 'first', traitA: 'a' });
+      expect(config.fetch).toHaveBeenCalledWith(
+        `${provider.flagsmithClient.getState().api}identities/`,
+        expect.objectContaining({
+          body: JSON.stringify({ identifier: 'first', traits: [{ trait_key: 'traitA', trait_value: 'a' }] }),
+        }),
+      );
+      await OpenFeature.setContext({ targetingKey: 'second', traitB: 'b' });
+      expect(config.fetch).toHaveBeenCalledWith(
+        `${provider.flagsmithClient.getState().api}identities/`,
+        expect.objectContaining({
+          body: JSON.stringify({ identifier: 'second', traits: [{ trait_key: 'traitB', trait_value: 'b' }] }),
+        }),
+      );
+    });
+
     it('should initialize with the targeting key and traits when passed to initialize', async () => {
       const targetingKey = 'test';
       const traits = { foo: 'bar', example: 123 };

libs/providers/flagsmith-client/src/lib/flagsmith-client-provider.ts

@@ -53,7 +53,14 @@ export class FlagsmithClientProvider implements Provider {
         ...(context || {}),
       });
       this.events.emit(ProviderEvents.Stale, { message: 'context has changed' });
-      return isLogout ? this._client.logout() : this._client.getFlags();
+      if (isLogout) {
+        return this._client.logout();
+      }
+      if (identity) {
+        const { targetingKey, ...traits } = context;
+        return this._client.identify(identity, traits as any);
+      }
+      return this._client.getFlags();
     }
 
     const serverState = this._config.state;