[Security Solution] Security Assistant: updates mocks (elastic#20)

## Summary - updates mocks to fix Jest tests
kertal · Jun 2, 2023 · ff5d0dd · ff5d0dd
1 parent 71f0da8
commit ff5d0dd
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 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
@@ -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
@@ -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={