[Multiple Datasource] Fix create data source wizard test

Signed-off-by: Zilong Xia <zilongx@amazon.com>
ZilongX · Apr 4, 2024 · 937926d · 937926d
1 parent cc34f97
commit 937926d
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 7 deletions.
diff --git a/...management/public/components/create_data_source_wizard/create_data_source_wizard.test.tsx b/...management/public/components/create_data_source_wizard/create_data_source_wizard.test.tsx
@@ -5,7 +5,7 @@
 
 import React from 'react';
 import {
-  fetchDataSourceVersion,
+  fetchDataSourceMetaData,
   getMappedDataSources,
   mockDataSourceAttributesWithAuth,
   mockManagementPlugin,
@@ -28,8 +28,8 @@ describe('Datasource Management: Create Datasource Wizard', () => {
   describe('case1: should load resources successfully', () => {
     beforeEach(async () => {
       spyOn(utils, 'getDataSources').and.returnValue(Promise.resolve(getMappedDataSources));
-      spyOn(utils, 'fetchDataSourceVersion').and.returnValue(
-        Promise.resolve(fetchDataSourceVersion)
+      spyOn(utils, 'fetchDataSourceMetaData').and.returnValue(
+        Promise.resolve(fetchDataSourceMetaData)
       );
       await act(async () => {
         component = mount(

diff --git a/...urce_management/public/components/create_data_source_wizard/create_data_source_wizard.tsx b/...urce_management/public/components/create_data_source_wizard/create_data_source_wizard.tsx
@@ -20,7 +20,7 @@ import {
   createSingleDataSource,
   getDataSources,
   testConnection,
-  fetchDataMetaData,
+  fetchDataSourceMetaData,
   handleSetDefaultDatasource,
 } from '../utils';
 import { LoadingMask } from '../loading_mask';
@@ -76,7 +76,7 @@ export const CreateDataSourceWizard: React.FunctionComponent<CreateDataSourceWiz
     setIsLoading(true);
     try {
       // Fetch data source metadata from added OS/ES domain/cluster
-      const metadata = await fetchDataMetaData(http, attributes);
+      const metadata = await fetchDataSourceMetaData(http, attributes);
       attributes.dataSourceVersion = metadata.dataSourceVersion;
       attributes.installedPlugins = metadata.installedPlugins;
       await createSingleDataSource(savedObjects.client, attributes);

diff --git a/src/plugins/data_source_management/public/components/utils.ts b/src/plugins/data_source_management/public/components/utils.ts
@@ -203,7 +203,7 @@ export async function testConnection(
   });
 }
 
-export async function fetchDataMetaData(
+export async function fetchDataSourceMetaData(
   http: HttpStart,
   { endpoint, auth: { type, credentials } }: DataSourceAttributes,
   dataSourceID?: string

diff --git a/src/plugins/data_source_management/public/mocks.ts b/src/plugins/data_source_management/public/mocks.ts
@@ -232,8 +232,9 @@ export const getMappedDataSources = [
   },
 ];
 
-export const fetchDataSourceVersion = {
+export const fetchDataSourceMetaData = {
   dataSourceVersion: '2.11.0',
+  installedPlugins: ['opensearch-ml', 'opensearch-sql'],
 };
 
 export const mockDataSourceAttributesWithAuth = {