tiny fix

yext · Jul 9, 2024 · 033286f · 033286f
1 parent fce78e5
commit 033286f
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/packages/chat-headless/tests/chatheadless.test.ts b/packages/chat-headless/tests/chatheadless.test.ts
@@ -30,10 +30,11 @@ const mockedMetaState: MetaState = {
 
 beforeEach(() => {
   jest.spyOn(coreLib, "provideChatCore").mockImplementation(() => {
-    return {
+    const client: ChatHttpClient = {
       getNextMessage: jest.fn(),
-      on: jest.fn(),
-    } as Partial<ChatHttpClient> as ChatHttpClient;
+      streamNextMessage: jest.fn(),
+    };
+    return client;
   });
   localStorage.clear();
 });