fix more test usages 3

elastic · Apr 19, 2023 · ff14b8e · ff14b8e
1 parent 235188e
commit ff14b8e
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/x-pack/plugins/rollup/server/collectors/register.test.ts b/x-pack/plugins/rollup/server/collectors/register.test.ts
@@ -10,17 +10,18 @@ import { registerRollupUsageCollector } from './register';
 
 describe('registerRollupUsageCollector', () => {
   const mockIndex = 'mock_index';
+  const getIndexForType = () => Promise.resolve(mockIndex);
 
   it('makes a usage collector and registers it`', () => {
     const mockCollectorSet = createUsageCollectionSetupMock();
-    registerRollupUsageCollector(mockCollectorSet, mockIndex);
+    registerRollupUsageCollector(mockCollectorSet, getIndexForType);
     expect(mockCollectorSet.makeUsageCollector).toBeCalledTimes(1);
     expect(mockCollectorSet.registerCollector).toBeCalledTimes(1);
   });
 
   it('makeUsageCollector configs fit the shape', () => {
     const mockCollectorSet = createUsageCollectionSetupMock();
-    registerRollupUsageCollector(mockCollectorSet, mockIndex);
+    registerRollupUsageCollector(mockCollectorSet, getIndexForType);
     expect(mockCollectorSet.makeUsageCollector).toHaveBeenCalledWith({
       type: 'rollups',
       isReady: expect.any(Function),
@@ -81,7 +82,7 @@ describe('registerRollupUsageCollector', () => {
 
   it('makeUsageCollector config.isReady returns true', () => {
     const mockCollectorSet = createUsageCollectionSetupMock();
-    registerRollupUsageCollector(mockCollectorSet, mockIndex);
+    registerRollupUsageCollector(mockCollectorSet, getIndexForType);
     const usageCollectorConfig = mockCollectorSet.makeUsageCollector.mock.calls[0][0];
     expect(usageCollectorConfig.isReady()).toBe(true);
   });