fix: unit test

Signed-off-by: SuZhou-Joe <[email protected]>
opensearch-project · Dec 26, 2023 · 01a30d1 · 01a30d1
1 parent 40c9ba3
commit 01a30d1
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 2 deletions.
diff --git a/server/routes/chat_routes.ts b/server/routes/chat_routes.ts
@@ -209,8 +209,9 @@ export function registerChatRoutes(router: IRouter, routeOptions: RoutesOptions)
            */
           const interaction = await storageService.getInteraction(sessionId, interactionId);
           resultPayload.interactions = [interaction].filter((item) => item);
-          resultPayload.messages =
-            (await storageService.getMessagesFromInteractions(resultPayload.interactions)) || [];
+          resultPayload.messages = resultPayload.interactions.length
+            ? await storageService.getMessagesFromInteractions(resultPayload.interactions)
+            : [];
         }
 
         return response.ok({

diff --git a/server/routes/regenerate.test.ts b/server/routes/regenerate.test.ts
@@ -113,6 +113,9 @@ describe('regenerate route when rootAgentId is provided', () => {
         create_time: 'create_time',
       };
     });
+    mockAgentFrameworkStorageService.getMessagesFromInteractions.mockImplementationOnce(
+      async () => []
+    );
     const result = (await regenerateRequest({
       sessionId: 'foo',
       interactionId: 'bar',

diff --git a/server/routes/send_message.test.ts b/server/routes/send_message.test.ts
@@ -98,6 +98,9 @@ describe('send_message route when rootAgentId is provided', () => {
         create_time: 'create_time',
       };
     });
+    mockAgentFrameworkStorageService.getMessagesFromInteractions.mockImplementationOnce(
+      async () => []
+    );
     const result = (await sendMessageRequest({
       input: {
         content: '1',
@@ -193,6 +196,9 @@ describe('send_message route when rootAgentId is provided', () => {
         create_time: 'create_time',
       };
     });
+    mockAgentFrameworkStorageService.getMessagesFromInteractions.mockImplementationOnce(
+      async () => []
+    );
     const result = (await sendMessageRequest({
       input: {
         content: '1',