update core context mock

Signed-off-by: tygao <[email protected]>
opensearch-project · May 28, 2024 · 7b8ec87 · 7b8ec87
1 parent d1f9ce8
commit 7b8ec87
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/public/contexts/__mocks__/core_context.tsx b/public/contexts/__mocks__/core_context.tsx
@@ -5,6 +5,7 @@
 
 import { BehaviorSubject } from 'rxjs';
 import { coreMock } from '../../../../../src/core/public/mocks';
+import { DataSourceServiceMock } from '../../services/data_source_service.mock';
 
 export const useCore = jest.fn(() => {
   const useCoreMock = {
@@ -24,11 +25,7 @@ export const useCore = jest.fn(() => {
         load: jest.fn(),
       },
       conversationLoad: {},
-      dataSource: {
-        getDataSourceQuery() {
-          return { dataSourceId: '' };
-        },
-      },
+      dataSource: new DataSourceServiceMock(),
     },
   };
   useCoreMock.services.http.delete.mockReturnValue(Promise.resolve());