Fix/register extension with launcher instead of integration

src/Rokt-Kit.ts

@@ -336,9 +336,11 @@ function extractRoktExtensionConfig(settingsString?: string): RoktExtensionConfi
   };
 }
 
-function registerLegacyExtensions(legacyExtensions: string[]) {
-  for (const extension of legacyExtensions) {
-    window.Rokt?.use(extension);
+function registerLegacyExtensions(legacyExtensions: string[], launcher: RoktLauncher | null) {
+  if (launcher) {
+    for (const extension of legacyExtensions) {
+      launcher.use(extension);
+    }
   }
 }
 
@@ -1107,7 +1109,7 @@ class RoktKit implements KitInterface {
         onLoad: () => {
           if (this.isLauncherReadyToAttach()) {
             this.attachLauncher(accountId, launcherOptions);
-            registerLegacyExtensions(legacyRoktExtensions);
+            registerLegacyExtensions(legacyRoktExtensions, this.launcher);
           } else {
             console.error('Rokt object is not available after script load.');
           }

test/src/tests.spec.ts

@@ -3348,11 +3348,18 @@ describe('Rokt Forwarder', () => {
       expect(tyeScript.src).toContain('/rokt-elements/rokt-element-thank-you.js');
     });
 
-    it('should call window.Rokt.use with ThankYouJourney when thank-you-journey extension is provided', async () => {
+    it('should call launcher.use with ThankYouJourney when thank-you-journey extension is provided', async () => {
       document.getElementById('rokt-thank-you-element')?.remove();
       document.getElementById('rokt-launcher')?.remove();
 
       const useCalls: string[] = [];
+      const mockLauncher = {
+        selectPlacements: () => {},
+        hashAttributes: () => {},
+        use: (name: string) => {
+          useCalls.push(name);
+        },
+      };
 
       (window as any).Rokt = undefined;
       (window as any).mParticle.Rokt = {
@@ -3375,17 +3382,98 @@ describe('Rokt Forwarder', () => {
         false,
       );
 
+      // Use a synchronous thenable so this.launcher is set before registerLegacyExtensions runs
       (window as any).Rokt = new (MockRoktForwarder as any)();
-      (window as any).Rokt.use = (name: string) => {
-        useCalls.push(name);
-      };
-      (window as any).Rokt.createLauncher = async () =>
-        Promise.resolve({ selectPlacements: () => {}, hashAttributes: () => {}, use: () => Promise.resolve() });
+      (window as any).Rokt.createLauncher = () => ({
+        then: (onFulfilled: (launcher: typeof mockLauncher) => void) => {
+          onFulfilled(mockLauncher);
+          return { catch: () => {} };
+        },
+      });
 
       const launcherScript = document.getElementById('rokt-launcher') as HTMLScriptElement;
       launcherScript.onload!(new Event('load'));
 
-      await waitForCondition(() => useCalls.length > 0);
+      expect(useCalls).toContain('ThankYouJourney');
+    });
+
+    it('should fetch thank you element resource when thank you element extension is provided', async () => {
+      document.getElementById('rokt-thank-you-element')?.remove();
+      document.getElementById('rokt-launcher')?.remove();
+
+      (window as any).Rokt = undefined;
+      (window as any).mParticle.Rokt = {
+        attachKit: async (kit: any) => {
+          (window as any).mParticle.Rokt.kit = kit;
+        },
+        filters: {
+          userAttributesFilters: [],
+          filterUserAttributes: (attrs: any) => attrs,
+          filteredUser: { getMPID: () => '123' },
+        },
+        use: () => Promise.resolve(),
+      };
+
+      await (window as any).mParticle.forwarder.init(
+        {
+          accountId: '123456',
+          roktExtensions: '[{"jsmap":null,"map":null,"maptype":"LegacyExtension","value":"thank-you-journey"}]',
+        },
+        reportService.cb,
+        false,
+      );
+
+      const tyeScript = document.getElementById('rokt-thank-you-element') as HTMLScriptElement;
+      expect(tyeScript).not.toBeNull();
+      expect(tyeScript.src).toContain('/rokt-elements/rokt-element-thank-you.js');
+    });
+
+    it('should call launcher.use with ThankYouJourney when thank-you-journey extension is provided', async () => {
+      document.getElementById('rokt-thank-you-element')?.remove();
+      document.getElementById('rokt-launcher')?.remove();
+
+      const useCalls: string[] = [];
+
+      (window as any).Rokt = undefined;
+      (window as any).mParticle.Rokt = {
+        attachKit: async (kit: any) => {
+          (window as any).mParticle.Rokt.kit = kit;
+        },
+        filters: {
+          userAttributesFilters: [],
+          filterUserAttributes: (attrs: any) => attrs,
+          filteredUser: { getMPID: () => '123' },
+        },
+      };
+
+      await (window as any).mParticle.forwarder.init(
+        {
+          accountId: '123456',
+          roktExtensions: '[{"jsmap":null,"map":null,"maptype":"LegacyExtension","value":"thank-you-journey"}]',
+        },
+        reportService.cb,
+        false,
+      );
+
+      const mockLauncher = {
+        selectPlacements: () => {},
+        hashAttributes: () => {},
+        use: (name: string) => {
+          useCalls.push(name);
+        },
+      };
+
+      // Use a synchronous thenable so this.launcher is set before registerLegacyExtensions runs
+      (window as any).Rokt = new (MockRoktForwarder as any)();
+      (window as any).Rokt.createLauncher = () => ({
+        then: (onFulfilled: (launcher: typeof mockLauncher) => void) => {
+          onFulfilled(mockLauncher);
+          return { catch: () => {} };
+        },
+      });
+
+      const launcherScript = document.getElementById('rokt-launcher') as HTMLScriptElement;
+      launcherScript.onload!(new Event('load'));
 
       expect(useCalls).toContain('ThankYouJourney');
     });