unit test for DataInsightHeader and DataModelPage components (#15076)

* added unit test for DataInsightHeader componnet

* added unit test for DataModelPage component

openmetadata-ui/src/main/resources/ui/src/pages/DataInsightPage/DataInsightHeader/DataInsightHeader.test.tsx

@@ -0,0 +1,100 @@
+/*
+ *  Copyright 2024 Collate.
+ *  Licensed under the Apache License, Version 2.0 (the "License");
+ *  you may not use this file except in compliance with the License.
+ *  You may obtain a copy of the License at
+ *  http://www.apache.org/licenses/LICENSE-2.0
+ *  Unless required by applicable law or agreed to in writing, software
+ *  distributed under the License is distributed on an "AS IS" BASIS,
+ *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *  See the License for the specific language governing permissions and
+ *  limitations under the License.
+ */
+import { render, screen } from '@testing-library/react';
+import userEvent from '@testing-library/user-event';
+import React from 'react';
+import { ROUTES } from '../../../constants/constants';
+import DataInsightHeader from './DataInsightHeader.component';
+
+const mockPush = jest.fn();
+
+jest.mock('react-router-dom', () => ({
+  useHistory: jest.fn().mockImplementation(() => ({
+    push: mockPush,
+  })),
+  useParams: jest.fn(() => ({ tab: 'tab' })),
+}));
+
+jest.mock('../../../components/DataInsightDetail/DataInsightSummary', () =>
+  jest.fn(() => <div>DataInsightSummary</div>)
+);
+
+jest.mock('../../../components/DataInsightDetail/KPIChart', () =>
+  jest.fn(() => <div>KPIChart</div>)
+);
+
+jest.mock('../../../components/DatePickerMenu/DatePickerMenu.component', () =>
+  jest.fn(() => <div>DatePickerMenu</div>)
+);
+
+jest.mock('../../../components/PermissionProvider/PermissionProvider', () => ({
+  usePermissionProvider: jest.fn(() => ({
+    permissions: {},
+  })),
+}));
+
+jest.mock('../../../components/SearchDropdown/SearchDropdown', () =>
+  jest.fn(() => <div>SearchDropdown</div>)
+);
+
+jest.mock('../../../utils/DataInsightUtils', () => ({
+  getOptionalDataInsightTabFlag: jest.fn(() => ({
+    showDataInsightSummary: true,
+    showKpiChart: true,
+  })),
+}));
+
+jest.mock('../../../utils/date-time/DateTimeUtils', () => ({
+  formatDate: jest.fn().mockReturnValue('formattedDate'),
+}));
+
+jest.mock('../../../utils/PermissionsUtils', () => ({
+  checkPermission: jest.fn().mockReturnValue(true),
+}));
+
+jest.mock('../DataInsightProvider', () => ({
+  useDataInsightProvider: jest
+    .fn()
+    .mockReturnValue({ chartFilter: {}, kpi: {} }),
+}));
+
+const mockProps = {
+  onScrollToChart: jest.fn(),
+};
+
+describe('DataInsightHeader component', () => {
+  it('should render all necessary elements', () => {
+    render(<DataInsightHeader {...mockProps} />);
+
+    expect(screen.getByText('label.data-insight-plural')).toBeInTheDocument();
+    expect(
+      screen.getByText('message.data-insight-subtitle')
+    ).toBeInTheDocument();
+
+    userEvent.click(
+      screen.getByRole('button', {
+        name: 'label.add-entity',
+      })
+    );
+
+    expect(mockPush).toHaveBeenCalledWith(ROUTES.ADD_KPI);
+
+    expect(screen.getAllByText('SearchDropdown')).toHaveLength(2);
+    expect(
+      screen.getByText('formattedDate - formattedDate')
+    ).toBeInTheDocument();
+    expect(screen.getByText('DatePickerMenu')).toBeInTheDocument();
+    expect(screen.getByText('DataInsightSummary')).toBeInTheDocument();
+    expect(screen.getByText('KPIChart')).toBeInTheDocument();
+  });
+});