From 477da55205933d68e5c8f9141d74cd7b5ae28cad Mon Sep 17 00:00:00 2001
From: SuZhou-Joe <suzhou@amazon.com>
Date: Thu, 12 Oct 2023 10:35:08 +0800
Subject: [PATCH] feat: add unit test for mountWrapper (#223)

Signed-off-by: SuZhou-Joe <suzhou@amazon.com>
---
 .../public/plugin.test.ts                     | 63 +++++++++++++++++--
 1 file changed, 58 insertions(+), 5 deletions(-)

diff --git a/src/plugins/saved_objects_management/public/plugin.test.ts b/src/plugins/saved_objects_management/public/plugin.test.ts
index c8e762f73dcc..149cee7c5c86 100644
--- a/src/plugins/saved_objects_management/public/plugin.test.ts
+++ b/src/plugins/saved_objects_management/public/plugin.test.ts
@@ -28,12 +28,23 @@
  * under the License.
  */
 
+const mountManagementSectionMock = jest.fn();
+jest.doMock('./management_section', () => ({
+  mountManagementSection: mountManagementSectionMock,
+}));
+import { waitFor } from '@testing-library/dom';
 import { coreMock } from '../../../core/public/mocks';
 import { homePluginMock } from '../../home/public/mocks';
 import { managementPluginMock } from '../../management/public/mocks';
 import { dataPluginMock } from '../../data/public/mocks';
 import { uiActionsPluginMock } from '../../ui_actions/public/mocks';
 import { SavedObjectsManagementPlugin } from './plugin';
+import {
+  MANAGE_LIBRARY_TITLE_WORDINGS,
+  SAVED_QUERIES_WORDINGS,
+  SAVED_SEARCHES_WORDINGS,
+} from './constants';
+import { DEFAULT_APP_CATEGORIES } from '../../../core/public';
 
 describe('SavedObjectsManagementPlugin', () => {
   let plugin: SavedObjectsManagementPlugin;
@@ -50,12 +61,22 @@ describe('SavedObjectsManagementPlugin', () => {
       const homeSetup = homePluginMock.createSetupContract();
       const managementSetup = managementPluginMock.createSetupContract();
       const uiActionsSetup = uiActionsPluginMock.createSetupContract();
+      const registerMock = jest.fn((params) => params.mount({} as any, {} as any));
 
-      await plugin.setup(coreSetup, {
-        home: homeSetup,
-        management: managementSetup,
-        uiActions: uiActionsSetup,
-      });
+      await plugin.setup(
+        {
+          ...coreSetup,
+          application: {
+            ...coreSetup.application,
+            register: registerMock,
+          },
+        },
+        {
+          home: homeSetup,
+          management: managementSetup,
+          uiActions: uiActionsSetup,
+        }
+      );
 
       expect(homeSetup.featureCatalogue.register).toHaveBeenCalledTimes(1);
       expect(homeSetup.featureCatalogue.register).toHaveBeenCalledWith(
@@ -63,6 +84,38 @@ describe('SavedObjectsManagementPlugin', () => {
           id: 'saved_objects',
         })
       );
+      expect(registerMock).toBeCalledWith(
+        expect.objectContaining({
+          id: 'objects',
+          title: MANAGE_LIBRARY_TITLE_WORDINGS,
+          order: 10000,
+          category: DEFAULT_APP_CATEGORIES.opensearchDashboards,
+        })
+      );
+      expect(registerMock).toBeCalledWith(
+        expect.objectContaining({
+          id: 'objects_searches',
+          title: SAVED_SEARCHES_WORDINGS,
+          order: 8000,
+          category: DEFAULT_APP_CATEGORIES.opensearchDashboards,
+        })
+      );
+      expect(registerMock).toBeCalledWith(
+        expect.objectContaining({
+          id: 'objects_query',
+          title: SAVED_QUERIES_WORDINGS,
+          order: 8001,
+          category: DEFAULT_APP_CATEGORIES.opensearchDashboards,
+        })
+      );
+      waitFor(
+        () => {
+          expect(mountManagementSectionMock).toBeCalledTimes(3);
+        },
+        {
+          container: document.body,
+        }
+      );
     });
   });
 });