@thoughtspot/visual-embed-sdk 1.44.1-test → 1.44.2

package/src/embed/embed.spec.ts

@@ -1,13 +1,18 @@
 import { logger } from '../utils/logger';
 import {
     init, AuthType, SearchEmbed, EmbedEvent,
+    RuntimeFilterOp,
+    Action,
 } from '../index';
 import {
     EVENT_WAIT_TIME,
     executeAfterWait,
+    expectUrlToHaveParamsWithValues,
     getDocumentBody,
     getIFrameEl,
+    getIFrameSrc,
     getRootEl,
+    postMessageToParent,
 } from '../test/test-utils';
 import * as authInstance from '../auth';
 
@@ -24,8 +29,8 @@ beforeAll(() => {
         thoughtSpotHost,
         authType: AuthType.None,
     });
-    spyOn(window, 'alert');
-    jest.spyOn(authInstance, 'postLoginService').mockResolvedValue(true);
+    jest.spyOn(window, 'alert');
+    jest.spyOn(authInstance, 'postLoginService').mockImplementation(() => Promise.resolve(undefined));
 });
 
 describe('test view config', () => {
@@ -52,4 +57,119 @@ describe('test view config', () => {
             expect(iframe.style.height).toBe(`${height}px`);
         });
     });
+    
+    test('should pass hideResults parameter when configured', async () => {
+        const searchEmbed = new SearchEmbed(getRootEl(), {
+            ...defaultViewConfig,
+            hideResults: true,
+        });
+        searchEmbed.render();
+
+        await executeAfterWait(() => {
+            const iframeSrc = getIFrameSrc();
+            expectUrlToHaveParamsWithValues(iframeSrc, {
+                hideResult: true,
+            });
+        });
+    });
+
+    test('should pass forceTable parameter when configured', async () => {
+        const searchEmbed = new SearchEmbed(getRootEl(), {
+            ...defaultViewConfig,
+            forceTable: true,
+        });
+        searchEmbed.render();
+
+        await executeAfterWait(() => {
+            const iframeSrc = getIFrameSrc();
+            expectUrlToHaveParamsWithValues(iframeSrc, {
+                forceTable: true,
+            });
+        });
+    });
+
+    test('should pass enableSearchAssist parameter when configured', async () => {
+        const searchEmbed = new SearchEmbed(getRootEl(), {
+            ...defaultViewConfig,
+            enableSearchAssist: true,
+        });
+        searchEmbed.render();
+
+        await executeAfterWait(() => {
+            const iframeSrc = getIFrameSrc();
+            expectUrlToHaveParamsWithValues(iframeSrc, {
+                enableSearchAssist: true,
+            });
+        });
+    });
+
+    test('should pass hideSearchBar parameter when configured', async () => {
+        const searchEmbed = new SearchEmbed(getRootEl(), {
+            ...defaultViewConfig,
+            hideSearchBar: true,
+        });
+        searchEmbed.render();
+
+        await executeAfterWait(() => {
+            const iframeSrc = getIFrameSrc();
+            expectUrlToHaveParamsWithValues(iframeSrc, {
+                hideSearchBar: true,
+            });
+        });
+    });
+
+    test('should register and trigger event listeners', async () => {
+        const searchEmbed = new SearchEmbed(getRootEl(), defaultViewConfig);
+        const mockCallback = jest.fn();
+
+        searchEmbed.on(EmbedEvent.Load, mockCallback);
+        await searchEmbed.render();
+
+        await executeAfterWait(() => {
+            const iframe = getIFrameEl();
+            postMessageToParent(iframe.contentWindow, {
+                type: EmbedEvent.Load,
+            });
+        });
+
+        await executeAfterWait(() => {
+            expect(mockCallback).toHaveBeenCalled();
+        }, EVENT_WAIT_TIME);
+    });
+
+    test('should pass disabledActions parameter when configured', async () => {
+        const disabledActions = [Action.Download, Action.Share];
+        const searchEmbed = new SearchEmbed(getRootEl(), {
+            ...defaultViewConfig,
+            disabledActions,
+        });
+        searchEmbed.render();
+
+        await executeAfterWait(() => {
+            const iframeSrc = getIFrameSrc();
+            expect(iframeSrc).toContain('disableAction');
+        });
+    });
+
+    test('should pass runtime filters when configured', async () => {
+        const runtimeFilters = [
+            {
+                columnName: 'revenue',
+                operator: RuntimeFilterOp.GT,
+                values: [1000],
+            },
+        ];
+        const searchEmbed = new SearchEmbed(getRootEl(), {
+            ...defaultViewConfig,
+            runtimeFilters,
+        });
+        searchEmbed.render();
+
+        await executeAfterWait(() => {
+            const iframeSrc = getIFrameSrc();
+            expect(iframeSrc).toContain('col1=revenue');
+            expect(iframeSrc).toContain('op1=GT');
+            expect(iframeSrc).toContain('val1=1000');
+        });
+    });
 });

