fix AssistanceChat.spec.tsx

Author: pd-redis

redisinsight/ui/src/components/side-panels/panels/ai-assistant/components/assistance-chat/AssistanceChat.spec.tsx

@@ -3,6 +3,7 @@ import { cloneDeep } from 'lodash'
 import {
   act,
   cleanup,
+  expectActionsToContain,
   fireEvent,
   mockedStore,
   mockedStoreFn,
@@ -122,7 +123,7 @@ describe('AssistanceChat', () => {
       sendQuestion(expect.objectContaining({ content: 'test' })),
     ])
 
-    expect(sendEventTelemetry).toBeCalledWith({
+    expect(sendEventTelemetry).toHaveBeenCalledWith({
       event: TelemetryEvent.AI_CHAT_MESSAGE_SENT,
       eventData: {
         chat: AiChatType.Assistance,
@@ -155,7 +156,7 @@ describe('AssistanceChat', () => {
 
     await waitForRiPopoverVisible()
 
-    expect(sendEventTelemetry).toBeCalledWith({
+    expect(sendEventTelemetry).toHaveBeenCalledWith({
       event: TelemetryEvent.AI_CHAT_BOT_TERMS_DISPLAYED,
       eventData: {
         chat: AiChatType.Assistance,
@@ -167,7 +168,7 @@ describe('AssistanceChat', () => {
       fireEvent.click(screen.getByTestId('ai-accept-agreements'))
     })
 
-    expect(sendEventTelemetry).toBeCalledWith({
+    expect(sendEventTelemetry).toHaveBeenCalledWith({
       event: TelemetryEvent.AI_CHAT_BOT_TERMS_ACCEPTED,
       eventData: {
         chat: AiChatType.Assistance,
@@ -204,12 +205,12 @@ describe('AssistanceChat', () => {
       fireEvent.click(screen.getByTestId('ai-chat-restart-confirm'))
     })
 
-    expect(store.getActions()).toEqual([
+    expectActionsToContain(store.getActions(), [
       ...afterRenderActions,
       removeAssistantChatHistory(),
     ])
 
-    expect(sendEventTelemetry).toBeCalledWith({
+    expect(sendEventTelemetry).toHaveBeenCalledWith({
       event: TelemetryEvent.AI_CHAT_SESSION_RESTARTED,
       eventData: {
         chat: AiChatType.Assistance,

redisinsight/ui/src/mocks/handlers/ai/assistantHandlers.ts

@@ -0,0 +1,46 @@
+import { http, HttpHandler, HttpResponse } from 'msw'
+import { ApiEndpoints } from 'uiSrc/constants'
+import { getMswURL } from 'uiSrc/utils/test-utils'
+
+const handlers: HttpHandler[] = [
+  http.get<{ id: string }>(
+    getMswURL(`/${ApiEndpoints.AI_ASSISTANT_CHATS}/:id`),
+    async ({ params }) => {
+      const { id } = params
+
+      return HttpResponse.json({ id, messages: [] }, { status: 200 })
+    },
+  ),
+  http.post<{ id: string }, { content: string }>(
+    getMswURL(`/${ApiEndpoints.AI_ASSISTANT_CHATS}/:id/messages`),
+    async ({ request }) => {
+      const { content } = await request.json()
+
+      return HttpResponse.json([content], { status: 200 })
+    },
+  ),
+  http.post<{ id: string }, { content: string }>(
+    getMswURL(`/${ApiEndpoints.AI_ASSISTANT_CHATS}`),
+    async ({ request }) => {
+      const { content } = await request.json()
+
+      return HttpResponse.json([content], { status: 200 })
+    },
+  ),
+  http.delete<{ id: string }>(
+    getMswURL(`/${ApiEndpoints.AI_ASSISTANT_CHATS}/:id`),
+    async () => {
+      return HttpResponse.text('', { status: 200 })
+    },
+  ),
+  http.options(getMswURL(`/${ApiEndpoints.AI_ASSISTANT_CHATS}*`), () => {
+    return new Response(null, {
+      status: 200,
+      headers: {
+        Allow: 'GET,HEAD,POST,DELETE',
+      },
+    })
+  }),
+]
+
+export default handlers

redisinsight/ui/src/mocks/handlers/ai/index.ts

@@ -0,0 +1,5 @@
+import { HttpHandler } from 'msw'
+import assistant from 'uiSrc/mocks/handlers/ai/assistantHandlers'
+
+const handlers: HttpHandler[] = [...assistant]
+export default handlers

redisinsight/ui/src/mocks/handlers/index.ts

@@ -9,6 +9,7 @@ import tutorials from './tutorials'
 import rdi from './rdi'
 import user from './user'
 import workbench from './workbench'
+import ai from './ai'
 
 // @ts-ignore
 export const handlers = [
@@ -23,4 +24,5 @@ export const handlers = [
   ...rdi,
   ...user,
   ...workbench,
+  ...ai,
 ]