@thoughtspot/visual-embed-sdk 1.45.1 → 1.45.3

package/cjs/src/embed/ts-embed.spec.js

@@ -24,6 +24,7 @@ const contracts_1 = require("./hostEventClient/contracts");
 const sessionInfoService = tslib_1.__importStar(require("../utils/sessionInfoService"));
 const authToken = tslib_1.__importStar(require("../authToken"));
 const apiIntercept = tslib_1.__importStar(require("../api-intercept"));
+const processData = tslib_1.__importStar(require("../utils/processData"));
 jest.mock('../utils/processTrigger');
 const mockProcessTrigger = processTrigger_1.processTrigger;
 const mockHandleInterceptEvent = jest.spyOn(apiIntercept, 'handleInterceptEvent');
@@ -140,7 +141,7 @@ describe('Unit test case for ts embed', () => {
                 expect(mockProcessTrigger).toHaveBeenCalledWith((0, test_utils_1.getIFrameEl)(), types_1.HostEvent.UIPassthrough, 'http://tshost', {
                     parameters: payload,
                     type: contracts_1.UIPassthroughEvent.PinAnswerToLiveboard,
-                });
+                }, undefined);
             });
         });
         test('Host event with empty param', async () => {
@@ -152,7 +153,7 @@ describe('Unit test case for ts embed', () => {
             mockProcessTrigger.mockResolvedValue({ session: 'test' });
             await (0, test_utils_1.executeAfterWait)(async () => {
                 await liveboardEmbed.trigger(types_1.HostEvent.Save);
-                expect(mockProcessTrigger).toHaveBeenCalledWith((0, test_utils_1.getIFrameEl)(), types_1.HostEvent.Save, 'http://tshost', {});
+                expect(mockProcessTrigger).toHaveBeenCalledWith((0, test_utils_1.getIFrameEl)(), types_1.HostEvent.Save, 'http://tshost', {}, undefined);
             });
         });
         test('Host event with falsy param', async () => {
@@ -164,7 +165,7 @@ describe('Unit test case for ts embed', () => {
             mockProcessTrigger.mockResolvedValue({ session: 'test' });
             await (0, test_utils_1.executeAfterWait)(async () => {
                 await liveboardEmbed.trigger(types_1.HostEvent.Save, false);
-                expect(mockProcessTrigger).toHaveBeenCalledWith((0, test_utils_1.getIFrameEl)(), types_1.HostEvent.Save, 'http://tshost', false);
+                expect(mockProcessTrigger).toHaveBeenCalledWith((0, test_utils_1.getIFrameEl)(), types_1.HostEvent.Save, 'http://tshost', false, undefined);
             });
         });
         test('should set proper height, width and min-height to iframe', async () => {
@@ -1120,7 +1121,7 @@ describe('Unit test case for ts embed', () => {
             const { mockPreauthInfoFetch, iFrame, } = await setup(true);
             expect(mockPreauthInfoFetch).toHaveBeenCalledTimes(1);
             await (0, test_utils_1.executeAfterWait)(() => {
-                expect(mockProcessTrigger).toHaveBeenCalledWith(iFrame, types_1.HostEvent.InfoSuccess, 'http://tshost', expect.objectContaining({ info: expect.any(Object) }));
+                expect(mockProcessTrigger).toHaveBeenCalledWith(iFrame, types_1.HostEvent.InfoSuccess, 'http://tshost', expect.objectContaining({ info: expect.any(Object) }), undefined);
             });
         });
         test('should not call InfoSuccess Event if overrideOrgId is true', async () => {
@@ -1223,28 +1224,28 @@ describe('Unit test case for ts embed', () => {
             const { mockGetPreauthInfo } = await setupPreauthTest('AppEmbed', undefined);
             await (0, test_utils_1.executeAfterWait)(() => {
                 expect(mockGetPreauthInfo).toHaveBeenCalledTimes(1);
-                expect(mockProcessTrigger).toHaveBeenCalledWith(expect.any(Object), types_1.HostEvent.InfoSuccess, 'http://tshost', expect.objectContaining({ info: expect.any(Object) }));
+                expect(mockProcessTrigger).toHaveBeenCalledWith(expect.any(Object), types_1.HostEvent.InfoSuccess, 'http://tshost', expect.objectContaining({ info: expect.any(Object) }), undefined);
             });
         });
         test('should enable preauth cache for FullAppEmbed with showPrimaryNavbar = false', async () => {
             const { mockGetPreauthInfo } = await setupPreauthTest('AppEmbed', false);
             await (0, test_utils_1.executeAfterWait)(() => {
                 expect(mockGetPreauthInfo).toHaveBeenCalledTimes(1);
-                expect(mockProcessTrigger).toHaveBeenCalledWith(expect.any(Object), types_1.HostEvent.InfoSuccess, 'http://tshost', expect.objectContaining({ info: expect.any(Object) }));
+                expect(mockProcessTrigger).toHaveBeenCalledWith(expect.any(Object), types_1.HostEvent.InfoSuccess, 'http://tshost', expect.objectContaining({ info: expect.any(Object) }), undefined);
             });
         });
         test('should enable preauth cache for SearchEmbed regardless of showPrimaryNavbar', async () => {
             const { mockGetPreauthInfo } = await setupPreauthTest('SearchEmbed', true);
             await (0, test_utils_1.executeAfterWait)(() => {
                 expect(mockGetPreauthInfo).toHaveBeenCalledTimes(1);
-                expect(mockProcessTrigger).toHaveBeenCalledWith(expect.any(Object), types_1.HostEvent.InfoSuccess, 'http://tshost', expect.objectContaining({ info: expect.any(Object) }));
+                expect(mockProcessTrigger).toHaveBeenCalledWith(expect.any(Object), types_1.HostEvent.InfoSuccess, 'http://tshost', expect.objectContaining({ info: expect.any(Object) }), undefined);
             });
         });
         test('should enable preauth cache for SearchEmbed (verifies fix for embed type regression)', async () => {
             const { mockGetPreauthInfo } = await setupPreauthTest('SearchEmbed', false);
             await (0, test_utils_1.executeAfterWait)(() => {
                 expect(mockGetPreauthInfo).toHaveBeenCalledTimes(1);
-                expect(mockProcessTrigger).toHaveBeenCalledWith(expect.any(Object), types_1.HostEvent.InfoSuccess, 'http://tshost', expect.objectContaining({ info: expect.any(Object) }));
+                expect(mockProcessTrigger).toHaveBeenCalledWith(expect.any(Object), types_1.HostEvent.InfoSuccess, 'http://tshost', expect.objectContaining({ info: expect.any(Object) }), undefined);
             });
         });
         test('should disable preauth cache for FullAppEmbed with overrideOrgId (combined condition)', async () => {
@@ -2330,7 +2331,7 @@ describe('Unit test case for ts embed', () => {
             jest.clearAllMocks();
             document.body.innerHTML = (0, test_utils_1.getDocumentBody)();
             mockPort.postMessage.mockClear();
-            jest.spyOn(authToken, 'getAuthenticationToken').mockResolvedValue('test-token');
+            jest.spyOn(authToken, 'getAuthenticationToken').mockResolvedValue('mock-test-token-placeholder');
             jest.spyOn(baseInstance, 'handleAuth').mockImplementation(() => Promise.resolve(true));
             jest.spyOn(baseInstance, 'notifyAuthFailure').mockImplementation(() => { });
         });
@@ -2354,7 +2355,7 @@ describe('Unit test case for ts embed', () => {
                 expect(baseInstance.handleAuth).toHaveBeenCalledTimes(1);
                 expect(mockPort.postMessage).toHaveBeenCalledWith({
                     type: index_1.EmbedEvent.AuthExpire,
-                    data: { authToken: 'test-token' },
+                    data: { authToken: 'mock-test-token-placeholder' },
                 });
             });
         });
@@ -2383,7 +2384,7 @@ describe('Unit test case for ts embed', () => {
                 expect(baseInstance.handleAuth).toHaveBeenCalledTimes(1);
                 expect(mockPort.postMessage).toHaveBeenCalledWith({
                     type: index_1.EmbedEvent.AuthExpire,
-                    data: { authToken: 'test-token' },
+                    data: { authToken: 'mock-test-token-placeholder' },
                 });
             });
         });
@@ -2427,6 +2428,83 @@ describe('Unit test case for ts embed', () => {
             expect(baseInstance.notifyAuthFailure).toHaveBeenCalledWith(authInstance.AuthFailureType.EXPIRY);
         });
     });
+    describe('AutoLogin behavior in tokenRefresh', () => {
+        const mockPort = { postMessage: jest.fn() };
+        const mockEmbedEventPayload = { type: index_1.EmbedEvent.RefreshAuthToken, data: {} };
+        beforeEach(() => {
+            jest.clearAllMocks();
+            document.body.innerHTML = (0, test_utils_1.getDocumentBody)();
+            mockPort.postMessage.mockClear();
+            jest.spyOn(authToken, 'getAuthenticationToken').mockResolvedValue('mock-test-token-placeholder');
+            jest.spyOn(processData, 'processAuthFailure').mockImplementation(() => ({}));
+            jest.spyOn(logger_1.logger, 'error').mockImplementation(() => { });
+        });
+        const renderAndTriggerRefreshAuthToken = async () => {
+            const spotterEmbed = new index_1.SpotterEmbed((0, test_utils_1.getRootEl)(), {
+                worksheetId: 'test-worksheet',
+                searchOptions: {
+                    searchQuery: 'test query',
+                },
+            });
+            await spotterEmbed.render();
+            await (0, test_utils_1.executeAfterWait)(() => {
+                const iframe = (0, test_utils_1.getIFrameEl)();
+                (0, test_utils_1.postMessageToParent)(iframe.contentWindow, mockEmbedEventPayload, mockPort);
+            });
+        };
+        test('Cookieless with autoLogin undefined should default to true and refresh token', async () => {
+            (0, index_1.init)({
+                thoughtSpotHost: 'tshost',
+                authType: index_1.AuthType.TrustedAuthTokenCookieless,
+                // autoLogin undefined
+            });
+            await renderAndTriggerRefreshAuthToken();
+            await (0, test_utils_1.executeAfterWait)(() => {
+                expect(authToken.getAuthenticationToken).toHaveBeenCalledWith(expect.any(Object), true);
+            });
+        });
+        test('Cookieless with autoLogin true should refresh token', async () => {
+            (0, index_1.init)({
+                thoughtSpotHost: 'tshost',
+                authType: index_1.AuthType.TrustedAuthTokenCookieless,
+                autoLogin: true,
+            });
+            await renderAndTriggerRefreshAuthToken();
+            await (0, test_utils_1.executeAfterWait)(() => {
+                expect(authToken.getAuthenticationToken).toHaveBeenCalledWith(expect.any(Object), true);
+            });
+        });
+        test('Cookieless with autoLogin false should not refresh token', async () => {
+            (0, index_1.init)({
+                thoughtSpotHost: 'tshost',
+                authType: index_1.AuthType.TrustedAuthTokenCookieless,
+                autoLogin: false,
+            });
+            await renderAndTriggerRefreshAuthToken();
+            await (0, test_utils_1.executeAfterWait)(() => {
+                expect(authToken.getAuthenticationToken).not.toHaveBeenCalled();
+                expect(mockPort.postMessage).not.toHaveBeenCalled();
+            });
+        });
+        test('Should handle error when getAuthenticationToken fails', async () => {
+            const error = new Error('Token fetch failed');
+            jest.spyOn(authToken, 'getAuthenticationToken').mockRejectedValue(error);
+            (0, index_1.init)({
+                thoughtSpotHost: 'tshost',
+                authType: index_1.AuthType.TrustedAuthTokenCookieless,
+                autoLogin: true,
+            });
+            await renderAndTriggerRefreshAuthToken();
+            await (0, test_utils_1.executeAfterWait)(() => {
+                expect(authToken.getAuthenticationToken).toHaveBeenCalledWith(expect.any(Object), true);
+                // Check that logger.error was called with the token refresh error
+                const errorCalls = logger_1.logger.error.mock.calls.filter((call) => { var _a, _b; return ((_a = call[0]) === null || _a === void 0 ? void 0 : _a.includes(errors_1.ERROR_MESSAGE.INVALID_TOKEN_ERROR)) && ((_b = call[0]) === null || _b === void 0 ? void 0 : _b.includes('Token fetch failed')); });
+                expect(errorCalls.length).toBeGreaterThan(0);
+                expect(processData.processAuthFailure).toHaveBeenCalledWith(error, expect.any(Element));
+                expect(mockPort.postMessage).not.toHaveBeenCalled();
+            });
+        });
+    });
     describe('Fullscreen Change Handler', () => {
         beforeEach(() => {
             document.body.innerHTML = (0, test_utils_1.getDocumentBody)();
@@ -2614,6 +2692,47 @@ describe('Unit test case for ts embed', () => {
             searchEmbed['executeAfterEmbedContainerLoaded'](callback3);
             expect(callback3).toHaveBeenCalledTimes(1);
         });
+        describe('getCurrentContext', () => {
+            const mockContext = {
+                stack: [
+                    {
+                        name: 'Liveboard',
+                        type: 'Liveboard',
+                        objectIds: { liveboardId: 'lb-123' },
+                    },
+                ],
+                currentContext: {
+                    name: 'Liveboard',
+                    type: 'Liveboard',
+                    objectIds: { liveboardId: 'lb-123' },
+                },
+            };
+            test('should return context when embed container is already loaded', async () => {
+                const searchEmbed = new index_1.SearchEmbed((0, test_utils_1.getRootEl)(), defaultViewConfig);
+                searchEmbed.isEmbedContainerLoaded = true;
+                const triggerSpy = jest.spyOn(searchEmbed, 'trigger')
+                    .mockResolvedValue(mockContext);
+                const context = await searchEmbed.getCurrentContext();
+                expect(context).toEqual(mockContext);
+                expect(triggerSpy).toHaveBeenCalledWith(types_1.HostEvent.GetPageContext, {});
+            });
+            test('should wait for embed container to load before returning context', async () => {
+                const searchEmbed = new index_1.SearchEmbed((0, test_utils_1.getRootEl)(), defaultViewConfig);
+                searchEmbed.isEmbedContainerLoaded = false;
+                const triggerSpy = jest.spyOn(searchEmbed, 'trigger')
+                    .mockResolvedValue(mockContext);
+                const contextPromise = searchEmbed.getCurrentContext();
+                // Context should not be resolved yet
+                await (0, test_utils_1.executeAfterWait)(() => {
+                    expect(triggerSpy).not.toHaveBeenCalled();
+                }, 10);
+                // Simulate embed container becoming ready
+                searchEmbed['executeEmbedContainerReadyCallbacks']();
+                const context = await contextPromise;
+                expect(context).toEqual(mockContext);
+                expect(triggerSpy).toHaveBeenCalledWith(types_1.HostEvent.GetPageContext, {});
+            });
+        });
         test('should register embed container event handlers during construction', () => {
             const searchEmbed = new index_1.SearchEmbed((0, test_utils_1.getRootEl)(), defaultViewConfig);
             // Check that the event handlers are registered
@@ -3328,7 +3447,7 @@ describe('Destroy error handling', () => {
             appEmbed.destroy();
         }).not.toThrow();
         expect(logSpy).toHaveBeenCalledWith('Error destroying TS Embed', expect.any(Error));
-        logSpy.mockRestore();
+        logSpy.mockReset();
     });
 });
 describe('Fullscreen change handler behavior', () => {
@@ -3362,7 +3481,7 @@ describe('Fullscreen change handler behavior', () => {
         const event = new Event('fullscreenchange');
         document.dispatchEvent(event);
         await (0, test_utils_1.executeAfterWait)(() => {
-            expect(mockProcessTrigger).toHaveBeenCalledWith(expect.any(Object), types_1.HostEvent.ExitPresentMode, expect.any(String), expect.any(Object));
+            expect(mockProcessTrigger).toHaveBeenLastCalledWith(expect.any(Object), types_1.HostEvent.ExitPresentMode, expect.any(String), expect.any(Object), undefined);
         });
     });
     test('should not trigger ExitPresentMode when entering fullscreen', async () => {
@@ -3429,9 +3548,9 @@ describe('ShowPreRender with UpdateEmbedParams', () => {
         });
         embed2.showPreRender();
         await (0, test_utils_1.executeAfterWait)(() => {
-            expect(mockProcessTrigger).toHaveBeenCalledWith(expect.any(Object), types_1.HostEvent.UpdateEmbedParams, expect.any(String), expect.objectContaining({
+            expect(mockProcessTrigger).toHaveBeenLastCalledWith(expect.any(Object), types_1.HostEvent.UpdateEmbedParams, expect.any(String), expect.objectContaining({
                 liveboardId: 'updated-lb',
-            }));
+            }), undefined);
         });
     });
     test('should trigger UpdateEmbedParams with runtime filters and visible vizs', async () => {
@@ -3455,7 +3574,7 @@ describe('ShowPreRender with UpdateEmbedParams', () => {
         });
         embed2.showPreRender();
         await (0, test_utils_1.executeAfterWait)(() => {
-            expect(mockProcessTrigger).toHaveBeenCalledWith(expect.any(Object), types_1.HostEvent.UpdateEmbedParams, expect.any(String), expect.objectContaining({
+            expect(mockProcessTrigger).toHaveBeenLastCalledWith(expect.any(Object), types_1.HostEvent.UpdateEmbedParams, expect.any(String), expect.objectContaining({
                 liveboardId: 'original-lb',
                 visibleVizs: ['viz-1'],
                 runtimeFilters: [
@@ -3470,7 +3589,7 @@ describe('ShowPreRender with UpdateEmbedParams', () => {
                         values: ['North'],
                     },
                 ],
-            }));
+            }), undefined);
         });
     });
     test('should trigger UpdateEmbedParams with updated config', async () => {
@@ -3498,7 +3617,7 @@ describe('ShowPreRender with UpdateEmbedParams', () => {
         });
         embed2.showPreRender();
         await (0, test_utils_1.executeAfterWait)(() => {
-            expect(mockProcessTrigger).toHaveBeenCalledWith(expect.any(Object), types_1.HostEvent.UpdateEmbedParams, expect.any(String), expect.objectContaining({
+            expect(mockProcessTrigger).toHaveBeenLastCalledWith(expect.any(Object), types_1.HostEvent.UpdateEmbedParams, expect.any(String), expect.objectContaining({
                 liveboardId: 'original-lb',
                 visibleVizs: ['viz-1', 'viz-2'],
                 runtimeFilters: [
@@ -3508,7 +3627,7 @@ describe('ShowPreRender with UpdateEmbedParams', () => {
                         values: ['North'],
                     },
                 ],
-            }));
+            }), undefined);
         });
     });
     test('should handle error when getUpdateEmbedParamsObject fails during showPreRender', async () => {