package/src/embed/events.spec.ts

@@ -36,8 +36,8 @@ beforeAll(() => {
         thoughtSpotHost,
         authType: AuthType.None,
     });
-    spyOn(window, 'alert');
-    jest.spyOn(authInstance, 'postLoginService').mockReturnValue(true);
+    jest.spyOn(window, 'alert');
+    jest.spyOn(authInstance, 'postLoginService').mockImplementation(() => Promise.resolve(undefined));
 });
 
 describe('test communication between host app and ThoughtSpot', () => {

package/src/embed/hostEventClient/host-event-client.spec.ts

@@ -1,5 +1,5 @@
 import { getIFrameEl, getRootEl } from '../../test/test-utils';
-import { HostEvent } from '../../types';
+import { AuthType, HostEvent } from '../../types';
 import { processTrigger } from '../../utils/processTrigger';
 import * as EmbedConfigService from '../embedConfig';
 import {
@@ -23,7 +23,7 @@ const createHostEventClient = () => {
 describe('HostEventClient', () => {
     const mockThoughtSpotHost = 'http://localhost';
     beforeEach(() => {
-        jest.spyOn(EmbedConfigService, 'getEmbedConfig').mockReturnValue({ thoughtSpotHost: mockThoughtSpotHost });
+        jest.spyOn(EmbedConfigService, 'getEmbedConfig').mockReturnValue({ thoughtSpotHost: mockThoughtSpotHost, authType: AuthType.None } as any);
     });
 
     afterEach(() => {

package/src/embed/liveboard.spec.ts

@@ -27,6 +27,7 @@ import * as processTriggerInstance from '../utils/processTrigger';
 import * as auth from '../auth';
 import * as previewService from '../utils/graphql/preview-service';
 import * as SessionInfoService from '../utils/sessionInfoService';
+import { logger } from '../utils/logger';
 
 const defaultViewConfig = {
     frameParams: {
@@ -57,7 +58,7 @@ beforeAll(() => {
         thoughtSpotHost,
         authType: AuthType.None,
     });
-    jest.spyOn(auth, 'postLoginService').mockImplementation(() => Promise.resolve({}));
+    jest.spyOn(auth, 'postLoginService').mockImplementation(() => Promise.resolve(undefined));
 });
 
 describe('Liveboard/viz embed tests', () => {
@@ -743,7 +744,7 @@ describe('Liveboard/viz embed tests', () => {
         });
     });
     test('should process the trigger, for vizEmbed', async () => {
-        const mockProcessTrigger = spyOn(processTriggerInstance, 'processTrigger');
+        const mockProcessTrigger = jest.spyOn(processTriggerInstance, 'processTrigger').mockImplementation(jest.fn());
         const liveboardEmbed = new LiveboardEmbed(getRootEl(), {
             enableVizTransformations: true,
             ...defaultViewConfig,
@@ -753,7 +754,7 @@ describe('Liveboard/viz embed tests', () => {
         liveboardEmbed.render();
         await executeAfterWait(async () => {
             await liveboardEmbed.trigger(HostEvent.Pin);
-            expect(mockProcessTrigger).toBeCalled();
+            expect(mockProcessTrigger).toHaveBeenCalled();
         });
     });
 
@@ -820,7 +821,7 @@ describe('Liveboard/viz embed tests', () => {
         });
     });
 
-    test('navigateToLiveboard should trigger the navigate event with the correct path', async (done) => {
+    test('navigateToLiveboard should trigger the navigate event with the correct path', async () => {
         mockMessageChannel();
         // mock getSessionInfo
 
@@ -830,7 +831,7 @@ describe('Liveboard/viz embed tests', () => {
         } as LiveboardViewConfig);
         const onSpy = jest.spyOn(liveboardEmbed, 'trigger');
         await liveboardEmbed.prerenderGeneric();
-        executeAfterWait(() => {
+        await executeAfterWait(() => {
             const iframe = getIFrameEl();
             postMessageToParent(iframe.contentWindow, {
                 type: EmbedEvent.APP_INIT,
@@ -841,13 +842,12 @@ describe('Liveboard/viz embed tests', () => {
             liveboardEmbed.navigateToLiveboard('lb1', 'viz1');
         });
 
-        executeAfterWait(() => {
+        await executeAfterWait(() => {
             expect(onSpy).toHaveBeenCalledWith(HostEvent.Navigate, 'embed/viz/lb1/viz1');
-            done();
         }, 1002);
     });
 
-    test('navigateToLiveboard with preRender', async (done) => {
+    test('navigateToLiveboard with preRender', async () => {
         mockMessageChannel();
 
         // mock getSessionInfo
@@ -857,7 +857,7 @@ describe('Liveboard/viz embed tests', () => {
             currentOrgId: 1,
             privileges: [],
             mixpanelToken: '1234567890',
-        });
+        } as any);
         mockGetSessionInfo();
 
         const liveboardEmbed = new LiveboardEmbed(getRootEl(), {
@@ -866,7 +866,7 @@ describe('Liveboard/viz embed tests', () => {
         } as LiveboardViewConfig);
         const onSpy = jest.spyOn(liveboardEmbed, 'trigger');
         await liveboardEmbed.prerenderGeneric();
-        executeAfterWait(() => {
+        await executeAfterWait(() => {
             const iframe = getIFrameEl();
             postMessageToParent(iframe.contentWindow, {
                 type: EmbedEvent.APP_INIT,
@@ -875,10 +875,9 @@ describe('Liveboard/viz embed tests', () => {
                 type: EmbedEvent.AuthInit,
             });
         });
-        executeAfterWait(() => {
+        await executeAfterWait(() => {
             liveboardEmbed.navigateToLiveboard('lb1', 'viz1');
             expect(onSpy).toHaveBeenCalledWith(HostEvent.Navigate, 'embed/viz/lb1/viz1');
-            done();
         }, 1002);
     });
     test('should set runtime parametere values in url params', async () => {
@@ -958,7 +957,7 @@ describe('Liveboard/viz embed tests', () => {
             const result = liveboardEmbed.trigger(HostEvent.SetActiveTab, {
                 tabId: newActiveTabId,
             });
-            expect(mockProcessTrigger).not.toBeCalled();
+            expect(mockProcessTrigger).not.toHaveBeenCalled();
         });
     });
 
@@ -988,14 +987,14 @@ describe('Liveboard/viz embed tests', () => {
                 authType: AuthType.None,
             });
         });
-        test('it should preRender generic with liveboard id is not passed', async (done) => {
+        test('it should preRender generic with liveboard id is not passed', async () => {
             const consoleSpy = jest.spyOn(console, 'error');
             const libEmbed = new LiveboardEmbed(getRootEl(), {
                 preRenderId: 'testPreRender',
             });
             const prerenderGenericSpy = jest.spyOn(libEmbed, 'prerenderGeneric');
             await libEmbed.preRender();
-            executeAfterWait(() => {
+            await executeAfterWait(() => {
                 const iFrame = document.getElementById(
                     libEmbed.getPreRenderIds().child,
                 ) as HTMLIFrameElement;
@@ -1005,8 +1004,6 @@ describe('Liveboard/viz embed tests', () => {
                 expect(iFrame.src).toMatch(/http:\/\/tshost\/.*&isLiveboardEmbed=true.*#$/);
 
                 expect(consoleSpy).toHaveBeenCalledTimes(0);
-
-                done();
             });
         });
 
@@ -1054,7 +1051,7 @@ describe('Liveboard/viz embed tests', () => {
             });
         });
 
-        test('it should navigateToLiveboard with liveboard id is not passed with EmbedListenerReady event', async (done) => {
+        test('it should navigateToLiveboard with liveboard id is not passed with EmbedListenerReady event', async () => {
             mockMessageChannel();
             const consoleSpy = jest.spyOn(console, 'error');
             const testPreRenderId = 'testPreRender';
@@ -1099,7 +1096,7 @@ describe('Liveboard/viz embed tests', () => {
             const navigateToLiveboardSpy = jest.spyOn(newLibEmbed, 'navigateToLiveboard');
             await newLibEmbed.showPreRender();
 
-            executeAfterWait(() => {
+            await executeAfterWait(() => {
                 const iFrame = document.getElementById(
                     libEmbed.getPreRenderIds().child,
                 ) as HTMLIFrameElement;
@@ -1109,12 +1106,10 @@ describe('Liveboard/viz embed tests', () => {
                 expect(iFrame.src).toMatch(/http:\/\/tshost\/.*&isLiveboardEmbed=true.*#$/);
 
                 expect(consoleSpy).toHaveBeenCalledTimes(0);
-
-                done();
             });
         });
 
-        test('it should navigateToLiveboard with liveboard id is not passed with AuthInit event', async (done) => {
+        test('it should navigateToLiveboard with liveboard id is not passed with AuthInit event', async () => {
             mockMessageChannel();
             const consoleSpy = jest.spyOn(console, 'error');
             const testPreRenderId = 'testPreRender';
@@ -1168,7 +1163,6 @@ describe('Liveboard/viz embed tests', () => {
                 expect(navigateToLiveboardSpy).toHaveBeenCalledWith(testLiveboardId, 'testVizId', 'testActiveTabId');
                 expect(iFrame.src).toMatch(/http:\/\/tshost\/.*&isLiveboardEmbed=true.*#$/);
                 expect(consoleSpy).toHaveBeenCalledTimes(0);
-                done();
             }, 1005);
         });
 
@@ -1277,6 +1271,7 @@ describe('Liveboard/viz embed tests', () => {
                 liveboardId,
                 fullHeight: true,
                 lazyLoadingForFullHeight: true,
+                lazyLoadingMargin: '10px',
             } as LiveboardViewConfig);
 
             await liveboardEmbed.render();
@@ -1330,6 +1325,7 @@ describe('Liveboard/viz embed tests', () => {
                 liveboardId,
                 fullHeight: true,
                 lazyLoadingForFullHeight: true,
+                lazyLoadingMargin: '10px',
             } as LiveboardViewConfig);
 
             await liveboardEmbed.render();
@@ -1348,6 +1344,7 @@ describe('Liveboard/viz embed tests', () => {
                 liveboardId,
                 fullHeight: true,
                 lazyLoadingForFullHeight: true,
+                lazyLoadingMargin: '10px',
             } as LiveboardViewConfig);
 
             const mockTrigger = jest.spyOn(liveboardEmbed, 'trigger');
@@ -1380,6 +1377,7 @@ describe('Liveboard/viz embed tests', () => {
                 liveboardId,
                 fullHeight: true,
                 lazyLoadingForFullHeight: true,
+                lazyLoadingMargin: '10px',
             } as LiveboardViewConfig);
 
             const mockTrigger = jest.spyOn(liveboardEmbed, 'trigger');
@@ -1405,6 +1403,7 @@ describe('Liveboard/viz embed tests', () => {
                 liveboardId,
                 fullHeight: true,
                 lazyLoadingForFullHeight: true,
+                lazyLoadingMargin: '10px',
             } as LiveboardViewConfig);
 
             await liveboardEmbed.render();
@@ -1426,6 +1425,7 @@ describe('Liveboard/viz embed tests', () => {
                 liveboardId,
                 fullHeight: true,
                 lazyLoadingForFullHeight: true,
+                lazyLoadingMargin: '10px',
             } as LiveboardViewConfig);
 
             await liveboardEmbed.render();
@@ -1454,6 +1454,7 @@ describe('Liveboard/viz embed tests', () => {
                 liveboardId,
                 fullHeight: true,
                 lazyLoadingForFullHeight: true,
+                lazyLoadingMargin: '10px',
             } as LiveboardViewConfig);
 
             // Set the iframe before render
@@ -1512,7 +1513,7 @@ describe('Liveboard/viz embed tests', () => {
                 ...defaultViewConfig,
             });
 
-            const navigateToLiveboardSpy = jest.spyOn(liveboardEmbed, 'navigateToLiveboard').mockResolvedValue(undefined);
+            const navigateToLiveboardSpy = jest.spyOn(liveboardEmbed, 'navigateToLiveboard').mockImplementation(() => Promise.resolve(undefined));
 
             // Mock embed container as not loaded initially
             liveboardEmbed.isEmbedContainerLoaded = false;
@@ -1544,7 +1545,7 @@ describe('Liveboard/viz embed tests', () => {
             };
 
             jest.spyOn(liveboardEmbed as any, 'getPreRenderObj').mockReturnValue(mockPreRenderObj as any);
-            jest.spyOn(liveboardEmbed, 'navigateToLiveboard').mockResolvedValue(undefined);
+            jest.spyOn(liveboardEmbed, 'navigateToLiveboard').mockImplementation(() => Promise.resolve(undefined));
 
             // Mock embed container as not loaded initially
             liveboardEmbed.isEmbedContainerLoaded = false;
@@ -1572,7 +1573,7 @@ describe('Liveboard/viz embed tests', () => {
                 ...defaultViewConfig,
             });
 
-            const navigateToLiveboardSpy = jest.spyOn(liveboardEmbed, 'navigateToLiveboard').mockResolvedValue(undefined);
+            const navigateToLiveboardSpy = jest.spyOn(liveboardEmbed, 'navigateToLiveboard').mockImplementation(() => Promise.resolve(undefined));
 
             // Mock embed container as already loaded
             liveboardEmbed.isEmbedContainerLoaded = true;
@@ -1593,7 +1594,7 @@ describe('Liveboard/viz embed tests', () => {
             });
 
             jest.spyOn(liveboardEmbed as any, 'getPreRenderObj').mockReturnValue(null);
-            const navigateToLiveboardSpy = jest.spyOn(liveboardEmbed, 'navigateToLiveboard').mockResolvedValue(undefined);
+            const navigateToLiveboardSpy = jest.spyOn(liveboardEmbed, 'navigateToLiveboard').mockImplementation(() => Promise.resolve(undefined));
 
             // Mock embed container as not loaded initially
             liveboardEmbed.isEmbedContainerLoaded = false;
@@ -1621,7 +1622,7 @@ describe('Liveboard/viz embed tests', () => {
                 ...defaultViewConfig,
             });
 
-            const navigateToLiveboardSpy = jest.spyOn(liveboardEmbed, 'navigateToLiveboard').mockResolvedValue(undefined);
+            const navigateToLiveboardSpy = jest.spyOn(liveboardEmbed, 'navigateToLiveboard').mockImplementation(() => Promise.resolve(undefined));
 
             // Mock embed container as already loaded
             liveboardEmbed.isEmbedContainerLoaded = true;
@@ -1639,7 +1640,7 @@ describe('Liveboard/viz embed tests', () => {
                 ...defaultViewConfig,
             });
 
-            const navigateToLiveboardSpy = jest.spyOn(liveboardEmbed, 'navigateToLiveboard').mockResolvedValue(undefined);
+            const navigateToLiveboardSpy = jest.spyOn(liveboardEmbed, 'navigateToLiveboard').mockImplementation(() => Promise.resolve(undefined));
 
             // Mock embed container as already loaded
             liveboardEmbed.isEmbedContainerLoaded = true;

package/src/embed/liveboard.ts

@@ -23,7 +23,7 @@ import {
     ErrorDetailsTypes,
     EmbedErrorCodes,
 } from '../types';
-import { calculateVisibleElementData, getQueryParamString, isUndefined } from '../utils';
+import { calculateVisibleElementData, getQueryParamString, isUndefined, isValidCssMargin } from '../utils';
 import { getAuthPromise } from './base';
 import { TsEmbed, V1Embed } from './ts-embed';
 import { addPreviewStylesIfNotPresent } from '../utils/global-styles';
@@ -528,7 +528,9 @@ export class LiveboardEmbed extends V1Embed {
             params[Param.fullHeight] = true;
             if (this.viewConfig.lazyLoadingForFullHeight) {
                 params[Param.IsLazyLoadingForEmbedEnabled] = true;
-                params[Param.RootMarginForLazyLoad] = this.viewConfig.lazyLoadingMargin;
+                if (isValidCssMargin(this.viewConfig.lazyLoadingMargin)) {
+                    params[Param.RootMarginForLazyLoad] = this.viewConfig.lazyLoadingMargin;
+                }
             }
         }
         this.defaultHeight = minimumHeight || defaultHeight || this.defaultHeight;

package/src/embed/pinboard.spec.ts

@@ -31,7 +31,7 @@ beforeAll(() => {
         thoughtSpotHost,
         authType: AuthType.None,
     });
-    jest.spyOn(auth, 'postLoginService').mockReturnValue(true);
+    jest.spyOn(auth, 'postLoginService').mockImplementation(() => Promise.resolve(undefined));
 });
 
 describe('Pinboard/viz embed tests', () => {

package/src/embed/sage.spec.ts

@@ -28,8 +28,8 @@ beforeAll(() => {
         thoughtSpotHost,
         authType: AuthType.None,
     });
-    spyOn(window, 'alert');
-    jest.spyOn(authInstance, 'postLoginService').mockResolvedValue(true);
+    jest.spyOn(window, 'alert');
+    jest.spyOn(authInstance, 'postLoginService').mockImplementation(() => Promise.resolve(true as any));
 });
 
 describe('Sage  embed tests', () => {

package/src/embed/search.spec.ts

@@ -40,8 +40,8 @@ beforeAll(() => {
         thoughtSpotHost,
         authType: AuthType.None,
     });
-    jest.spyOn(authInstance, 'postLoginService').mockImplementation(() => Promise.resolve({}));
-    spyOn(window, 'alert');
+    jest.spyOn(authInstance, 'postLoginService').mockImplementation(() => Promise.resolve(undefined));
+    jest.spyOn(window, 'alert').mockImplementation(() => {});
 });
 
 describe('Search embed tests', () => {
@@ -666,3 +666,134 @@ describe('Search embed tests', () => {
         expect(getIFrameSrc().includes('executeSearch')).toBeFalsy();
     });
 });
+
+// Add these tests to src/embed/search.spec.ts
+
+// ============================================
+// SearchEmbed tests for uncovered lines 427, 430
+// ============================================
+
+// TEST 1: hideResults parameter (line 427)
+test('should pass hideResult parameter when hideResults is true', async () => {
+    const searchEmbed = new SearchEmbed(getRootEl(), {
+        ...defaultViewConfig,
+        hideResults: true,
+    });
+    searchEmbed.render();
+    await executeAfterWait(() => {
+        const iframeSrc = getIFrameSrc();
+        expect(iframeSrc).toContain('hideResult=true');
+    });
+});
+
+// TEST 2: forceTable parameter (line 430)
+test('should pass forceTable parameter when forceTable is true', async () => {
+    const searchEmbed = new SearchEmbed(getRootEl(), {
+        ...defaultViewConfig,
+        forceTable: true,
+    });
+    searchEmbed.render();
+    await executeAfterWait(() => {
+        const iframeSrc = getIFrameSrc();
+        expect(iframeSrc).toContain('forceTable=true');
+    });
+});
+
+describe('SearchBarEmbed tests', () => {
+    test('should pass dataSources parameter when dataSources array is provided', async () => {
+        const searchBarEmbed = new SearchBarEmbed(getRootEl() as any, {
+            ...defaultViewConfig,
+            dataSources: ['source-1', 'source-2'],
+        });
+        searchBarEmbed.render();
+        await executeAfterWait(() => {
+            const iframeSrc = getIFrameSrc();
+            expect(iframeSrc).toContain('dataSources');
+            expect(iframeSrc).toContain('source-1');
+        });
+    });
+
+    test('should pass dataSource parameter when single dataSource is provided', async () => {
+        const searchBarEmbed = new SearchBarEmbed(getRootEl() as any, {
+            ...defaultViewConfig,
+            dataSource: 'single-source-id',
+        });
+        searchBarEmbed.render();
+        await executeAfterWait(() => {
+            const iframeSrc = getIFrameSrc();
+            expect(iframeSrc).toContain('dataSources');
+            expect(iframeSrc).toContain('single-source-id');
+        });
+    });
+
+    test('should pass searchTokenString and executeSearch when searchOptions provided', async () => {
+        const searchBarEmbed = new SearchBarEmbed(getRootEl() as any, {
+            ...defaultViewConfig,
+            searchOptions: {
+                searchTokenString: '[revenue][region]',
+                executeSearch: true,
+            },
+        });
+        searchBarEmbed.render();
+        await executeAfterWait(() => {
+            const iframeSrc = getIFrameSrc();
+            expect(iframeSrc).toContain('searchTokenString');
+            expect(iframeSrc).toContain('executeSearch=true');
+        });
+    });
+
+    test('should set useLastSelectedSources to false when dataSource is provided', async () => {
+        const searchBarEmbed = new SearchBarEmbed(getRootEl() as any, {
+            ...defaultViewConfig,
+            dataSource: 'my-source',
+            useLastSelectedSources: true, // This should be overridden to false
+        });
+        searchBarEmbed.render();
+        await executeAfterWait(() => {
+            const iframeSrc = getIFrameSrc();
+            expect(iframeSrc).toContain('useLastSelectedSources=false');
+        });
+    });
+
+    test('should include searchOptions in APP_INIT when excludeSearchTokenStringFromURL is true', async () => {
+        const searchOptions = {
+            searchTokenString: '[quantity][product]',
+            executeSearch: true,
+        };
+        const searchBarEmbed = new SearchBarEmbed(getRootEl() as any, {
+            ...defaultViewConfig,
+            searchOptions,
+            excludeSearchTokenStringFromURL: true,
+        });
+
+        const mockEmbedEventPayload = {
+            type: EmbedEvent.APP_INIT,
+            data: {},
+        };
+
+        searchBarEmbed.render();
+
+        const mockPort: any = {
+            postMessage: jest.fn(),
+        };
+
+        await executeAfterWait(() => {
+            const iframe = getIFrameEl();
+            postMessageToParent(iframe.contentWindow, mockEmbedEventPayload, mockPort);
+        });
+
+        expect(getIFrameSrc().includes('searchTokenString')).toBeFalsy();
+
+        await executeAfterWait(() => {
+            expect(mockPort.postMessage).toHaveBeenCalledWith({
+                type: EmbedEvent.APP_INIT,
+                data: expect.objectContaining({
+                    searchOptions: {
+                        searchTokenString: '[quantity][product]',
+                        executeSearch: true,
+                    },
+                }),
+            });
+        });
+    });
+});

package/src/embed/ts-embed-trigger.spec.ts

@@ -13,9 +13,9 @@ import * as authInstance from '../auth';
 describe('Trigger', () => {
     beforeEach(() => {
         document.body.innerHTML = getDocumentBody();
-        jest.spyOn(authInstance, 'postLoginService').mockResolvedValue(true);
+        jest.spyOn(authInstance, 'postLoginService').mockImplementation(() => Promise.resolve(true as any));
     });
-    test('should trigger the event', async (done) => {
+    test('should trigger the event', async () => {
         init({
             thoughtSpotHost: 'https://tshost',
             authType: AuthType.None,
@@ -36,7 +36,6 @@ describe('Trigger', () => {
                 type: HostEvent.DownloadAsCsv,
                 data: { vizId: 'testId' },
             }), 'https://tshost', expect.anything());
-            done();
         });
     });
 });