Fix UT

Signed-off-by: Heng Qian <[email protected]>
qianheng-aws · Sep 4, 2024 · a0e1a89 · a0e1a89
1 parent d2278f1
commit a0e1a89
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/public/components/incontext_insight/generate_popover_body.test.tsx b/public/components/incontext_insight/generate_popover_body.test.tsx
@@ -37,7 +37,7 @@ describe('GeneratePopoverBody', () => {
     contextProvider: jest.fn(),
     suggestions: ['Test summarization question'],
     datasourceId: 'test-datasource',
-    key: 'test-key',
+    key: 'alerts',
   };
 
   const closePopoverMock = jest.fn();
@@ -49,7 +49,7 @@ describe('GeneratePopoverBody', () => {
         case SUMMARY_ASSISTANT_API.SUMMARIZE:
           value = {
             summary: 'Generated summary content',
-            insightAgentId: 'insight_agent_id',
+            insightAgentIdExists: true,
           };
           break;
 
@@ -122,7 +122,7 @@ describe('GeneratePopoverBody', () => {
         case SUMMARY_ASSISTANT_API.SUMMARIZE:
           value = {
             summary: 'Generated summary content',
-            insightAgentId: undefined,
+            insightAgentIdExists: false,
           };
           break;
 
@@ -192,7 +192,7 @@ describe('GeneratePopoverBody', () => {
         case SUMMARY_ASSISTANT_API.SUMMARIZE:
           value = {
             summary: 'Generated summary content',
-            insightAgentId: 'insight_agent_id',
+            insightAgentIdExists: true,
           };
           break;