more cases

Author: ssundahlTTD

src/integrationTests/secureSignal.test.ts

@@ -100,6 +100,7 @@ describe('Secure Signal Tests', () => {
 
       afterEach(() => {
         mocks.resetFakeTime();
+        mocks.removeUid2LocalStorage();
       });
 
       test('should send signal to Google ESP when SDK initialized', async () => {
@@ -117,6 +118,57 @@ describe('Secure Signal Tests', () => {
           identity.advertising_token
         );
       });
+
+      test('should send signal to Google ESP when SDK initialized and Identity is available', async () => {
+        mocks.setUid2LocalStorage(identity);
+        __uid2SSProviderScriptLoad();
+        __uid2InternalHandleScriptLoad();
+        (window.__uid2 as UID2).init({});
+        expect(secureSignalProvidersPushMock).toHaveBeenCalled();
+        await expect(secureSignalProvidersPushMock).toHaveBeenCalledWith(
+          expect.objectContaining({
+            id: 'uidapi.com',
+          })
+        );
+        await mocks.flushPromises();
+        expect(await secureSignalProvidersPushMock.mock.results[0].value).toBe(
+          identity.advertising_token
+        );
+      });
+
+      test('should send signal to Google ESP when SDK initialized and Identity is set after', async () => {
+        __uid2SSProviderScriptLoad();
+        __uid2InternalHandleScriptLoad();
+        (window.__uid2 as UID2).init({});
+        (window.__uid2 as UID2).setIdentity(identity);
+        expect(secureSignalProvidersPushMock).toHaveBeenCalled();
+        await expect(secureSignalProvidersPushMock).toHaveBeenCalledWith(
+          expect.objectContaining({
+            id: 'uidapi.com',
+          })
+        );
+        await mocks.flushPromises();
+        expect(await secureSignalProvidersPushMock.mock.results[0].value).toBe(
+          identity.advertising_token
+        );
+      });
+
+      test('should send signal to Google ESP when SDK initialized and SS script is included later', async () => {
+        __uid2InternalHandleScriptLoad();
+        (window.__uid2 as UID2).init({ identity });
+        __uid2SSProviderScriptLoad();
+        // will fire on both InitCompleted and SdkLoaded
+        expect(secureSignalProvidersPushMock).toHaveBeenCalledTimes(2);
+        await expect(secureSignalProvidersPushMock).toHaveBeenCalledWith(
+          expect.objectContaining({
+            id: 'uidapi.com',
+          })
+        );
+        await mocks.flushPromises();
+        expect(await secureSignalProvidersPushMock.mock.results[0].value).toBe(
+          identity.advertising_token
+        );
+      });
     });
 
     describe('When script loaded after SDK loaded', () => {
@@ -184,6 +236,7 @@ describe('Secure Signal Tests', () => {
     describe('When SDK initialized after both SDK and SS script loaded - UID2', () => {
       beforeEach(() => {
         window.__uid2 = new UID2();
+        mocks.removeUid2LocalStorage();
       });
       test('should send identity to Google ESP', async () => {
         __uid2InternalHandleScriptLoad();

src/secureSignalEuid.ts

@@ -20,13 +20,12 @@ export function __euidSSProviderScriptLoad() {
   window.__euid = window.__euid || {
     callbacks: [],
   };
-  window.__euid.callbacks?.push((eventType) => {
-    if (eventType === 'SdkLoaded') {
-      if (window.__euid?.getIdentity()) {
-        window.__euidSecureSignalProvider.registerSecureSignalProvider();
-      }
-    }
-    if (eventType === 'IdentityUpdated') {
+  window.__euid.callbacks.push((eventType) => {
+    if (
+      eventType === 'InitCompleted' ||
+      eventType === 'SdkLoaded' ||
+      eventType === 'IdentityUpdated'
+    ) {
       if (window.__euid.getIdentity()) {
         window.__euidSecureSignalProvider.registerSecureSignalProvider();
       }

src/secureSignalUid2.ts

@@ -17,14 +17,13 @@ export function __uid2SSProviderScriptLoad() {
   window.__uid2 = window.__uid2 || {
     callbacks: [],
   };
-  window.__uid2.callbacks?.push((eventType) => {
-    if (eventType === 'SdkLoaded') {
-      if (window.__uid2?.getIdentity()) {
-        window.__uid2SecureSignalProvider.registerSecureSignalProvider();
-      }
-    }
-    if (eventType === 'IdentityUpdated') {
-      if (window.__uid2?.getIdentity()) {
+  window.__uid2.callbacks.push((eventType) => {
+    if (
+      eventType === 'InitCompleted' ||
+      eventType === 'SdkLoaded' ||
+      eventType === 'IdentityUpdated'
+    ) {
+      if (window.__uid2.getIdentity()) {
         window.__uid2SecureSignalProvider.registerSecureSignalProvider();
       }
     }