test: validate default extensions and graceful degredation of slim bundle

Author: dustinbyrne

packages/browser/src/__tests__/extension-classes.test.ts

@@ -35,6 +35,12 @@ describe('__extensionClasses enrollment', () => {
         expect(posthog.webVitalsAutocapture).toBeUndefined()
         expect(posthog.productTours).toBeUndefined()
         expect(posthog.siteApps).toBeUndefined()
+        expect(posthog.surveys).toBeUndefined()
+        expect(posthog.toolbar).toBeUndefined()
+        expect(posthog.exceptions).toBeUndefined()
+        expect(posthog.conversations).toBeUndefined()
+        expect(posthog.logs).toBeUndefined()
+        expect(posthog.experiments).toBeUndefined()
     })
 
     it('initializes no extensions when none are provided and no defaults exist', async () => {
@@ -53,6 +59,12 @@ describe('__extensionClasses enrollment', () => {
         expect(posthog.webVitalsAutocapture).toBeUndefined()
         expect(posthog.productTours).toBeUndefined()
         expect(posthog.siteApps).toBeUndefined()
+        expect(posthog.surveys).toBeUndefined()
+        expect(posthog.toolbar).toBeUndefined()
+        expect(posthog.exceptions).toBeUndefined()
+        expect(posthog.conversations).toBeUndefined()
+        expect(posthog.logs).toBeUndefined()
+        expect(posthog.experiments).toBeUndefined()
     })
 
     it('__extensionClasses overrides __defaultExtensionClasses', async () => {
@@ -85,6 +97,12 @@ describe('__extensionClasses enrollment', () => {
         expect(posthog.webVitalsAutocapture).toBeDefined()
         expect(posthog.productTours).toBeDefined()
         expect(posthog.siteApps).toBeDefined()
+        expect(posthog.surveys).toBeDefined()
+        expect(posthog.toolbar).toBeDefined()
+        expect(posthog.exceptions).toBeDefined()
+        expect(posthog.conversations).toBeDefined()
+        expect(posthog.logs).toBeDefined()
+        expect(posthog.experiments).toBeDefined()
     })
 })
 
@@ -107,13 +125,19 @@ describe('extension lifecycle', () => {
             const allKeys = Object.keys(AllExtensions).sort()
             expect(allKeys).toEqual([
                 'autocapture',
+                'conversations',
                 'deadClicksAutocapture',
                 'exceptionObserver',
+                'exceptions',
+                'experiments',
                 'heatmaps',
                 'historyAutocapture',
+                'logs',
                 'productTours',
                 'sessionRecording',
                 'siteApps',
+                'surveys',
+                'toolbar',
                 'tracingHeaders',
                 'webVitalsAutocapture',
             ])
@@ -173,13 +197,11 @@ describe('extension lifecycle', () => {
                 }
             }
 
-            // Use extension keys that don't have hardcoded method calls
-            // in set_config, so a spy class works without needing stubs.
             const posthog = await createPosthogInstance(undefined, {
                 __preview_deferred_init_extensions: false,
                 __extensionClasses: {
-                    historyAutocapture: SpyExtension as any,
-                    siteApps: SpyExtension as any,
+                    toolbar: SpyExtension as any,
+                    conversations: SpyExtension as any,
                 },
                 capture_pageview: false,
             })
@@ -195,4 +217,48 @@ describe('extension lifecycle', () => {
             expect(onRemoteConfigSpy).toHaveBeenCalledWith(remoteConfig)
         })
     })
+
+    describe('graceful degradation without extensions (slim bundle)', () => {
+        it('onSurveysLoaded calls back with error when extension is not loaded', async () => {
+            PostHog.__defaultExtensionClasses = {}
+
+            const posthog = await createPosthogInstance(undefined, {
+                __preview_deferred_init_extensions: false,
+                capture_pageview: false,
+            })
+
+            const callback = jest.fn()
+            posthog.onSurveysLoaded(callback)
+
+            expect(callback).toHaveBeenCalledWith([], { isLoaded: false, error: 'Surveys module not available' })
+        })
+
+        it('getSurveys calls back with error when extension is not loaded', async () => {
+            PostHog.__defaultExtensionClasses = {}
+
+            const posthog = await createPosthogInstance(undefined, {
+                __preview_deferred_init_extensions: false,
+                capture_pageview: false,
+            })
+
+            const callback = jest.fn()
+            posthog.getSurveys(callback)
+
+            expect(callback).toHaveBeenCalledWith([], { isLoaded: false, error: 'Surveys module not available' })
+        })
+
+        it('getActiveMatchingSurveys calls back with error when extension is not loaded', async () => {
+            PostHog.__defaultExtensionClasses = {}
+
+            const posthog = await createPosthogInstance(undefined, {
+                __preview_deferred_init_extensions: false,
+                capture_pageview: false,
+            })
+
+            const callback = jest.fn()
+            posthog.getActiveMatchingSurveys(callback)
+
+            expect(callback).toHaveBeenCalledWith([], { isLoaded: false, error: 'Surveys module not available' })
+        })
+    })
 })