Fix false positives in query signals routes tests

* Refactors mock expectations so that they're actually evaluated; they can't go within a mockImplementation call as it's evaluated in the wrong scope. * Fixes duplicated test to use the proper assertions
elastic · Feb 13, 2020 · 56fc3f5 · 56fc3f5
1 parent 668702b
commit 56fc3f5
Showing 1 changed file with 24 additions and 19 deletions.
diff --git a/...egacy/plugins/siem/server/lib/detection_engine/routes/signals/query_signals_route.test.ts b/...egacy/plugins/siem/server/lib/detection_engine/routes/signals/query_signals_route.test.ts
@@ -40,37 +40,42 @@ describe('query for signal', () => {
 
   describe('query and agg on signals index', () => {
     test('returns 200 when using single query', async () => {
-      clients.clusterClient.callAsCurrentUser.mockImplementation((endpoint, params) => {
-        expect(params!.body).toMatchObject({ ...typicalSignalsQueryAggs() });
-        return Promise.resolve(true);
-      });
-      const { statusCode } = await server.inject(getSignalsAggsQueryRequest());
+      const { statusCode } = await server.inject(getSignalsQueryRequest());
+
       expect(statusCode).toBe(200);
+      expect(clients.clusterClient.callAsCurrentUser).toHaveBeenCalledWith(
+        'search',
+        expect.objectContaining({ body: typicalSignalsQuery() })
+      );
       expect(myUtils.getIndex).toHaveReturnedWith('fakeindex');
     });
 
     test('returns 200 when using single agg', async () => {
-      clients.clusterClient.callAsCurrentUser.mockImplementation((endpoint, params) => {
-        expect(params!.body).toMatchObject({ ...typicalSignalsQueryAggs() });
-        return Promise.resolve(true);
-      });
       const { statusCode } = await server.inject(getSignalsAggsQueryRequest());
+
       expect(statusCode).toBe(200);
+      expect(clients.clusterClient.callAsCurrentUser).toHaveBeenCalledWith(
+        'search',
+        expect.objectContaining({ body: typicalSignalsQueryAggs() })
+      );
       expect(myUtils.getIndex).toHaveReturnedWith('fakeindex');
     });
 
     test('returns 200 when using aggs and query together', async () => {
-      const allTogether = getSignalsQueryRequest();
-      allTogether.payload = { ...typicalSignalsQueryAggs(), ...typicalSignalsQuery() };
-      clients.clusterClient.callAsCurrentUser.mockImplementation((endpoint, params) => {
-        expect(params!.body).toMatchObject({
-          ...typicalSignalsQueryAggs(),
-          ...typicalSignalsQuery(),
-        });
-        return Promise.resolve(true);
-      });
-      const { statusCode } = await server.inject(allTogether);
+      const request = getSignalsQueryRequest();
+      request.payload = { ...typicalSignalsQueryAggs(), ...typicalSignalsQuery() };
+      const { statusCode } = await server.inject(request);
+
       expect(statusCode).toBe(200);
+      expect(clients.clusterClient.callAsCurrentUser).toHaveBeenCalledWith(
+        'search',
+        expect.objectContaining({
+          body: {
+            ...typicalSignalsQuery(),
+            ...typicalSignalsQueryAggs(),
+          },
+        })
+      );
       expect(myUtils.getIndex).toHaveReturnedWith('fakeindex');
     });