update mock function

Signed-off-by: tygao <tygao@amazon.com>
raintygao · Jun 5, 2024 · be5233d · be5233d
1 parent 1c8c44c
commit be5233d
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 7 deletions.
diff --git a/public/components/__tests__/chat_window_header_title.test.tsx b/public/components/__tests__/chat_window_header_title.test.tsx
@@ -42,7 +42,7 @@ const setup = ({
         }),
         reload: jest.fn(),
       },
-      dataSource: rest.dataSource ?? new DataSourceServiceMock(),
+      dataSource: rest.dataSource ?? new DataSourceServiceMock(false),
     },
   };
   useCoreMock.services.http.put.mockImplementation(() => Promise.resolve());

diff --git a/public/services/data_source_service.mock.ts b/public/services/data_source_service.mock.ts
@@ -4,17 +4,19 @@
  */
 
 export class DataSourceServiceMock {
-  private _isMDSEnabled = false;
+  private _isMDSEnabled = true;
   constructor(isMDSEnabled?: boolean) {
-    this._isMDSEnabled = isMDSEnabled ?? false;
+    this._isMDSEnabled = isMDSEnabled ?? true;
   }
 
   getDataSourceQuery() {
-    const dataSourceId = {
-      dataSourceId: this._isMDSEnabled ? 'data_source_id' : '',
-    };
+    const result = this._isMDSEnabled
+      ? {
+          dataSourceId: '',
+        }
+      : {};
     return new Promise((resolve) => {
-      resolve(dataSourceId);
+      resolve(result);
     });
   }