Update tests

elastic · Sep 4, 2021 · 7b85d04 · 7b85d04
1 parent 872facd
commit 7b85d04
Showing 1 changed file with 76 additions and 37 deletions.
diff --git a/src/plugins/data/public/query/saved_query/saved_query_service.test.ts b/src/plugins/data/public/query/saved_query/saved_query_service.test.ts
@@ -54,7 +54,7 @@ const mockSavedObjectsClient = {
   create: jest.fn(),
   error: jest.fn(),
   find: jest.fn(),
-  get: jest.fn(),
+  resolve: jest.fn(),
   delete: jest.fn(),
 };
 
@@ -74,7 +74,7 @@ describe('saved query service', () => {
   afterEach(() => {
     mockSavedObjectsClient.create.mockReset();
     mockSavedObjectsClient.find.mockReset();
-    mockSavedObjectsClient.get.mockReset();
+    mockSavedObjectsClient.resolve.mockReset();
     mockSavedObjectsClient.delete.mockReset();
   });
 
@@ -267,85 +267,124 @@ describe('saved query service', () => {
 
   describe('getSavedQuery', function () {
     it('should retrieve a saved query by id', async () => {
-      mockSavedObjectsClient.get.mockReturnValue({ id: 'foo', attributes: savedQueryAttributes });
+      mockSavedObjectsClient.resolve.mockReturnValue({
+        saved_object: {
+          id: 'foo',
+          attributes: savedQueryAttributes,
+        },
+        outcome: 'exactMatch',
+      });
 
       const response = await getSavedQuery('foo');
       expect(response).toEqual({ id: 'foo', attributes: savedQueryAttributes });
     });
     it('should only return saved queries', async () => {
-      mockSavedObjectsClient.get.mockReturnValue({ id: 'foo', attributes: savedQueryAttributes });
+      mockSavedObjectsClient.resolve.mockReturnValue({
+        saved_object: {
+          id: 'foo',
+          attributes: savedQueryAttributes,
+        },
+        outcome: 'exactMatch',
+      });
 
       await getSavedQuery('foo');
-      expect(mockSavedObjectsClient.get).toHaveBeenCalledWith('query', 'foo');
+      expect(mockSavedObjectsClient.resolve).toHaveBeenCalledWith('query', 'foo');
     });
 
     it('should parse a json query', async () => {
-      mockSavedObjectsClient.get.mockReturnValue({
-        id: 'food',
-        attributes: {
-          title: 'food',
-          description: 'bar',
-          query: {
-            language: 'kuery',
-            query: '{"x": "y"}',
+      mockSavedObjectsClient.resolve.mockReturnValue({
+        saved_object: {
+          id: 'food',
+          attributes: {
+            title: 'food',
+            description: 'bar',
+            query: {
+              language: 'kuery',
+              query: '{"x": "y"}',
+            },
           },
         },
+        outcome: 'exactMatch',
       });
 
       const response = await getSavedQuery('food');
       expect(response.attributes.query.query).toEqual({ x: 'y' });
     });
 
     it('should handle null string', async () => {
-      mockSavedObjectsClient.get.mockReturnValue({
-        id: 'food',
-        attributes: {
-          title: 'food',
-          description: 'bar',
-          query: {
-            language: 'kuery',
-            query: 'null',
+      mockSavedObjectsClient.resolve.mockReturnValue({
+        saved_object: {
+          id: 'food',
+          attributes: {
+            title: 'food',
+            description: 'bar',
+            query: {
+              language: 'kuery',
+              query: 'null',
+            },
           },
         },
+        outcome: 'exactMatch',
       });
 
       const response = await getSavedQuery('food');
       expect(response.attributes.query.query).toEqual('null');
     });
 
     it('should handle null quoted string', async () => {
-      mockSavedObjectsClient.get.mockReturnValue({
-        id: 'food',
-        attributes: {
-          title: 'food',
-          description: 'bar',
-          query: {
-            language: 'kuery',
-            query: '"null"',
+      mockSavedObjectsClient.resolve.mockReturnValue({
+        saved_object: {
+          id: 'food',
+          attributes: {
+            title: 'food',
+            description: 'bar',
+            query: {
+              language: 'kuery',
+              query: '"null"',
+            },
           },
         },
+        outcome: 'exactMatch',
       });
 
       const response = await getSavedQuery('food');
       expect(response.attributes.query.query).toEqual('"null"');
     });
 
     it('should not lose quotes', async () => {
-      mockSavedObjectsClient.get.mockReturnValue({
-        id: 'food',
-        attributes: {
-          title: 'food',
-          description: 'bar',
-          query: {
-            language: 'kuery',
-            query: '"Bob"',
+      mockSavedObjectsClient.resolve.mockReturnValue({
+        saved_object: {
+          id: 'food',
+          attributes: {
+            title: 'food',
+            description: 'bar',
+            query: {
+              language: 'kuery',
+              query: '"Bob"',
+            },
           },
         },
+        outcome: 'exactMatch',
       });
 
       const response = await getSavedQuery('food');
       expect(response.attributes.query.query).toEqual('"Bob"');
     });
+
+    it('should throw if conflict', async () => {
+      mockSavedObjectsClient.resolve.mockReturnValue({
+        saved_object: {
+          id: 'foo',
+          attributes: savedQueryAttributes,
+        },
+        outcome: 'conflict',
+      });
+
+      const result = getSavedQuery('food');
+      expect(result).rejects.toMatchInlineSnapshot(
+        `[Error: Multiple saved queries found with ID: food]`
+      );
+    });
   });
 
   describe('deleteSavedQuery', function () {