update ut

Signed-off-by: Jackie Han <jkhanjob@gmail.com>

Author: jackiehanyang

public/pages/ConfigureModel/containers/__tests__/ConfigureModel.test.tsx

@@ -25,6 +25,14 @@ import { CoreServicesContext } from '../../../../components/CoreServices/CoreSer
 import { INITIAL_DETECTOR_DEFINITION_VALUES } from '../../../DefineDetector/utils/constants';
 import { INITIAL_MODEL_CONFIGURATION_VALUES } from '../../utils/constants';
 
+jest.mock('../../../../services', () => ({
+  ...jest.requireActual('../../../../services'),
+
+  getDataSourcePlugin: () => {
+    return false;
+  }
+}));
+
 const renderWithRouter = (isEdit: boolean = false) => ({
   ...render(
     <Provider store={configureStore(httpClientMock)}>
@@ -34,6 +42,7 @@ const renderWithRouter = (isEdit: boolean = false) => ({
             render={(props: RouteComponentProps) => (
               <CoreServicesContext.Provider value={coreServicesMock}>
                 <ConfigureModel
+                  setActionMenu={jest.fn()}
                   isEdit={isEdit}
                   detectorDefinitionValues={INITIAL_DETECTOR_DEFINITION_VALUES}
                   initialValues={INITIAL_MODEL_CONFIGURATION_VALUES}

public/pages/DefineDetector/containers/__tests__/DefineDetector.test.tsx

@@ -27,6 +27,14 @@ import {
   testDetectorDefinitionValues,
 } from '../../utils/constants';
 
+jest.mock('../../../../services', () => ({
+  ...jest.requireActual('../../../../services'),
+
+  getDataSourcePlugin: () => {
+    return false;
+  }
+}));
+
 const renderWithRouterEmpty = (isEdit: boolean = false) => ({
   ...render(
     <Provider store={configureStore(httpClientMock)}>
@@ -36,6 +44,7 @@ const renderWithRouterEmpty = (isEdit: boolean = false) => ({
             render={(props: RouteComponentProps) => (
               <CoreServicesContext.Provider value={coreServicesMock}>
                 <DefineDetector
+                  setActionMenu={jest.fn()}
                   isEdit={isEdit}
                   initialValues={INITIAL_DETECTOR_DEFINITION_VALUES}
                   {...props}

public/pages/DetectorsList/containers/List/__tests__/List.test.tsx

@@ -31,6 +31,14 @@ import { DetectorList, ListRouterParams } from '../List';
 import { DETECTOR_STATE } from '../../../../../../server/utils/constants';
 import { CoreServicesContext } from '../../../../../components/CoreServices/CoreServices';
 
+jest.mock('../../../../../services', () => ({
+  ...jest.requireActual('../../../../../services'),
+
+  getDataSourcePlugin: () => {
+    return false;
+  }
+}));
+
 const renderWithRouter = (
   initialAdState: Detectors = initialDetectorsState
 ) => ({

public/pages/utils/helpers.ts

@@ -20,7 +20,7 @@ import { DETECTORS_QUERY_PARAMS, SORT_DIRECTION } from '../../../server/utils/co
 import { ALL_INDICES, ALL_DETECTOR_STATES, MAX_DETECTORS, DEFAULT_QUERY_PARAMS } from './constants';
 import { DETECTOR_STATE } from '../../../server/utils/constants';
 import { timeFormatter } from '@elastic/charts';
-import { getDataSourcePlugin } from '../../services';
+import { getDataSourceEnabled, getDataSourcePlugin } from '../../services';
 
 export function sanitizeSearchText(searchValue: string): string {
   if (!searchValue || searchValue == '*') {
@@ -155,7 +155,7 @@ export const constructHrefWithDataSourceId = (
   dataSourceId: string,
   withHash: Boolean
 ): string => {
-  const dataSourceEnabled = getDataSourcePlugin()?.dataSourceEnabled || false;
+  const dataSourceEnabled = getDataSourceEnabled().enabled;
   const url = new URLSearchParams();
 
   // Set up base parameters for '/detectors'

public/plugin.ts

@@ -37,7 +37,7 @@ import {
   setQueryService,
   setSavedObjectsClient,
   setDataSourceManagementPlugin,
-  setDataSourcePlugin,
+  setDataSourceEnabled,
 } from './services';
 import { AnomalyDetectionOpenSearchDashboardsPluginStart } from 'public';
 import {
@@ -100,7 +100,9 @@ export class AnomalyDetectionOpenSearchDashboardsPlugin
 
     setDataSourceManagementPlugin(plugins.dataSourceManagement);
 
-    setDataSourcePlugin(plugins.dataSource);
+    const enabled = !!plugins.dataSource;
+
+    setDataSourceEnabled({ enabled });
 
     // Create context menu actions
     const actions = getActions();

public/services.ts

@@ -17,6 +17,10 @@ import { UiActionsStart } from '../../../src/plugins/ui_actions/public';
 import { SavedAugmentVisLoader } from '../../../src/plugins/vis_augmenter/public';
 import { DataSourcePluginSetup } from '../../../src/plugins/data_source/public';
 
+export interface DataSourceEnabled {
+  enabled: boolean;
+}
+
 export const [getSavedFeatureAnywhereLoader, setSavedFeatureAnywhereLoader] =
   createGetterSetter<SavedAugmentVisLoader>('savedFeatureAnywhereLoader');
 
@@ -47,8 +51,8 @@ export const [getSavedObjectsClient, setSavedObjectsClient] =
 export const [getDataSourceManagementPlugin, setDataSourceManagementPlugin] =
   createGetterSetter<DataSourceManagementPluginSetup>('DataSourceManagement');
 
-export const [getDataSourcePlugin, setDataSourcePlugin] =
-  createGetterSetter<DataSourcePluginSetup>('DataSource');
+export const [getDataSourceEnabled, setDataSourceEnabled] =
+  createGetterSetter<DataSourceEnabled>('DataSourceEnabled');
 
 // This is primarily used for mocking this module and each of its fns in tests.
 export default {