disable in mds use case

Signed-off-by: Jackie Han <[email protected]>

Author: jackiehanyang

public/pages/DetectorResults/__tests__/AnomalyResultsTable.test.tsx

@@ -13,7 +13,7 @@ import React from 'react';
 import { render, screen, fireEvent, waitFor } from '@testing-library/react';
 import '@testing-library/jest-dom';
 import { AnomalyResultsTable } from '../containers/AnomalyResultsTable';
-import { getSavedObjectsClient, getNotifications } from '../../../services';
+import { getSavedObjectsClient, getNotifications, getDataSourceEnabled } from '../../../services';
 import { CoreServicesContext } from '../../../components/CoreServices/CoreServices';
 
 const mockWindowOpen = jest.fn();
@@ -25,6 +25,7 @@ Object.defineProperty(window, 'open', {
 jest.mock('../../../services', () => ({
   getSavedObjectsClient: jest.fn(),
   getNotifications: jest.fn(),
+  getDataSourceEnabled: jest.fn(),
 }));
 
 const mockCoreServices = {
@@ -75,6 +76,8 @@ describe('AnomalyResultsTable', () => {
         addDanger: jest.fn(),
       },
     });
+
+    (getDataSourceEnabled as jest.Mock).mockReturnValue({ enabled: false });
   });
 
   it('shows no anomalies message when there are no anomalies', () => {
@@ -168,4 +171,30 @@ describe('AnomalyResultsTable', () => {
       });
     }
   });
+
+  describe('mds feature flag', () => {
+    it('shows Actions column when mds is disabled', () => {
+      (getDataSourceEnabled as jest.Mock).mockReturnValue({ enabled: false });
+      
+      renderWithContext(<AnomalyResultsTable {...defaultProps} />);
+      
+      const actionsColumn = screen.getByText('Actions');
+      expect(actionsColumn).toBeInTheDocument();
+      
+      const discoverButton = screen.getByTestId('discoverIcon');
+      expect(discoverButton).toBeInTheDocument();
+    });
+
+    it('hides Actions column when mds is enabled', () => {
+      (getDataSourceEnabled as jest.Mock).mockReturnValue({ enabled: true });
+      
+      renderWithContext(<AnomalyResultsTable {...defaultProps} />);
+      
+      const actionsColumn = screen.queryByText('Actions');
+      expect(actionsColumn).not.toBeInTheDocument();
+      
+      const discoverButton = screen.queryByTestId('discoverIcon');
+      expect(discoverButton).not.toBeInTheDocument();
+    });
+  });
 }); 

public/pages/DetectorResults/containers/AnomalyResultsTable.tsx

@@ -28,7 +28,7 @@ import { AnomalyData } from '../../../models/interfaces';
 import { getTitleWithCount } from '../../../utils/utils';
 import { convertToCategoryFieldAndEntityString } from '../../utils/anomalyResultUtils';
 import { HeatmapCell } from '../../AnomalyCharts/containers/AnomalyHeatmapChart';
-import { getSavedObjectsClient, getNotifications } from '../../../services';
+import { getSavedObjectsClient, getNotifications, getDataSourceEnabled } from '../../../services';
 
 //@ts-ignore
 const EuiBasicTable = EuiBasicTableComponent as any;
@@ -130,22 +130,31 @@ export function AnomalyResultsTable(props: AnomalyResultsTableProps) {
   };
 
   const getCustomColumns = () => {
+    const dataSourceEnabled = getDataSourceEnabled().enabled;
     const columns = [...staticColumn] as any[];
-    const actionsColumnIndex = columns.findIndex((column: any) => column.field === 'actions');
 
-    if (actionsColumnIndex !== -1) {
-      const actionsColumn = { ...columns[actionsColumnIndex] } as any;
+    if (!dataSourceEnabled) {
+      const actionsColumnIndex = columns.findIndex((column: any) => column.field === 'actions');
 
-      if (actionsColumn.actions && Array.isArray(actionsColumn.actions)) {
-        actionsColumn.actions = [
-          {
-            ...actionsColumn.actions[0],
-            onClick: (item: any) => handleOpenDiscover(item.startTime, item.endTime, item),
-          },
-        ];
+      if (actionsColumnIndex !== -1) {
+        const actionsColumn = { ...columns[actionsColumnIndex] } as any;
+        
+        if (actionsColumn.actions && Array.isArray(actionsColumn.actions)) {
+          actionsColumn.actions = [
+            {
+              ...actionsColumn.actions[0],
+              onClick: (item: any) => handleOpenDiscover(item.startTime, item.endTime, item),
+            },
+          ];
+        }
+        
+        columns[actionsColumnIndex] = actionsColumn;
+      }
+    } else {
+      const actionsColumnIndex = columns.findIndex((column: any) => column.field === 'actions');
+      if (actionsColumnIndex !== -1) {
+        columns.splice(actionsColumnIndex, 1);
       }
-      
-      columns[actionsColumnIndex] = actionsColumn;
     }
 
     return columns;