@thoughtspot/visual-embed-sdk 1.29.1-alpha.1 → 1.29.1

package/src/authToken.spec.ts

@@ -0,0 +1,31 @@
+import { getAuthenticationToken, resetCachedAuthToken } from './authToken';
+import * as authServiceInstance from './utils/authService/authService';
+import { EmbedConfig } from './types';
+
+describe('AuthToken Unit tests', () => {
+    test('getAuthenticationToken: When verification is disabled', async () => {
+        jest.spyOn(authServiceInstance, 'verifyTokenService');
+
+        const token = await getAuthenticationToken({
+            getAuthToken: async () => 'abc3',
+            disableTokenVerification: true,
+        } as EmbedConfig);
+
+        expect(token).toBe('abc3');
+        expect(authServiceInstance.verifyTokenService).not.toBeCalled();
+    });
+
+    test.only('getAuthenticationToken: When verification is enabled', async () => {
+        resetCachedAuthToken();
+        jest.clearAllMocks();
+        jest.spyOn(authServiceInstance, 'verifyTokenService').mockImplementation(() => true);
+        const token = await getAuthenticationToken({
+            thoughtSpotHost: 'test',
+            getAuthToken: async () => 'abc2',
+            disableTokenVerification: false,
+        } as EmbedConfig);
+
+        expect(token).toBe('abc2');
+        expect(authServiceInstance.verifyTokenService).toBeCalledWith('test', 'abc2');
+    });
+});

package/src/authToken.ts

@@ -10,8 +10,10 @@ const INVALID_TOKEN_ERR = 'Invalid token received form token callback or authTok
 let cachedAuthToken: string | null = null;
 
 // This method can be used to get the authToken using the embedConfig
