fix unit test

Signed-off-by: Hailong Cui <ihailong@amazon.com>
ruanyl · Oct 18, 2023 · dcb3a60 · dcb3a60
1 parent cbcb5d7
commit dcb3a60
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/src/plugins/workspace/public/plugin.test.ts b/src/plugins/workspace/public/plugin.test.ts
@@ -11,6 +11,7 @@ import { applicationServiceMock, chromeServiceMock, coreMock } from '../../../co
 import { WorkspacePlugin } from './plugin';
 import { WORKSPACE_FATAL_ERROR_APP_ID, WORKSPACE_OVERVIEW_APP_ID } from '../common/constants';
 import { Observable, Subscriber } from 'rxjs';
+import { savedObjectsManagementPluginMock } from 'src/plugins/saved_objects_management/public/mocks';
 
 describe('Workspace plugin', () => {
   beforeEach(() => {
@@ -19,11 +20,15 @@ describe('Workspace plugin', () => {
   });
   it('#setup', async () => {
     const setupMock = coreMock.createSetup();
+    const savedObjectManagementSetupMock = savedObjectsManagementPluginMock.createSetupContract();
     const workspacePlugin = new WorkspacePlugin();
-    await workspacePlugin.setup(setupMock);
+    await workspacePlugin.setup(setupMock, {
+      savedObjectsManagement: savedObjectManagementSetupMock,
+    });
     expect(setupMock.application.register).toBeCalledTimes(1);
     expect(WorkspaceClientMock).toBeCalledTimes(1);
     expect(workspaceClientMock.enterWorkspace).toBeCalledTimes(0);
+    expect(savedObjectManagementSetupMock.columns.register).toBeCalledTimes(1);
   });
 
   it('#setup when workspace id is in url and enterWorkspace return error', async () => {
@@ -55,7 +60,9 @@ describe('Workspace plugin', () => {
     });
 
     const workspacePlugin = new WorkspacePlugin();
-    await workspacePlugin.setup(setupMock);
+    await workspacePlugin.setup(setupMock, {
+      savedObjectsManagement: savedObjectsManagementPluginMock.createSetupContract(),
+    });
     expect(setupMock.application.register).toBeCalledTimes(1);
     expect(WorkspaceClientMock).toBeCalledTimes(1);
     expect(workspaceClientMock.enterWorkspace).toBeCalledWith('workspaceId');
@@ -109,7 +116,9 @@ describe('Workspace plugin', () => {
     });
 
     const workspacePlugin = new WorkspacePlugin();
-    await workspacePlugin.setup(setupMock);
+    await workspacePlugin.setup(setupMock, {
+      savedObjectsManagement: savedObjectsManagementPluginMock.createSetupContract(),
+    });
     currentAppIdSubscriber?.next(WORKSPACE_FATAL_ERROR_APP_ID);
     expect(applicationStartMock.navigateToApp).toBeCalledWith(WORKSPACE_OVERVIEW_APP_ID);
     windowSpy.mockRestore();