fix unit tests.

elastic · Jul 7, 2021 · 48074a4 · 48074a4
1 parent 96a1a60
commit 48074a4
Showing 1 changed file with 25 additions and 5 deletions.
diff --git a/x-pack/plugins/apm/server/lib/search_strategies/correlations/search_strategy.test.ts b/x-pack/plugins/apm/server/lib/search_strategies/correlations/search_strategy.test.ts
@@ -122,6 +122,8 @@ describe('APM Correlations search strategy', () => {
       } as unknown) as SearchStrategyDependencies;
       params = {
         index: 'apm-*',
+        start: '2020',
+        end: '2021',
       };
     });
 
@@ -154,10 +156,22 @@ describe('APM Correlations search strategy', () => {
               },
               query: {
                 bool: {
-                  filter: [{ term: { 'processor.event': 'transaction' } }],
+                  filter: [
+                    { term: { 'processor.event': 'transaction' } },
+                    {
+                      range: {
+                        '@timestamp': {
+                          format: 'epoch_millis',
+                          gte: '2020',
+                          lte: '2021',
+                        },
+                      },
+                    },
+                  ],
                 },
               },
               size: 0,
+              track_total_hits: true,
             })
           );
         });
@@ -167,11 +181,17 @@ describe('APM Correlations search strategy', () => {
         it('retrieves the current request', async () => {
           const searchStrategy = await apmCorrelationsSearchStrategyProvider();
           const response = await searchStrategy
-            .search({ id: 'my-search-id', params }, {}, mockDeps)
+            .search({ params }, {}, mockDeps)
             .toPromise();
 
-          expect(response).toEqual(
-            expect.objectContaining({ id: 'my-search-id' })
+          const searchStrategyId = response.id;
+
+          const response2 = await searchStrategy
+            .search({ id: searchStrategyId, params }, {}, mockDeps)
+            .toPromise();
+
+          expect(response2).toEqual(
+            expect.objectContaining({ id: searchStrategyId })
           );
         });
       });
@@ -226,7 +246,7 @@ describe('APM Correlations search strategy', () => {
 
         expect(response2.id).toEqual(response1.id);
         expect(response2).toEqual(
-          expect.objectContaining({ loaded: 10, isRunning: false })
+          expect.objectContaining({ loaded: 100, isRunning: false })
         );
       });
     });