From ff5d0dd8ee42c14e97a1c5bfe24ce0692e5eb558 Mon Sep 17 00:00:00 2001
From: Andrew Macri <andrew.macri@elastic.co>
Date: Fri, 2 Jun 2023 09:56:55 -0600
Subject: [PATCH] [Security Solution] Security Assistant: updates mocks (#20)

## Summary

- updates mocks to fix Jest tests
---
 .../common/mock/storybook_providers.tsx       | 10 ++++++----
 .../public/common/mock/test_providers.tsx     | 20 +++++++++++--------
 2 files changed, 18 insertions(+), 12 deletions(-)

diff --git a/x-pack/plugins/security_solution/public/common/mock/storybook_providers.tsx b/x-pack/plugins/security_solution/public/common/mock/storybook_providers.tsx
index c5a5cd99ea6dc..c40fd2b6cebf3 100644
--- a/x-pack/plugins/security_solution/public/common/mock/storybook_providers.tsx
+++ b/x-pack/plugins/security_solution/public/common/mock/storybook_providers.tsx
@@ -55,8 +55,10 @@ const KibanaReactContext = createKibanaReactContext(coreMock);
  */
 export const StorybookProviders: React.FC = ({ children }) => {
   const store = createStore(mockGlobalState, SUB_PLUGINS_REDUCER, kibanaObservable, storage);
-  const http = httpServiceMock.createStartContract({ basePath: '/test' });
   const actionTypeRegistry = actionTypeRegistryMock.create();
+  const mockGetInitialConversations = jest.fn(() => ({}));
+  const mockGetComments = jest.fn(() => []);
+  const mockHttp = httpServiceMock.createStartContract({ basePath: '/test' });
 
   return (
     <I18nProvider>
@@ -67,10 +69,10 @@ export const StorybookProviders: React.FC = ({ children }) => {
               <AssistantProvider
                 actionTypeRegistry={actionTypeRegistry}
                 augmentMessageCodeBlocks={jest.fn()}
-                getComments={jest.fn()}
-                getInitialConversations={jest.fn()}
-                http={http}
+                getComments={mockGetComments}
+                getInitialConversations={mockGetInitialConversations}
                 setConversations={jest.fn()}
+                http={mockHttp}
               >
                 {children}
               </AssistantProvider>
diff --git a/x-pack/plugins/security_solution/public/common/mock/test_providers.tsx b/x-pack/plugins/security_solution/public/common/mock/test_providers.tsx
index f55c39c168fce..89f5a6351411f 100644
--- a/x-pack/plugins/security_solution/public/common/mock/test_providers.tsx
+++ b/x-pack/plugins/security_solution/public/common/mock/test_providers.tsx
@@ -64,8 +64,10 @@ export const TestProvidersComponent: React.FC<Props> = ({
   cellActions = [],
 }) => {
   const queryClient = new QueryClient();
-  const http = httpServiceMock.createStartContract({ basePath: '/test' });
   const actionTypeRegistry = actionTypeRegistryMock.create();
+  const mockGetInitialConversations = jest.fn(() => ({}));
+  const mockGetComments = jest.fn(() => []);
+  const mockHttp = httpServiceMock.createStartContract({ basePath: '/test' });
 
   return (
     <I18nProvider>
@@ -75,10 +77,10 @@ export const TestProvidersComponent: React.FC<Props> = ({
             <AssistantProvider
               actionTypeRegistry={actionTypeRegistry}
               augmentMessageCodeBlocks={jest.fn()}
-              getComments={jest.fn()}
-              getInitialConversations={jest.fn()}
-              http={http}
+              getComments={mockGetComments}
+              getInitialConversations={mockGetInitialConversations}
               setConversations={jest.fn()}
+              http={mockHttp}
             >
               <QueryClientProvider client={queryClient}>
                 <ExpandableFlyoutProvider>
@@ -109,8 +111,10 @@ const TestProvidersWithPrivilegesComponent: React.FC<Props> = ({
   onDragEnd = jest.fn(),
   cellActions = [],
 }) => {
-  const http = httpServiceMock.createStartContract({ basePath: '/test' });
   const actionTypeRegistry = actionTypeRegistryMock.create();
+  const mockGetInitialConversations = jest.fn(() => ({}));
+  const mockGetComments = jest.fn(() => []);
+  const mockHttp = httpServiceMock.createStartContract({ basePath: '/test' });
 
   return (
     <I18nProvider>
@@ -120,10 +124,10 @@ const TestProvidersWithPrivilegesComponent: React.FC<Props> = ({
             <AssistantProvider
               actionTypeRegistry={actionTypeRegistry}
               augmentMessageCodeBlocks={jest.fn()}
-              getComments={jest.fn()}
-              getInitialConversations={jest.fn()}
-              http={http}
+              getComments={mockGetComments}
+              getInitialConversations={mockGetInitialConversations}
               setConversations={jest.fn()}
+              http={mockHttp}
             >
               <UserPrivilegesProvider
                 kibanaCapabilities={