feat: add unit test

Signed-off-by: SuZhou-Joe <[email protected]>
ruanyl · Apr 10, 2024 · d5b8676 · d5b8676
1 parent 57ddd07
commit d5b8676
Showing 1 changed file with 25 additions and 0 deletions.
diff --git a/src/plugins/workspace/server/saved_objects/workspace_id_consumer_wrapper.test.ts b/src/plugins/workspace/server/saved_objects/workspace_id_consumer_wrapper.test.ts
@@ -8,6 +8,7 @@ import { SavedObject } from '../../../../core/public';
 import { httpServerMock, savedObjectsClientMock, coreMock } from '../../../../core/server/mocks';
 import { WorkspaceIdConsumerWrapper } from './workspace_id_consumer_wrapper';
 import { DATA_SOURCE_SAVED_OBJECT_TYPE } from '../../../../plugins/data_source/common';
+import { UI_SETTINGS_SAVED_OBJECTS_TYPE } from '../../../../core/server';
 
 describe('WorkspaceIdConsumerWrapper', () => {
   const requestHandlerContext = coreMock.createRequestHandlerContext();
@@ -86,6 +87,30 @@ describe('WorkspaceIdConsumerWrapper', () => {
         }
       );
     });
+
+    it(`Should throw error when trying to create unallowed type within a workspace`, async () => {
+      expect(() =>
+        wrapperClient.bulkCreate([
+          getSavedObject({
+            type: DATA_SOURCE_SAVED_OBJECT_TYPE,
+            id: 'foo',
+          }),
+        ])
+      ).toThrow('type: data-source is not allowed to create within a workspace.');
+
+      expect(() =>
+        wrapperClient.bulkCreate([
+          getSavedObject({
+            type: DATA_SOURCE_SAVED_OBJECT_TYPE,
+            id: 'foo',
+          }),
+          getSavedObject({
+            type: UI_SETTINGS_SAVED_OBJECTS_TYPE,
+            id: 'bar',
+          }),
+        ])
+      ).toThrow('type: data-source, type: config are not allowed to create within a workspace.');
+    });
   });
 
   describe('checkConflict', () => {