hide live queries in 3.3 below

Signed-off-by: Kishore Kumaar Natarajan <[email protected]>

Author: Kishore Kumaar Natarajan

public/pages/WorkloadManagement/WLMMain/WLMMain.test.tsx

@@ -11,6 +11,7 @@ import { WorkloadManagementMain } from './WLMMain';
 import { CoreStart } from 'opensearch-dashboards/public';
 import userEvent from '@testing-library/user-event';
 import { DataSourceContext } from '../WorkloadManagement';
+import * as versionUtils from '../../../utils/version-utils';
 
 jest.mock('echarts-for-react', () => () => <div data-testid="MockedChart">Mocked Chart</div>);
 jest.mock('../../../components/PageHeader', () => ({
@@ -63,6 +64,10 @@ beforeEach(() => {
   jest.clearAllMocks();
   capturedOptions.length = 0;
 
+  // Mock version utils
+  jest.spyOn(versionUtils, 'getVersionOnce').mockResolvedValue('3.3.0');
+  jest.spyOn(versionUtils, 'isVersion33OrHigher').mockReturnValue(true);
+
   // Restore the data source menu mock after reset
   mockDataSourceManagement.ui.getDataSourceMenu.mockReturnValue(MockDataSourceMenu);
 
@@ -76,6 +81,11 @@ beforeEach(() => {
         ],
       });
     }
+    if (url === '/api/live_queries') {
+      return Promise.resolve({
+        response: { live_queries: [] }
+      });
+    }
     if (url === '/api/_wlm/stats') {
       return Promise.resolve({
         node1: {
@@ -116,6 +126,12 @@ beforeEach(() => {
         },
       });
     }
+    if (url === '/api/_wlm/thresholds') {
+      return Promise.resolve({
+        cpuRejectionThreshold: 0.8,
+        memoryRejectionThreshold: 0.8
+      });
+    }
   });
 });
 
@@ -502,4 +518,26 @@ describe('WorkloadManagementMain', () => {
       expect(screen.getByPlaceholderText(/search workload groups/i)).toBeInTheDocument();
     });
   });
+
+  it('does not call Query Insights API when version < 3.3', async () => {
+    jest.spyOn(versionUtils, 'isVersion33OrHigher').mockReturnValue(false);
+    
+    renderComponent();
+    
+    await waitFor(() => {
+      expect(mockCore.http.get).not.toHaveBeenCalledWith('/api/live_queries', expect.any(Object));
+    });
+  });
+
+  it('calls Query Insights API when version >= 3.3', async () => {
+    jest.spyOn(versionUtils, 'isVersion33OrHigher').mockReturnValue(true);
+    
+    renderComponent();
+    
+    await waitFor(() => {
+      expect(mockCore.http.get).toHaveBeenCalledWith('/api/live_queries', expect.objectContaining({
+        query: { dataSourceId: 'default' }
+      }));
+    });
+  });
 });

public/pages/WorkloadManagement/WLMMain/WLMMain.tsx

@@ -28,6 +28,7 @@ import { WLM_CREATE } from '../WorkloadManagement';
 import { DataSourceContext } from '../WorkloadManagement';
 import { WLMDataSourceMenu } from '../../../components/DataSourcePicker';
 import { getDataSourceEnabledUrl } from '../../../utils/datasource-utils';
+import { getVersionOnce, isVersion33OrHigher } from '../../../utils/version-utils';
 
 export const WLM = '/workloadManagement';
 
@@ -160,9 +161,14 @@ export const WorkloadManagementMain = ({
     }
   };
 
-  // === API Calls ===
   const checkQueryInsightsAvailability = async () => {
     try {
+      const version = await getVersionOnce(dataSource?.id || '');
+      if (!isVersion33OrHigher(version)) {
+        setIsQueryInsightsAvailable(false);
+        return;
+      }
+
       const res = await core.http.get('/api/live_queries', {
         query: { dataSourceId: dataSource.id },
       });