Updating test

elastic · Aug 1, 2024 · dd9ff57 · dd9ff57
1 parent 74bdf34
commit dd9ff57
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 8 deletions.
diff --git a/x-pack/plugins/task_manager/server/lib/task_partitioner.test.ts b/x-pack/plugins/task_manager/server/lib/task_partitioner.test.ts
@@ -9,7 +9,7 @@ import {
   createDiscoveryServiceMock,
   createFindSO,
 } from '../kibana_discovery_service/mock_kibana_discovery_service';
-import { TaskPartitioner } from './task_partitioner';
+import { CACHE_INTERVAL, TaskPartitioner } from './task_partitioner';
 
 const POD_NAME = 'test-pod';
 
@@ -79,19 +79,16 @@ describe('getPartitions()', () => {
 
   test('correctly caches the partitions on 10 second interval ', async () => {
     const taskPartitioner = new TaskPartitioner(POD_NAME, discoveryServiceMock);
-    await taskPartitioner.getPartitions();
-    jest.advanceTimersByTime(3000);
+    const shorterInterval = CACHE_INTERVAL / 2;
 
     await taskPartitioner.getPartitions();
-    jest.advanceTimersByTime(3000);
 
+    jest.advanceTimersByTime(shorterInterval);
     await taskPartitioner.getPartitions();
-    jest.advanceTimersByTime(3000);
 
+    jest.advanceTimersByTime(shorterInterval);
     await taskPartitioner.getPartitions();
-    jest.advanceTimersByTime(3000);
 
-    await taskPartitioner.getPartitions();
     expect(discoveryServiceMock.getActiveKibanaNodes).toHaveBeenCalledTimes(2);
   });
 });
diff --git a/x-pack/plugins/task_manager/server/lib/task_partitioner.ts b/x-pack/plugins/task_manager/server/lib/task_partitioner.ts
@@ -17,7 +17,7 @@ function range(start: number, end: number) {
 }
 
 export const MAX_PARTITIONS = 256;
-const CACHE_INTERVAL = 10000;
+export const CACHE_INTERVAL = 10000;
 
 export class TaskPartitioner {
   private readonly allPartitions: number[];