-export async function getAuthenticationToken(embedConfig: EmbedConfig): Promise<string> {
-    if (cachedAuthToken) {
+export const getAuthenticationToken = async (embedConfig: EmbedConfig): Promise<string> => {
+    // Since we don't have token validation enabled , we cannot tell if the
+    // cached token is valid or not. So we will always fetch a new token.
+    if (cachedAuthToken && !embedConfig.disableTokenVerification) {
         let isCachedTokenStillValid;
         try {
             isCachedTokenStillValid = await validateAuthToken(embedConfig, cachedAuthToken, true);
@@ -32,23 +34,22 @@ export async function getAuthenticationToken(embedConfig: EmbedConfig): Promise<
         authToken = await response.text();
     }
 
-    try {
-        // this will throw error if the token is not valid
-        await validateAuthToken(embedConfig, authToken);
-    } catch (e) {
-        logger.error(`Received invalid token from getAuthToken callback or authToken endpoint. Error : ${e.message}`);
-        throw e;
-    }
+    // this will throw error if the token is not valid
+    await validateAuthToken(embedConfig, authToken);
 
     cachedAuthToken = authToken;
     return authToken;
-}
+};
 
 const validateAuthToken = async (
     embedConfig: EmbedConfig,
     authToken: string,
     suppressAlert?: boolean,
 ): Promise<boolean> => {
+    if (embedConfig.disableTokenVerification) {
+        logger.info('Token verification is disabled. Assuming token is valid.');
+        return true;
+    }
     try {
         const isTokenValid = await verifyTokenService(embedConfig.thoughtSpotHost, authToken);
         if (isTokenValid) return true;
@@ -69,12 +70,11 @@ const validateAuthToken = async (
 
 /**
  * Resets the auth token and a new token will be fetched on the next request.
- *
  * @example
  * ```js
  * resetCachedAuthToken();
  * ```
- * @version SDK: 1.28.0 | ThoughtSpot: *
+ * @version SDK: 1.28.2 | ThoughtSpot: *
  * @group Authentication / Init
  */
 export const resetCachedAuthToken = (): void => {

package/src/config.ts

@@ -5,7 +5,6 @@ import { EmbedConfig } from './types';
  * Copyright (c) 2023
  *
  * Utilities related to reading configuration objects
- *
  * @summary Config-related utils
  * @author Ayon Ghosh <ayon.ghosh@thoughtspot.com>
  */
@@ -23,7 +22,6 @@ const urlRegex = new RegExp(
 /**
  * Parse and construct the ThoughtSpot hostname or IP address
  * from the embed configuration object.
- *
  * @param config
  */
 export const getThoughtSpotHost = (config: EmbedConfig): string => {

package/src/embed/app.spec.ts

@@ -1,4 +1,9 @@
-import { AppEmbed, AppViewConfig, Page } from './app';
+import {
+    AppEmbed,
+    AppViewConfig,
+    DataPanelCustomColumnGroupsAccordionState,
+    Page,
+} from './app';
 import { init } from '../index';
 import {
     Action, AuthType, EmbedEvent, HostEvent, RuntimeFilterOp,
@@ -19,7 +24,6 @@ import { version } from '../../package.json';
 import * as config from '../config';
 import { TsEmbed, V1Embed } from './ts-embed';
 import { logger } from '../utils/logger';
-import * as auth from '../auth';
 
 const defaultViewConfig = {
     frameParams: {
@@ -36,7 +40,6 @@ beforeAll(() => {
         thoughtSpotHost,
         authType: AuthType.None,
     });
-    jest.spyOn(auth, 'postLoginService').mockImplementation(() => Promise.resolve({}));
     (window as any).ResizeObserver = window.ResizeObserver
             || jest.fn().mockImplementation(() => ({
                 disconnect: jest.fn(),
@@ -69,13 +72,13 @@ describe('App embed tests', () => {
     test('should hide the primary nav bar', async () => {
         const appEmbed = new AppEmbed(getRootEl(), {
             ...defaultViewConfig,
-            showPrimaryNavbar: false,
+            showPrimaryNavbar: true,
         } as AppViewConfig);
         appEmbed.render();
         await executeAfterWait(() => {
             expectUrlMatchesWithParams(
                 getIFrameSrc(),
-                `http://${thoughtSpotHost}/?embedApp=true&primaryNavHidden=true&profileAndHelpInNavBarHidden=false${defaultParams}${defaultParamsPost}#/home`,
+                `http://${thoughtSpotHost}/?embedApp=true&primaryNavHidden=false&profileAndHelpInNavBarHidden=false${defaultParams}${defaultParamsPost}#/home`,
             );
         });
     });
@@ -291,6 +294,21 @@ describe('App embed tests', () => {
         });
     });
 
+    test('should set enable2ColumnLayout to true in url', async () => {
+        const appEmbed = new AppEmbed(getRootEl(), {
+            ...defaultViewConfig,
+            enable2ColumnLayout: true,
+        } as AppViewConfig);
+
+        appEmbed.render();
+        await executeAfterWait(() => {
+            expectUrlMatchesWithParams(
+                getIFrameSrc(),
+                `http://${thoughtSpotHost}/?embedApp=true&profileAndHelpInNavBarHidden=false&enable2ColumnLayout=true${defaultParamsPost}#/home`,
+            );
+        });
+    });
+
     test('Should add the tag to the iframe src', async () => {
         const appEmbed = new AppEmbed(getRootEl(), {
             ...defaultViewConfig,
@@ -411,6 +429,22 @@ describe('App embed tests', () => {
         });
     });
 
+    test('Should add dataPanelCustomGroupsAccordionInitialState flag to the iframe src', async () => {
+        const appEmbed = new AppEmbed(getRootEl(), {
+            ...defaultViewConfig,
+            // eslint-disable-next-line max-len
+            dataPanelCustomGroupsAccordionInitialState: DataPanelCustomColumnGroupsAccordionState.EXPAND_FIRST,
+        } as AppViewConfig);
+
+        appEmbed.render();
+        await executeAfterWait(() => {
+            expectUrlMatchesWithParams(
+                getIFrameSrc(),
+                `http://${thoughtSpotHost}/?embedApp=true&primaryNavHidden=true&profileAndHelpInNavBarHidden=false&modularHomeExperience=false&dataPanelCustomGroupsAccordionInitialState=EXPAND_FIRST${defaultParams}${defaultParamsPost}#/home`,
+            );
+        });
+    });
+
     test('should register event handlers to adjust iframe height', async () => {
         const onSpy = jest.spyOn(AppEmbed.prototype, 'on')
             .mockImplementation((event, callback) => {

package/src/embed/app.ts

@@ -3,22 +3,16 @@
  *
  * Full application embedding
  * https://developers.thoughtspot.com/docs/?pageid=full-embed
- *
  * @summary Full app embed
  * @module
  * @author Ayon Ghosh <ayon.ghosh@thoughtspot.com>
  */
 
-import { logger } from '../utils/logger';
-import { getQueryParamString } from '../utils';
 import {
-    Param,
-    DOMSelector,
-    HostEvent,
-    ViewConfig,
-    EmbedEvent,
-    MessagePayload,
+    DOMSelector, EmbedEvent, HostEvent, MessagePayload, Param, ViewConfig,
 } from '../types';
+import { getQueryParamString } from '../utils';
+import { logger } from '../utils/logger';
 import { V1Embed } from './ts-embed';
 
 /**
@@ -56,9 +50,27 @@ export enum Page {
     SpotIQ = 'insights',
 }
 
+/**
+ * Define the initial state os column custom group accordions
+ * in data panel v2.
+ */
+export enum DataPanelCustomColumnGroupsAccordionState {
+    /**
+     * Expand all the accordion initially in data panel v2.
+     */
+    EXPAND_ALL = 'EXPAND_ALL',
+    /**
+     * Collapse all the accordions initially in data panel v2.
+     */
+    COLLAPSE_ALL = 'COLLAPSE_ALL',
+    /**
+     * Expand the first accordion and collapse the rest.
+     */
+    EXPAND_FIRST = 'EXPAND_FIRST',
+}
+
 /**
  * The view configuration for full app embedding.
- *
  * @group Embed components
  */
 export interface AppViewConfig extends Omit<ViewConfig, 'visibleTabs'> {
@@ -66,10 +78,8 @@ export interface AppViewConfig extends Omit<ViewConfig, 'visibleTabs'> {
      * If true, the top navigation bar within the ThoughtSpot app
      * is displayed. By default, the navigation bar is hidden.
      * This flag also controls the homepage left navigation bar.
-     *
      * @default true
      * @version SDK: 1.2.0 | Thoughtspot: 8.4.0.cl
-     *
      * @example
      * ```js
      * const embed = new AppEmbed('#tsEmbed', {
@@ -85,13 +95,12 @@ export interface AppViewConfig extends Omit<ViewConfig, 'visibleTabs'> {
      * nav-bars are visible, this flag will only hide the homepage left nav-bar.
      * The showPrimaryNavbar flag takes precedence over the hideHomepageLeftNav.
      *
+     *
      * **Note**: This option does not apply to the classic homepage.
      * To access the updated modular homepage, set
      * `modularHomeExperience` to `true` (available as Early Access feature in 9.12.5.cl).
-     *
      * @default false
      * @version SDK: 1.28.0 | Thoughtspot: 9.12.5.cl
-     *
      * @example
      * ```js
      * const embed = new AppEmbed('#tsEmbed', {
@@ -106,7 +115,6 @@ export interface AppViewConfig extends Omit<ViewConfig, 'visibleTabs'> {
      * Global nav-bar. By default, these buttons are visible on the nav-bar.
      * @default false
      * @version SDK: 1.2.0 | Thoughtspot: 8.4.0.cl
-     *
      * @example
      * ```js
      * const embed = new AppEmbed('#tsEmbed', {
@@ -120,13 +128,12 @@ export interface AppViewConfig extends Omit<ViewConfig, 'visibleTabs'> {
      * Control the visibility of the application switcher button on the nav-bar.
      * By default, the application switcher is shown.
      *
+     *
      * **Note**: This option does not apply to the classic homepage.
      * To access the updated modular homepage, set
      * `modularHomeExperience` to `true` (available as Early Access feature in 9.12.5.cl).
-     *
      * @version SDK: 1.28.0 | Thoughtspot: 9.12.5.cl
      * @default false
-     *
      * @example
      * ```js
      * const embed = new AppEmbed('#tsEmbed', {
@@ -140,13 +147,12 @@ export interface AppViewConfig extends Omit<ViewConfig, 'visibleTabs'> {
      * Control the visibility of the Org switcher button on the nav-bar.
      * By default, the Org switcher button is shown.
      *
+     *
      * **Note**: This option does not apply to the classic homepage.
      * To access the updated modular homepage, set
      * `modularHomeExperience` to `true` (available as Early Access feature in 9.12.5.cl).
-     *
      * @version SDK: 1.28.0 | Thoughtspot: 9.12.5.cl
      * @default true
-     *
      * @example
      * ```js
      * const embed = new AppEmbed('#tsEmbed', {
@@ -164,9 +170,7 @@ export interface AppViewConfig extends Omit<ViewConfig, 'visibleTabs'> {
      *
      * For eg, if you want the component to open to a specific Liveboard
      * you could set the path to `pinboard/<liveboardId>/tab/<tabId>`.
-     *
      * @version SDK: 1.1.0 | Thoughtspot: 9.4.0.cl
-     *
      * @example
      * ```js
      * const embed = new AppEmbed('#tsEmbed', {
@@ -182,9 +186,7 @@ export interface AppViewConfig extends Omit<ViewConfig, 'visibleTabs'> {
      *
      * Use this to open to particular page in the app. To open to a specific
      * path within the app, use the `path` attribute which is more flexible.
-     *
      * @version SDK: 1.1.0 | Thoughtspot: 9.4.0.cl
-     *
      * @example
      * ```js
      * const embed = new AppEmbed('#tsEmbed', {
@@ -198,7 +200,6 @@ export interface AppViewConfig extends Omit<ViewConfig, 'visibleTabs'> {
      * This puts a filter tag on the application. All metadata lists in the
      * application, such as Liveboards and answers, would be filtered by this
      * tag.
-     *
      * @version SDK: 1.1.0 | Thoughtspot: 9.4.0.cl
      * @example
      * ```js
@@ -211,9 +212,7 @@ export interface AppViewConfig extends Omit<ViewConfig, 'visibleTabs'> {
     tag?: string;
     /**
      * The array of GUIDs to be hidden
-     *
      * @version SDK: 1.11.0 | ThoughtSpot: 8.3.0.cl, 8.4.1-sw
-     *
      * @example
      * ```js
      * const embed = new AppEmbed('#tsEmbed', {
@@ -229,7 +228,6 @@ export interface AppViewConfig extends Omit<ViewConfig, 'visibleTabs'> {
     /**
      * Render liveboards using the new v2 rendering mode
      * This is a transient flag which is primarily meant for internal use
-     *
      * @version SDK: 1.11.0 | ThoughtSpot: 8.3.0.cl, 8.4.1-sw
      * @hidden
      */
@@ -237,9 +235,7 @@ export interface AppViewConfig extends Omit<ViewConfig, 'visibleTabs'> {
     /**
      * If set to true, the Search Assist feature is enabled.
      * @default true
-     *
      * @version SDK: 1.13.0 | ThoughtSpot: 8.5.0.cl, 8.8.1-sw
-     *
      * @example
      * ```js
      * const embed = new AppEmbed('#tsEmbed', {
@@ -252,9 +248,7 @@ export interface AppViewConfig extends Omit<ViewConfig, 'visibleTabs'> {
     /**
      * If set to true, the embedded object container dynamically resizes
      * according to the height of the pages which support fullHeight mode.
-     *
      * @version SDK: 1.21.0 | ThoughtSpot: 9.4.0.cl, 9.4.0-sw
-     *
      * @example
      * ```js
      * const embed = new AppEmbed('#tsEmbed', {
@@ -266,7 +260,6 @@ export interface AppViewConfig extends Omit<ViewConfig, 'visibleTabs'> {
     fullHeight?: boolean;
     /**
      * Flag to control Data panel experience
-     *
      * @default false
      * @version SDK: 1.26.0 | Thoughtspot: 9.7.0.cl
      * @hidden
@@ -274,10 +267,8 @@ export interface AppViewConfig extends Omit<ViewConfig, 'visibleTabs'> {
     dataPanelV2?: boolean;
     /**
      * Show or hide Liveboard header
-     *
      * @version SDK: 1.26.0 | Thoughtspot: 9.7.0.cl
      * @default false
-     *
      * @example
      * ```js
      * const embed = new AppEmbed('#tsEmbed', {
@@ -289,10 +280,8 @@ export interface AppViewConfig extends Omit<ViewConfig, 'visibleTabs'> {
     hideLiveboardHeader?: boolean;
     /**
      * Show or hide Liveboard title
-     *
      * @version SDK: 1.26.0 | Thoughtspot: 9.7.0.cl
      * @default false
-     *
      * @example
      * ```js
      * const embed = new AppEmbed('#tsEmbed', {
@@ -304,10 +293,8 @@ export interface AppViewConfig extends Omit<ViewConfig, 'visibleTabs'> {
     showLiveboardTitle?: boolean;
     /**
      * Show or hide Liveboard description
-     *
      * @version SDK: 1.26.0 | Thoughtspot: 9.7.0.cl
      * @default false
-     *
      * @example
      * ```js
      * const embed = new AppEmbed('#tsEmbed', {
@@ -319,10 +306,8 @@ export interface AppViewConfig extends Omit<ViewConfig, 'visibleTabs'> {
     showLiveboardDescription?: boolean;
     /**
      * Flag to control new Modular Home experience.
-     *
      * @default false
      * @version SDK: 1.28.0 | Thoughtspot: 9.12.5.cl
-     *
      * @example
      * ```js
      * const embed = new AppEmbed('#tsEmbed', {
@@ -334,7 +319,6 @@ export interface AppViewConfig extends Omit<ViewConfig, 'visibleTabs'> {
     modularHomeExperience?: boolean;
     /**
      * Boolean to control if Liveboard header is sticky or not.
-     *
      * @example
      * ```js
      * const embed = new AppEmbed('#embed', {
@@ -347,23 +331,71 @@ export interface AppViewConfig extends Omit<ViewConfig, 'visibleTabs'> {
     isLiveboardHeaderSticky?: boolean;
     /**
      * enable or disable ask sage
-     *
      * @version SDK: 1.29.0 | Thoughtspot: 9.12.0.cl
      * @default false
      */
     enableAskSage?: boolean;
     /**
      * To set the initial state of the search bar in case of saved-answers.
-     *
      * @version SDK: 1.32.0 | Thoughtspot: 10.0.0.cl
      * @default false
      */
     collapseSearchBarInitially?: boolean;
+    /**
+     * To enable custom column groups in data panel v2.
+     * @version SDK: 1.32.0 | Thoughtspot: 10.0.0.cl
+     * @default false
+     * @example
+     * ```js
+     * const embed = new AppEmbed('#embed', {
+     *   ... // other app view config
+     *   enableCustomColumnGroups: true,
+     * });
+     * ```
+     */
+    enableCustomColumnGroups?: boolean;
+    /**
+     * This controls the initial behaviour of custom column groups accordion.
+     * It takes DataPanelCustomColumnGroupsAccordionState enum values as input.
+     * List of different enum values:-
+     * - EXPAND_ALL: Expand all the accordion initially in data panel v2.
+     * - COLLAPSE_ALL: Collapse all the accordions initially in data panel v2.
+     * - EXPAND_FIRST: Expand the first accordion and collapse the rest.
+     * @version SDK: 1.32.0 | Thoughtspot: 10.0.0.cl
+     * @default DataPanelCustomColumnGroupsAccordionState.EXPAND_ALL
+     * @example
+     * ```js
+     * const embed = new AppEmbed('#embed', {
+     *   ... // other app view config
+     *   dataPanelCustomGroupsAccordionInitialState:
+     *      DataPanelCustomColumnGroupsAccordionState.EXPAND_ALL,
+     * });
+     * ```
+     */
+    dataPanelCustomGroupsAccordionInitialState?: DataPanelCustomColumnGroupsAccordionState;
+    /**
+     * This flag is used to enable the 2 column layout in liveboard
+     * @type {boolean}
+     * @default false
+     * @version SDK: 1.32.0 | ThoughtSpot:10.1.0.cl
+     * @example
+     * ```js
+     * const embed = new LiveboardEmbed('#embed-container', {
+     *    ... // other options
+     *    enable2ColumnLayout: true,
+     * })
+     * ```
+     */
+    enable2ColumnLayout?: boolean;
+    /**
+     * Flag to use OnBeforeSearchExecute embed event
+     * @version SDK : 1.29.0 | Thoughtspot : 10.1.0.cl
+     */
+    isOnBeforeGetVizDataInterceptEnabled?: boolean;
 }
 
 /**
  * Embeds full ThoughtSpot experience in a host application.
- *
  * @group Embed components
  */
 export class AppEmbed extends V1Embed {
@@ -406,6 +438,11 @@ export class AppEmbed extends V1Embed {
             isLiveboardHeaderSticky = true,
             enableAskSage,
             collapseSearchBarInitially = false,
+            enable2ColumnLayout,
+            enableCustomColumnGroups = false,
+            isOnBeforeGetVizDataInterceptEnabled = false,
+            /* eslint-disable-next-line max-len */
+            dataPanelCustomGroupsAccordionInitialState = DataPanelCustomColumnGroupsAccordionState.EXPAND_ALL,
         } = this.viewConfig;
 
         let params = {};
@@ -418,6 +455,7 @@ export class AppEmbed extends V1Embed {
         params[Param.ShowLiveboardTitle] = showLiveboardTitle;
         params[Param.ShowLiveboardDescription] = !!showLiveboardDescription;
         params[Param.LiveboardHeaderSticky] = isLiveboardHeaderSticky;
+        params[Param.IsFullAppEmbed] = true;
 
         params = this.getBaseQueryParams(params);
 
@@ -439,14 +477,40 @@ export class AppEmbed extends V1Embed {
             params[Param.EnableSearchAssist] = enableSearchAssist;
         }
 
+        if (enable2ColumnLayout !== undefined) {
+            params[Param.Enable2ColumnLayout] = enable2ColumnLayout;
+        }
+
         if (enableAskSage) {
             params[Param.enableAskSage] = enableAskSage;
         }
 
+        if (isOnBeforeGetVizDataInterceptEnabled) {
+            /* eslint-disable-next-line max-len */
+            params[
+                Param.IsOnBeforeGetVizDataInterceptEnabled
+            ] = isOnBeforeGetVizDataInterceptEnabled;
+        }
+
         params[Param.DataPanelV2Enabled] = dataPanelV2;
         params[Param.HideHomepageLeftNav] = hideHomepageLeftNav;
         params[Param.ModularHomeExperienceEnabled] = modularHomeExperience;
         params[Param.CollapseSearchBarInitially] = collapseSearchBarInitially;
+        params[Param.EnableCustomColumnGroups] = enableCustomColumnGroups;
+        if (
+            dataPanelCustomGroupsAccordionInitialState
+                === DataPanelCustomColumnGroupsAccordionState.COLLAPSE_ALL
+            || dataPanelCustomGroupsAccordionInitialState
+                === DataPanelCustomColumnGroupsAccordionState.EXPAND_FIRST
+        ) {
+            /* eslint-disable-next-line max-len */
+            params[
+                Param.DataPanelCustomGroupsAccordionInitialState
+            ] = dataPanelCustomGroupsAccordionInitialState;
+        } else {
+            /* eslint-disable-next-line max-len */
+            params[Param.DataPanelCustomGroupsAccordionInitialState] = DataPanelCustomColumnGroupsAccordionState.EXPAND_ALL;
+        }
         const queryParams = getQueryParamString(params, true);
 
         return queryParams;
@@ -454,7 +518,6 @@ export class AppEmbed extends V1Embed {
 
     /**
      * Constructs the URL of the ThoughtSpot app page to be rendered.
-     *
      * @param pageId The ID of the page to be embedded.
      */
     public getIFrameSrc(): string {
@@ -471,7 +534,6 @@ export class AppEmbed extends V1Embed {
     /**
      * Set the iframe height as per the computed height received
      * from the ThoughtSpot app.
-     *
      * @param data The event payload
      */
     protected updateIFrameHeight = (data: MessagePayload) => {
@@ -492,7 +554,6 @@ export class AppEmbed extends V1Embed {
 
     /**
      * Gets the ThoughtSpot route of the page for a particular page ID.
-     *
      * @param pageId The identifier for a page in the ThoughtSpot app.
      * @param modularHomeExperience
      */
@@ -518,7 +579,6 @@ export class AppEmbed extends V1Embed {
 
     /**
      * Formats the path provided by the user.
-     *
      * @param path The URL path.
      * @returns The URL path that the embedded app understands.
      */
@@ -539,7 +599,6 @@ export class AppEmbed extends V1Embed {
      * Navigate to particular page for app embed. eg:answers/pinboards/home
      * This is used for embedding answers, pinboards, visualizations and full application
      * only.
-     *
      * @param path string | number The string, set to iframe src and navigate to new page
      * eg: appEmbed.navigateToPage('pinboards')
      * When used with `noReload` (default: true) this can also be a number
@@ -571,7 +630,6 @@ export class AppEmbed extends V1Embed {
 
     /**
      * Renders the embedded application pages in the ThoughtSpot app.
-     *
      * @param renderOptions An object containing the page ID
      * to be embedded.
      */

package/src/embed/base.spec.ts

@@ -1,11 +1,11 @@
 /* eslint-disable no-console */
 /* eslint-disable @typescript-eslint/no-shadow */
 import EventEmitter from 'eventemitter3';
-import { EmbedConfig } from '../index';
 import * as auth from '../auth';
-import * as authService from '../utils/authService/authService';
 import * as authTokenService from '../authToken';
 import * as index from '../index';
+import { EmbedConfig } from '../index';
+import * as authService from '../utils/authService/authService';
 import * as base from './base';
 import * as embedConfigInstance from './embedConfig';
 
@@ -13,8 +13,8 @@ import {
     executeAfterWait,
     getAllIframeEl,
     getDocumentBody,
-    getRootEl,
     getIFrameSrc,
+    getRootEl,
 } from '../test/test-utils';
 import * as tokenizedFetchInstance from '../tokenizedFetch';
 import { logger } from '../utils/logger';
@@ -28,7 +28,6 @@ describe('Base TS Embed', () => {
             thoughtSpotHost,
             authType: index.AuthType.None,
         }) as EventEmitter;
-        jest.spyOn(auth, 'postLoginService').mockImplementation(() => Promise.resolve({}));
     });
 
     beforeEach(() => {
@@ -260,6 +259,8 @@ describe('Base TS Embed', () => {
         expect(prefetchIframe.length).toBe(1);
         const firstIframe = <HTMLIFrameElement>prefetchIframe[0];
         expect(firstIframe.src).toBe(url);
+        expect(firstIframe.style.width).toBe('0px');
+        expect(firstIframe.classList.contains('prefetchIframeNum-0')).toBe(true);
     });
 
     test('Should add the prefetch iframe when prefetch is called with multiple options', async () => {