Add fetchsegments on setuser test

optimizely · Jul 20, 2023 · 3a40344 · 3a40344
1 parent 887d912
commit 3a40344
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 4 deletions.
diff --git a/src/client.spec.ts b/src/client.spec.ts
@@ -220,11 +220,11 @@ describe('ReactSDKClient', () => {
       });
     });
 
-    it('adds and removes update handlers', () => {
+    it('adds and removes update handlers', async () => {
       const instance = createInstance(config);
       const onUserUpdateListener = jest.fn();
       const dispose = instance.onUserUpdate(onUserUpdateListener);
-      instance.setUser({
+      await instance.setUser({
         id: 'newUser',
       });
       expect(onUserUpdateListener).toBeCalledTimes(1);
@@ -233,12 +233,27 @@ describe('ReactSDKClient', () => {
         attributes: {},
       });
       dispose();
-      instance.setUser({
+      await instance.setUser({
         id: 'newUser2',
       });
       expect(onUserUpdateListener).toBeCalledTimes(1);
     });
 
+    it('calls fetchqualifiedsegements internally on each setuser call', async () => {
+      const instance = createInstance(config);
+      jest.spyOn(instance, 'fetchQualifiedSegments').mockImplementation(async () => true);
+
+      await instance.setUser({
+        id: 'xxfueaojfe8&86',
+      });
+
+      await instance.setUser({
+        id: 'xxfueaojfe8&87',
+      });
+
+      expect(instance.fetchQualifiedSegments).toBeCalledTimes(2);
+    });
+
     describe('pre-set user and user overrides', () => {
       let instance: ReactSDKClient;
       beforeEach(() => {

diff --git a/src/client.ts b/src/client.ts
@@ -45,7 +45,7 @@ export interface ReactSDKClient extends Omit<optimizely.Client, 'createUserConte
   user: UserInfo;
 
   onReady(opts?: { timeout?: number }): Promise<any>;
-  setUser(userInfo: UserInfo): void;
+  setUser(userInfo: UserInfo): Promise<void>;
   onUserUpdate(handler: OnUserUpdateHandler): DisposeFn;
   isReady(): boolean;
   getIsReadyPromiseFulfilled(): boolean;