From 4b374c7de708ea08461d55cc6836de5a61b1cb66 Mon Sep 17 00:00:00 2001
From: Harsh Vador <58542468+harsh-vador@users.noreply.github.com>
Date: Wed, 28 Feb 2024 13:11:13 +0530
Subject: [PATCH] navbar test case (#15352)

---
 .../ui/src/components/NavBar/NavBar.test.tsx  | 215 ++++++++++++++++++
 .../ui/src/components/NavBar/NavBar.tsx       |   1 +
 2 files changed, 216 insertions(+)
 create mode 100644 openmetadata-ui/src/main/resources/ui/src/components/NavBar/NavBar.test.tsx

diff --git a/openmetadata-ui/src/main/resources/ui/src/components/NavBar/NavBar.test.tsx b/openmetadata-ui/src/main/resources/ui/src/components/NavBar/NavBar.test.tsx
new file mode 100644
index 00000000000..105093056b3
--- /dev/null
+++ b/openmetadata-ui/src/main/resources/ui/src/components/NavBar/NavBar.test.tsx
@@ -0,0 +1,215 @@
+/*
+ *  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 { act, fireEvent, render, screen } from '@testing-library/react';
+import React from 'react';
+import { mockUserData } from '../Settings/Users/mocks/User.mocks';
+import NavBar from './NavBar';
+
+const mockHandleSearchBoxOpen = jest.fn();
+const mockFeatureModal = jest.fn();
+const mockHandleSearchChange = jest.fn();
+const mockHandleOnClick = jest.fn();
+const mockHandleKeyDown = jest.fn();
+const mockHandleClear = jest.fn();
+const mockProps = {
+  supportDropdown: [],
+  searchValue: 'searchValue',
+  isTourRoute: false,
+  isFeatureModalOpen: true,
+  pathname: '',
+  isSearchBoxOpen: false,
+  handleSearchBoxOpen: mockHandleSearchBoxOpen,
+  handleFeatureModal: mockFeatureModal,
+  handleSearchChange: mockHandleSearchChange,
+  handleOnClick: mockHandleOnClick,
+  handleClear: mockHandleClear,
+  handleKeyDown: mockHandleKeyDown,
+};
+jest.mock('../../context/GlobalSearchProvider/GlobalSearchProvider', () => ({
+  useGlobalSearchProvider: jest.fn().mockImplementation(() => ({
+    searchCriteria: '',
+    updateSearchCriteria: jest.fn(),
+  })),
+}));
+jest.mock('../../context/WebSocketProvider/WebSocketProvider', () => ({
+  useWebSocketConnector: jest.fn().mockImplementation(() => ({
+    socket: {
+      on: jest.fn(),
+      off: jest.fn(),
+    },
+  })),
+}));
+jest.mock('../../utils/BrowserNotificationUtils', () => ({
+  hasNotificationPermission: jest.fn(),
+  shouldRequestPermission: jest.fn(),
+}));
+jest.mock('../../utils/CommonUtils', () => ({
+  refreshPage: jest.fn(),
+  getEntityDetailLink: jest.fn(),
+}));
+jest.mock('../../utils/FeedUtils', () => ({
+  getEntityFQN: jest.fn().mockReturnValue('entityFQN'),
+  getEntityType: jest.fn().mockReturnValue('entityType'),
+  prepareFeedLink: jest.fn().mockReturnValue('entity-link'),
+}));
+jest.mock('../Domain/DomainProvider/DomainProvider', () => ({
+  useDomainProvider: jest.fn().mockImplementation(() => ({
+    domainOptions: jest.fn().mockReturnValue('domainOptions'),
+    activeDomain: jest.fn().mockReturnValue('activeDomain'),
+    updateActiveDomain: jest.fn(),
+  })),
+}));
+jest.mock('../Modals/WhatsNewModal/WhatsNewModal', () => {
+  return jest.fn().mockImplementation(
+    ({ onCancel, header }) =>
+      mockProps.isFeatureModalOpen && (
+        <div data-testid="whats-new-dialog">
+          WhatsNewModal
+          <p>{header}</p>
+          <button data-testid="cancel-button" onClick={onCancel}>
+            onCancel
+          </button>
+        </div>
+      )
+  );
+});
+jest.mock('../NotificationBox/NotificationBox.component', () => {
+  return jest.fn().mockImplementation(({ onTabChange }) => (
+    <div data-testid="tab-change" onClick={onTabChange}>
+      tab change
+    </div>
+  ));
+});
+
+jest.mock(
+  '../Settings/Users/UserProfileIcon/UserProfileIcon.component',
+  () => ({
+    UserProfileIcon: jest
+      .fn()
+      .mockReturnValue(
+        <div data-testid="user-profile-icon">UserProfileIcon</div>
+      ),
+  })
+);
+jest.mock('../Auth/AuthProviders/AuthProvider', () => ({
+  useAuthContext: jest.fn(() => ({
+    currentUser: mockUserData,
+  })),
+}));
+jest.mock('react-router-dom', () => ({
+  useLocation: jest
+    .fn()
+    .mockReturnValue({ search: 'search', pathname: '/my-data' }),
+  useHistory: jest.fn(),
+}));
+
+jest.mock('../common/CmdKIcon/CmdKIcon.component', () => {
+  return jest.fn().mockReturnValue(<div data-testid="cmd">CmdKIcon</div>);
+});
+jest.mock('../AppBar/SearchOptions', () => {
+  return jest.fn().mockReturnValue(<div data-testid="cmd">SearchOptions</div>);
+});
+jest.mock('../AppBar/Suggestions', () => {
+  return jest.fn().mockReturnValue(<div data-testid="cmd">Suggestions</div>);
+});
+jest.mock('antd', () => ({
+  ...jest.requireActual('antd'),
+
+  Dropdown: jest.fn().mockImplementation(({ dropdownRender }) => {
+    return (
+      <div data-testid="dropdownRender">
+        <div>{dropdownRender}</div>
+      </div>
+    );
+  }),
+}));
+
+describe('Test NavBar Component', () => {
+  it('Should render NavBar component', async () => {
+    render(<NavBar {...mockProps} />);
+
+    expect(
+      await screen.findByTestId('navbar-search-container')
+    ).toBeInTheDocument();
+    expect(
+      await screen.findByTestId('global-search-selector')
+    ).toBeInTheDocument();
+    expect(await screen.findByTestId('searchBox')).toBeInTheDocument();
+    expect(await screen.findByTestId('cmd')).toBeInTheDocument();
+    expect(await screen.findByTestId('user-profile-icon')).toBeInTheDocument();
+    expect(
+      await screen.findByTestId('whats-new-alert-card')
+    ).toBeInTheDocument();
+    expect(
+      await screen.findByTestId('whats-new-alert-header')
+    ).toBeInTheDocument();
+    expect(
+      await screen.findByTestId('close-whats-new-alert')
+    ).toBeInTheDocument();
+    expect(
+      await screen.findByText('label.whats-new-version')
+    ).toBeInTheDocument();
+  });
+
+  it('should handle search box open', () => {
+    render(<NavBar {...mockProps} />);
+    const searchBox = screen.getByTestId('searchBox');
+    fireEvent.click(searchBox);
+
+    expect(mockHandleSearchBoxOpen).toHaveBeenCalled();
+  });
+
+  it('should handle search change', () => {
+    render(<NavBar {...mockProps} />);
+    const searchBox = screen.getByTestId('searchBox');
+    fireEvent.change(searchBox, { target: { value: 'test' } });
+
+    expect(mockHandleSearchChange).toHaveBeenCalledWith('test');
+  });
+
+  it('should handle key down', () => {
+    render(<NavBar {...mockProps} />);
+    const searchBox = screen.getByTestId('searchBox');
+    fireEvent.keyDown(searchBox, { key: 'Enter', code: 'Enter' });
+
+    expect(mockHandleKeyDown).toHaveBeenCalled();
+  });
+
+  it('should render cancel icon', () => {
+    render(<NavBar {...mockProps} />);
+    const searchBox = screen.getByTestId('searchBox');
+    fireEvent.keyDown(searchBox, { key: 'Enter', code: 'Enter' });
+
+    expect(mockHandleKeyDown).toHaveBeenCalled();
+  });
+
+  it('should call function on icon search', async () => {
+    render(<NavBar {...mockProps} searchValue="" />);
+    const searchBox = await screen.findByTestId('search-icon');
+    await act(async () => {
+      fireEvent.click(searchBox);
+    });
+
+    expect(mockHandleOnClick).toHaveBeenCalled();
+  });
+
+  it('should call onCancel on modal close', async () => {
+    render(<NavBar {...mockProps} searchValue="" />);
+    const cancelButton = await screen.findAllByTestId('cancel-button');
+    await act(async () => {
+      fireEvent.click(cancelButton[0]);
+    });
+
+    expect(mockFeatureModal).toHaveBeenCalled();
+  });
+});
diff --git a/openmetadata-ui/src/main/resources/ui/src/components/NavBar/NavBar.tsx b/openmetadata-ui/src/main/resources/ui/src/components/NavBar/NavBar.tsx
index 9edeeadabd7..66eb0bdce8d 100644
--- a/openmetadata-ui/src/main/resources/ui/src/components/NavBar/NavBar.tsx
+++ b/openmetadata-ui/src/main/resources/ui/src/components/NavBar/NavBar.tsx
@@ -401,6 +401,7 @@ const NavBar = ({
                           'text-primary': !isSearchBlur,
                         })}
                         component={IconSearch}
+                        data-testid="search-icon"
                         style={{ fontSize: '16px' }}
                         onClick={(e) => {
                           e.preventDefault();