updated uts for WorkloadManagement

Signed-off-by: Lingxi Chen <[email protected]>

Author: Lingxi Chen

public/pages/WorkloadManagement/WorkloadManagement.test.tsx

@@ -3,21 +3,14 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 
-/**
- * @jest-environment jsdom
- */
-
 import React from 'react';
 import { render, screen } from '@testing-library/react';
 import { MemoryRouter, Route } from 'react-router-dom';
 import '@testing-library/jest-dom/extend-expect';
-import { WorkloadManagement, WLM_MAIN, WLM_DETAILS } from './WorkloadManagement';
+import { WorkloadManagement, WLM_MAIN, WLM_DETAILS, WLM_CREATE } from './WorkloadManagement';
 import { CoreStart } from 'opensearch-dashboards/public';
 import { QueryInsightsDashboardsPluginStartDependencies } from '../../types';
-
-// Mock the subcomponents
-jest.mock('./WLMMain/WLMMain', () => () => <div>Mocked WLM Main</div>);
-jest.mock('./WLMDetails/WLMDetails', () => () => <div>Mocked WLM Details</div>);
+import { DataSourceContext } from './WorkloadManagement';
 
 const mockCore = ({
   http: {
@@ -27,32 +20,69 @@ const mockCore = ({
   uiSettings: {
     get: jest.fn().mockReturnValue(false),
   },
+  chrome: {
+    setBreadcrumbs: jest.fn(),
+  },
+  notifications: {
+    toasts: {
+      addSuccess: jest.fn(),
+      addDanger: jest.fn(),
+    },
+  },
 } as unknown) as CoreStart;
 
-const mockDepsStart: QueryInsightsDashboardsPluginStartDependencies = {};
+const mockDepsStart = {} as QueryInsightsDashboardsPluginStartDependencies;
+
+const mockDataSource = {
+  id: 'default',
+  name: 'default',
+} as any;
+
+jest.mock('../../components/PageHeader', () => ({
+  PageHeader: () => <div data-testid="mock-page-header">Mocked PageHeader</div>,
+}));
 
-const renderWithRoute = (initialRoute: string) =>
-  render(
+const renderWithRoute = (initialRoute: string) => {
+  return render(
     <MemoryRouter initialEntries={[initialRoute]}>
       <Route path="*">
-        <WorkloadManagement core={mockCore} depsStart={mockDepsStart} />
+        <DataSourceContext.Provider value={{ dataSource: mockDataSource, setDataSource: jest.fn() }}>
+          <WorkloadManagement
+            core={mockCore}
+            depsStart={mockDepsStart}
+            params={{} as any}
+            dataSourceManagement={{} as any}
+          />
+        </DataSourceContext.Provider>
       </Route>
     </MemoryRouter>
   );
+};
 
 describe('WorkloadManagement Routing', () => {
+  beforeEach(() => {
+    jest.clearAllMocks();
+  });
+
   it('renders WLMMain component at WLM_MAIN route', () => {
     renderWithRoute(WLM_MAIN);
-    expect(screen.getByText('Mocked WLM Main')).toBeInTheDocument();
+    expect(screen.getByRole('heading', { name: /Workload groups/i })).toBeInTheDocument();
   });
 
-  it('renders WLMDetails component at WLM_DETAILS route', () => {
-    renderWithRoute(WLM_DETAILS);
-    expect(screen.getByText('Mocked WLM Details')).toBeInTheDocument();
+  it('renders WLMDetails component at WLM_DETAILS route', async () => {
+    renderWithRoute(`${WLM_DETAILS}?name=DEFAULT_WORKLOAD_GROUP`);
+    expect(await screen.findByRole('heading', { name: /DEFAULT_WORKLOAD_GROUP/i })).toBeInTheDocument();
   });
 
   it('redirects to WLM_MAIN for unknown routes', () => {
-    renderWithRoute('/some/invalid/route');
-    expect(screen.getByText('Mocked WLM Main')).toBeInTheDocument();
+    renderWithRoute('/invalid/route');
+    expect(screen.getByRole('heading', { name: /Workload groups/i })).toBeInTheDocument();
+  });
+
+  it('renders WLMCreate component at WLM_CREATE route', () => {
+    renderWithRoute(WLM_CREATE);
+    expect(screen.getByRole('heading', { name: /Create workload group/i })).toBeInTheDocument();
   });
+
 });
+