@thoughtspot/visual-embed-sdk 1.37.1 → 1.38.0-alpha.1

package/src/embed/bodyless-conversation.ts

@@ -8,14 +8,28 @@ import { getQueryParamString } from '../utils';
  * Configuration for bodyless conversation options.
  * @group Embed components
  */
-export interface BodylessConversationViewConfig extends ViewConfig {
+export interface SpotterAgentEmbedViewConfig extends ViewConfig {
     /**
      * The ID of the worksheet to use for the conversation.
      */
     worksheetId: string;
+    
+    /**
+     * Optional CSS class name to add to the container div.
+     */
+    containerClassName?: string;
 }
 
-interface ConversationMessageViewConfig extends BodylessConversationViewConfig {
+/**
+ * Configuration for conversation options.
+ * @deprecated from SDK: 1.38.0 | ThoughtSpot: 10.10.0.cl
+ * Use {@link SpotterAgentEmbedViewConfig} instead
+ * @group Embed components
+ */
+// eslint-disable-next-line @typescript-eslint/no-empty-object-type
+export interface BodylessConversationViewConfig extends SpotterAgentEmbedViewConfig {}
+
+interface SpotterAgentMessageViewConfig extends SpotterAgentEmbedViewConfig {
     sessionId: string;
     genNo: number;
     acSessionId: string;
@@ -23,7 +37,7 @@ interface ConversationMessageViewConfig extends BodylessConversationViewConfig {
 }
 
 class ConversationMessage extends TsEmbed {
-    constructor(container: HTMLElement, protected viewConfig: ConversationMessageViewConfig) {
+    constructor(container: HTMLElement, protected viewConfig: SpotterAgentMessageViewConfig) {
         viewConfig.embedComponentType = 'bodyless-conversation';
         super(container, viewConfig);
     }
@@ -39,6 +53,7 @@ class ConversationMessage extends TsEmbed {
         const queryParams = this.getBaseQueryParams();
 
         queryParams[Param.HideActions] = [...(queryParams[Param.HideActions] ?? [])];
+        queryParams[Param.isSpotterAgentEmbed] = true;
         let query = '';
         const queryParamsString = getQueryParamString(queryParams, true);
         if (queryParamsString) {
@@ -68,9 +83,9 @@ class ConversationMessage extends TsEmbed {
  * chatbots or other conversational interfaces.
  * @example
  * ```js
- * import { BodylessConversation } from '@thoughtspot/visual-embed-sdk';
+ * import { SpotterAgentEmbed } from '@thoughtspot/visual-embed-sdk';
  *
- * const conversation = new BodylessConversation({
+ * const conversation = new SpotterAgentEmbed({
  *  worksheetId: 'worksheetId',
  * });
  *
@@ -80,12 +95,12 @@ class ConversationMessage extends TsEmbed {
  * document.body.appendChild(container); // or to any other element
  * ```
  * @group Embed components
- * @version SDK: 1.33.1 | ThoughtSpot: 10.5.0.cl
+ * @version SDK: 1.37.0 | ThoughtSpot: 10.9.0.cl
  */
-export class BodylessConversation {
+export class SpotterAgentEmbed {
     private conversationService: ConversationService;
 
-    constructor(private viewConfig: BodylessConversationViewConfig) {
+    constructor(private viewConfig: SpotterAgentEmbedViewConfig) {
         const embedConfig = getEmbedConfig();
 
         this.conversationService = new ConversationService(
@@ -101,6 +116,10 @@ export class BodylessConversation {
         }
 
         const container = document.createElement('div');
+        if (this.viewConfig.containerClassName) {
+            container.className = this.viewConfig.containerClassName;
+        }
+
         const embed = new ConversationMessage(container, {
             ...this.viewConfig,
             sessionId: data.sessionId,
@@ -112,3 +131,29 @@ export class BodylessConversation {
         return { container, viz: embed };
     }
 }
+
+/**
+ * Create a conversation embed, which can be integrated inside
+ * chatbots or other conversational interfaces.
+ * @deprecated from SDK: 1.38.0 | ThoughtSpot: 10.10.0.cl
+ * Use {@link SpotterAgentEmbed} instead
+ * @example
+ * ```js
+ * import { SpotterAgentEmbed } from '@thoughtspot/visual-embed-sdk';
+ *
+ * const conversation = new SpotterAgentEmbed({
+ *  worksheetId: 'worksheetId',
+ * });
+ *
+ * const { container, error } = await conversation.sendMessage('show me sales by region');
+ *
+ * // append the container to the DOM
+ * document.body.appendChild(container); // or to any other element
+ * ```
+ * @group Embed components
+ */
+export class BodylessConversation extends SpotterAgentEmbed {
+    constructor(viewConfig: BodylessConversationViewConfig) {
+        super(viewConfig);
+    }
+}

package/src/embed/conversation.spec.ts

@@ -1,6 +1,6 @@
 import {
-    ConversationEmbed,
-    ConversationViewConfig,
+    SpotterEmbed,
+    SpotterEmbedViewConfig,
 } from './conversation';
 import * as authInstance from '../auth';
 import { init } from '../index';
@@ -28,14 +28,14 @@ beforeAll(() => {
 
 describe('ConversationEmbed', () => {
     it('should render the conversation embed', async () => {
-        const viewConfig: ConversationViewConfig = {
+        const viewConfig: SpotterEmbedViewConfig = {
             worksheetId: 'worksheetId',
             searchOptions: {
                 searchQuery: 'searchQuery',
             },
         };
 
-        const conversationEmbed = new ConversationEmbed(getRootEl(), viewConfig);
+        const conversationEmbed = new SpotterEmbed(getRootEl(), viewConfig);
         await conversationEmbed.render();
         expectUrlMatchesWithParams(
             getIFrameSrc(),
@@ -44,7 +44,7 @@ describe('ConversationEmbed', () => {
     });
 
     it('should render the conversation embed with worksheets disabled', async () => {
-        const viewConfig: ConversationViewConfig = {
+        const viewConfig: SpotterEmbedViewConfig = {
             worksheetId: 'worksheetId',
             searchOptions: {
                 searchQuery: 'searchQuery',
@@ -52,7 +52,7 @@ describe('ConversationEmbed', () => {
             disableSourceSelection: true,
         };
 
-        const conversationEmbed = new ConversationEmbed(getRootEl(), viewConfig);
+        const conversationEmbed = new SpotterEmbed(getRootEl(), viewConfig);
         await conversationEmbed.render();
         expectUrlMatchesWithParams(
             getIFrameSrc(),
@@ -61,7 +61,7 @@ describe('ConversationEmbed', () => {
     });
 
     it('should render the conversation embed with spotter limitations text if flag is set', async () => {
-        const viewConfig: ConversationViewConfig = {
+        const viewConfig: SpotterEmbedViewConfig = {
             worksheetId: 'worksheetId',
             searchOptions: {
                 searchQuery: 'searchQuery',
@@ -69,7 +69,7 @@ describe('ConversationEmbed', () => {
             showSpotterLimitations: true,
         };
 
-        const conversationEmbed = new ConversationEmbed(getRootEl(), viewConfig);
+        const conversationEmbed = new SpotterEmbed(getRootEl(), viewConfig);
         await conversationEmbed.render();
         expectUrlMatchesWithParams(
             getIFrameSrc(),
@@ -78,7 +78,7 @@ describe('ConversationEmbed', () => {
     });
 
     it('should render the conversation embed with sample questions hidden', async () => {
-        const viewConfig: ConversationViewConfig = {
+        const viewConfig: SpotterEmbedViewConfig = {
             worksheetId: 'worksheetId',
             searchOptions: {
                 searchQuery: 'searchQuery',
@@ -86,7 +86,7 @@ describe('ConversationEmbed', () => {
             hideSampleQuestions: true,
         };
 
-        const conversationEmbed = new ConversationEmbed(getRootEl(), viewConfig);
+        const conversationEmbed = new SpotterEmbed(getRootEl(), viewConfig);
         await conversationEmbed.render();
         expectUrlMatchesWithParams(
             getIFrameSrc(),
@@ -95,7 +95,7 @@ describe('ConversationEmbed', () => {
     });
 
     it('should render the conversation embed with worksheets hidden', async () => {
-        const viewConfig: ConversationViewConfig = {
+        const viewConfig: SpotterEmbedViewConfig = {
             worksheetId: 'worksheetId',
             searchOptions: {
                 searchQuery: 'searchQuery',
@@ -103,7 +103,7 @@ describe('ConversationEmbed', () => {
             hideSourceSelection: true,
         };
 
-        const conversationEmbed = new ConversationEmbed(getRootEl(), viewConfig);
+        const conversationEmbed = new SpotterEmbed(getRootEl(), viewConfig);
         await conversationEmbed.render();
         expectUrlMatchesWithParams(
             getIFrameSrc(),
@@ -112,13 +112,13 @@ describe('ConversationEmbed', () => {
     });
 
     it('should handle error when worksheetId is not provided', async () => {
-        const viewConfig: ConversationViewConfig = {
+        const viewConfig: SpotterEmbedViewConfig = {
             worksheetId: '',
             searchOptions: {
                 searchQuery: 'searchQuery',
             },
         };
-        const conversationEmbed = new ConversationEmbed(getRootEl(), viewConfig);
+        const conversationEmbed = new SpotterEmbed(getRootEl(), viewConfig);
         (conversationEmbed as any).handleError = jest.fn();
         await conversationEmbed.render();
         expect((conversationEmbed as any).handleError).toHaveBeenCalledWith(
@@ -127,7 +127,7 @@ describe('ConversationEmbed', () => {
     });
 
     it('should render the conversation embed if data panel v2 flag is true', async () => {
-        const viewConfig: ConversationViewConfig = {
+        const viewConfig: SpotterEmbedViewConfig = {
             worksheetId: 'worksheetId',
             searchOptions: {
                 searchQuery: 'searchQuery',
@@ -135,7 +135,7 @@ describe('ConversationEmbed', () => {
             dataPanelV2: true,
         };
 
-        const conversationEmbed = new ConversationEmbed(getRootEl(), viewConfig);
+        const conversationEmbed = new SpotterEmbed(getRootEl(), viewConfig);
         await conversationEmbed.render();
         expectUrlMatchesWithParams(
             getIFrameSrc(),

package/src/embed/conversation.ts

@@ -15,10 +15,10 @@ export interface SearchOptions {
 }
 
 /**
- * The configuration for the embedded conversationEmbed options.
+ * The configuration for the embedded spotterEmbed options.
  * @group Embed components
  */
-export interface ConversationViewConfig extends ViewConfig {
+export interface SpotterEmbedViewConfig extends ViewConfig {
     /**
      * The ID of the worksheet to use for the conversation.
      */
@@ -32,7 +32,7 @@ export interface ConversationViewConfig extends ViewConfig {
      * but still display the selected data source.
      * @example
      * ```js
-     * const embed = new ConversationEmbed('#tsEmbed', {
+     * const embed = new SpotterEmbed('#tsEmbed', {
      *    ... // other options
      *    disableSourceSelection : true,
      * })
@@ -44,7 +44,7 @@ export interface ConversationViewConfig extends ViewConfig {
      * hideSourceSelection : Hide data source selection
      * @example
      * ```js
-     * const embed = new ConversationEmbed('#tsEmbed', {
+     * const embed = new SpotterEmbed('#tsEmbed', {
      *    ... // other options
      *    hideSourceSelection : true,
      * })
@@ -71,7 +71,7 @@ export interface ConversationViewConfig extends ViewConfig {
      * default is false.
      * @example
      * ```js
-     * const embed = new ConversationEmbed('#tsEmbed', {
+     * const embed = new SpotterEmbed('#tsEmbed', {
      *    ... // other options
      *    showSpotterLimitations : true,
      * })
@@ -84,7 +84,7 @@ export interface ConversationViewConfig extends ViewConfig {
      * the initial screen of the conversation.
      * @example
      * ```js
-     * const embed = new ConversationEmbed('#tsEmbed', {
+     * const embed = new SpotterEmbed('#tsEmbed', {
      *    ... // other options
      *    hideSampleQuestions : true,
      * })
@@ -94,12 +94,21 @@ export interface ConversationViewConfig extends ViewConfig {
     hideSampleQuestions?: boolean;
 }
 
+/**
+ * The configuration for the embedded spotterEmbed options.
+ * @deprecated from SDK: 1.38.0 | ThoughtSpot: 10.10.0.cl
+ * Use {@link SpotterEmbedViewConfig} instead
+ * @group Embed components
+ */
+// eslint-disable-next-line @typescript-eslint/no-empty-object-type
+export interface ConversationViewConfig extends SpotterEmbedViewConfig {}
+
 /**
  * Embed ThoughtSpot AI Conversation.
  * @group Embed components
  * @example
  * ```js
- * const conversation = new ConversationEmbed('#tsEmbed', {
+ * const conversation = new SpotterEmbed('#tsEmbed', {
  *   worksheetId: 'worksheetId',
  *   searchOptions: {
  *     searchQuery: 'searchQuery',
@@ -107,10 +116,10 @@ export interface ConversationViewConfig extends ViewConfig {
  * });
  * conversation.render();
  * ```
- * @version SDK: 1.33.1 | ThoughtSpot: 10.5.0.cl
+ * @version SDK: 1.37.0 | ThoughtSpot: 10.9.0.cl
  */
-export class ConversationEmbed extends TsEmbed {
-    constructor(container: HTMLElement, protected viewConfig: ConversationViewConfig) {
+export class SpotterEmbed extends TsEmbed {
+    constructor(container: HTMLElement, protected viewConfig: SpotterEmbedViewConfig) {
         viewConfig.embedComponentType = 'conversation';
         super(container, viewConfig);
     }
@@ -163,7 +172,7 @@ export class ConversationEmbed extends TsEmbed {
         return `${this.getEmbedBasePath(query)}/embed/${path}${tsPostHashParams}`;
     }
 
-    public async render(): Promise<ConversationEmbed> {
+    public async render(): Promise<SpotterEmbed> {
         await super.render();
 
         const src = this.getIframeSrc();
@@ -171,3 +180,27 @@ export class ConversationEmbed extends TsEmbed {
         return this;
     }
 }
+
+/**
+ * Embed ThoughtSpot AI Conversation.
+ * @deprecated from SDK: 1.38.0 | ThoughtSpot: 10.10.0.cl
+ * Use {@link SpotterEmbed} instead
+ * @group Embed components
+ * @example
+ * ```js
+ * const conversation = new SpotterEmbed('#tsEmbed', {
+ *   worksheetId: 'worksheetId',
+ *   searchOptions: {
+ *     searchQuery: 'searchQuery',
+ *   },
+ * });
+ * conversation.render();
+ * ```
+ * @version SDK: 1.37.0 | ThoughtSpot: 10.9.0.cl
+ */
+export class ConversationEmbed extends SpotterEmbed {
+    constructor(container: HTMLElement, protected viewConfig: ConversationViewConfig) {
+        viewConfig.embedComponentType = 'conversation';
+        super(container, viewConfig);
+    }
+}

package/src/embed/liveboard.ts

@@ -24,13 +24,8 @@ import { getAuthPromise } from './base';
 import { TsEmbed, V1Embed } from './ts-embed';
 import { addPreviewStylesIfNotPresent } from '../utils/global-styles';
 import { TriggerPayload, TriggerResponse } from './hostEventClient/contracts';
+import { logger } from '../utils/logger';
 
-const liveboardHeightWhitelistedRoutes = [
-    '/embed/viz/',
-    '/embed/insights/viz/',
-    '/tsl-editor/PINBOARD_ANSWER_BOOK/',
-    '/import-tsl/PINBOARD_ANSWER_BOOK/',
-];
 
 /**
  * The configuration for the embedded Liveboard or visualization page view.
@@ -429,11 +424,16 @@ export class LiveboardEmbed extends V1Embed {
 
     private defaultHeight = 500;
 
-    // eslint-disable-next-line no-useless-constructor
+     
     constructor(domSelector: DOMSelector, viewConfig: LiveboardViewConfig) {
         viewConfig.embedComponentType = 'LiveboardEmbed';
         super(domSelector, viewConfig);
         if (this.viewConfig.fullHeight === true) {
+            if (this.viewConfig.vizId) {
+                logger.warn('Full height is currently only supported for Liveboard embeds.' +
+                  'Using full height with vizId might lead to unexpected behavior.');
+            }
+
             this.on(EmbedEvent.RouteChange, this.setIframeHeightForNonEmbedLiveboard);
             this.on(EmbedEvent.EmbedHeight, this.updateIFrameHeight);
             this.on(EmbedEvent.EmbedIframeCenter, this.embedIframeCenter);
@@ -592,11 +592,26 @@ export class LiveboardEmbed extends V1Embed {
     };
 
     private setIframeHeightForNonEmbedLiveboard = (data: MessagePayload) => {
-        const routePath = data.data.currentPath;
-        if (liveboardHeightWhitelistedRoutes.some((path) => routePath.startsWith(path))) {
+        const { height: frameHeight } = this.viewConfig.frameParams || {};
+
+        const liveboardRelatedRoutes = [
+            '/pinboard/',
+            '/insights/pinboard/',
+            '/schedules/',
+            '/embed/viz/',
+            '/embed/insights/viz/',
+            '/liveboard/',
+            '/insights/liveboard/',
+            '/tsl-editor/PINBOARD_ANSWER_BOOK/',
+            '/import-tsl/PINBOARD_ANSWER_BOOK/',
+        ];
+
+        if (liveboardRelatedRoutes.some((path) => data.data.currentPath.startsWith(path))) {
+            // Ignore the height reset of the frame, if the navigation is
+            // only within the liveboard page.
             return;
         }
-        this.setIFrameHeight(this.defaultHeight);
+        this.setIFrameHeight(frameHeight || this.defaultHeight);
     };
 
     private setActiveTab(data: { tabId: string }) {

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

@@ -14,8 +14,6 @@ import {
     AppViewConfig,
     SageEmbed,
     SageViewConfig,
-    ConversationViewConfig,
-    ConversationEmbed,
     SearchViewConfig,
     AnswerService,
 } from '../index';

package/src/index.ts

@@ -9,7 +9,12 @@
  * @author Ayon Ghosh <ayon.ghosh@thoughtspot.com>
  */
 import {
-    AppEmbed, Page, AppViewConfig, HomePageSearchBarMode,
+    AppEmbed,
+    Page,
+    AppViewConfig,
+    HomePageSearchBarMode,
+    PrimaryNavbarVersion,
+    HomePage,
 } from './embed/app';
 import {
     init,
@@ -23,8 +28,8 @@ import {
 import { PinboardEmbed, LiveboardViewConfig, LiveboardEmbed } from './embed/liveboard';
 import { SearchEmbed, SearchViewConfig } from './embed/search';
 import { SearchBarEmbed, SearchBarViewConfig } from './embed/search-bar';
-import { BodylessConversation, BodylessConversationViewConfig } from './embed/bodyless-conversation';
-import { ConversationEmbed, ConversationViewConfig } from './embed/conversation';
+import { SpotterAgentEmbed, SpotterAgentEmbedViewConfig, BodylessConversation, BodylessConversationViewConfig} from './embed/bodyless-conversation';
+import { SpotterEmbed, SpotterEmbedViewConfig, ConversationEmbed, ConversationViewConfig } from './embed/conversation';
 import {
     AuthFailureType, AuthStatus, AuthEvent, AuthEventEmitter,
 } from './auth';
@@ -85,10 +90,14 @@ export {
     LiveboardEmbed,
     SageEmbed,
     AppEmbed,
-    BodylessConversation,
+    SpotterAgentEmbed,
+    SpotterAgentEmbedViewConfig,
     BodylessConversationViewConfig,
-    ConversationEmbed,
+    BodylessConversation,
+    SpotterEmbed,
+    SpotterEmbedViewConfig,
     ConversationViewConfig,
+    ConversationEmbed,
     AuthFailureType,
     AuthStatus,
     AuthEvent,
@@ -129,6 +138,8 @@ export {
     uploadMixpanelEvent,
     MIXPANEL_EVENT,
     HomePageSearchBarMode,
+    PrimaryNavbarVersion,
+    HomePage,
     VizPoint,
     CustomActionPayload,
     UIPassthroughEvent,

package/src/react/all-types-export.ts

@@ -9,8 +9,10 @@ export {
     PreRenderedAppEmbed,
     SageEmbed,
     PreRenderedSageEmbed,
+    SpotterEmbed,
     ConversationEmbed,
     PreRenderedConversationEmbed,
+    SpotterAgentEmbed,
     useEmbedRef,
     useInit,
 } from './index';

package/src/react/index.spec.tsx

@@ -16,6 +16,7 @@ import {
 } from '../test/test-utils';
 import {
     SearchEmbed, AppEmbed, LiveboardEmbed, useEmbedRef, SearchBarEmbed, PreRenderedLiveboardEmbed,
+    SpotterAgentEmbed
 } from './index';
 import * as allExports from './index';
 import {
@@ -235,6 +236,161 @@ describe('React Components', () => {
         });
     });
 
+    describe('SpotterAgentEmbed', () => {
+        it('Should work as a React component integrating BodylessConversation', async () => {
+            const mockDiv = document.createElement('div');
+            
+            let conversationService: any = null;
+            
+            render(
+                <SpotterAgentEmbed
+                    ref={(instance: any) => {
+                        conversationService = instance;
+                        
+                        if (instance) {
+                            instance.sendMessage = jest.fn().mockResolvedValue({
+                                container: mockDiv
+                            });
+                        }
+                    }}
+                    worksheetId="test-worksheet-id"
+                />
+            );
+
+            expect(conversationService).not.toBeNull();
+            
+            if (conversationService) {
+                expect(typeof conversationService.sendMessage).toBe('function');
+                
+                const response = await conversationService.sendMessage("What are my sales this month?");
+                expect(response.container).toBe(mockDiv);
+            }
+        });
+
+        it('Should work as a React component with ref support', () => {
+            const mockSendMessage = jest.fn().mockResolvedValue({
+                container: document.createElement('div'),
+                viz: {}
+            });
+            
+            const TestComponent = () => {
+                const conversationRef = React.useRef(null);
+                
+                const handleClick = () => {
+                    if (conversationRef.current) {
+                        conversationRef.current.sendMessage = mockSendMessage;
+                        
+                        conversationRef.current.sendMessage("Test message");
+                    }
+                };
+                
+                return (
+                    <>
+                        <SpotterAgentEmbed
+                            ref={conversationRef}
+                            worksheetId="test-worksheet-id"
+                        />
+                        <button data-testid="test-button" onClick={handleClick}>
+                            Send Message
+                        </button>
+                    </>
+                );
+            };
+            
+            const { getByTestId } = render(<TestComponent />);
+            
+            fireEvent.click(getByTestId('test-button'));
+            
+            expect(mockSendMessage).toHaveBeenCalledWith("Test message");
+        });
+
+        it('Should work with the useEmbedRef hook', () => {
+            const mockSendMessage = jest.fn().mockResolvedValue({
+                container: document.createElement('div'),
+                viz: {}
+            });
+            
+            const TestComponent = () => {
+                const embedRef = useEmbedRef<typeof SpotterAgentEmbed>();
+                
+                const handleClick = () => {
+                    if (embedRef.current) {
+                        const service = embedRef.current as unknown as { 
+                            sendMessage: typeof mockSendMessage 
+                        };
+                        
+                        service.sendMessage = mockSendMessage;
+                        
+                        service.sendMessage("Test with useEmbedRef");
+                    }
+                };
+                
+                return (
+                    <>
+                        <SpotterAgentEmbed
+                            ref={embedRef}
+                            worksheetId="test-worksheet-id"
+                        />
+                        <button data-testid="use-embed-ref-button" onClick={handleClick}>
+                            Send with useEmbedRef
+                        </button>
+                    </>
+                );
+            };
+            
+            const { getByTestId } = render(<TestComponent />);
+            
+            fireEvent.click(getByTestId('use-embed-ref-button'));
+            
+            expect(mockSendMessage).toHaveBeenCalledWith("Test with useEmbedRef");
+        });
+
+        it('Should work with the className prop', async () => {
+            let capturedInstance: any = null;
+            
+            const TestComponent = () => {
+                const embedRef = useEmbedRef<typeof SpotterAgentEmbed>();
+                
+                React.useEffect(() => {
+                    capturedInstance = embedRef.current;
+                    
+                    if (capturedInstance) {
+                        const mockConversationService = {
+                            sendMessage: jest.fn().mockResolvedValue({
+                                data: {
+                                    sessionId: 'test-session',
+                                    genNo: 1,
+                                    stateKey: {
+                                        transactionId: 'test-transaction',
+                                        generationNumber: 1
+                                    }
+                                }
+                            })
+                        };
+                        (capturedInstance as any).conversationService = mockConversationService;
+                    }
+                }, []);
+                
+                return (
+                    <SpotterAgentEmbed
+                        ref={embedRef}
+                        worksheetId="test-worksheet-id"
+                        className="embedClass"
+                    />
+                );
+            };
+            
+            render(<TestComponent />);
+            
+            expect(capturedInstance).not.toBeNull();
+            
+            if (capturedInstance) {
+                const result = await capturedInstance.sendMessage("test");
+                expect(result.container.className).toBe("embedClass");
+            }
+        });
+    });
+
     describe('PreRenderedLiveboardEmbed', () => {
         it('should preRender the liveboard ', async () => {
             const preRenderId = 'tsEmbed-pre-render-wrapper-test';
@@ -285,7 +441,7 @@ describe('allExports', () => {
     it('should not have undefined exports', () => {
         Object.keys(allExports).forEach(
             (exportKey) => expect(
-                Boolean(allExports[exportKey]),
+                Boolean(allExports[exportKey as keyof typeof allExports]),
             )
                 .toBe(true),
         );