fix more test usages

elastic · Apr 18, 2023 · 8466aa5 · 8466aa5
1 parent 4537691
commit 8466aa5
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/...e_collection/server/collectors/saved_objects_counts/saved_objects_count_collector.test.ts b/...e_collection/server/collectors/saved_objects_counts/saved_objects_count_collector.test.ts
@@ -18,9 +18,10 @@ describe('saved_objects_count_collector', () => {
   const fetchContextMock = createCollectorFetchContextMock();
 
   const kibanaIndex = '.kibana-tests';
+  const getAllIndices = () => Promise.resolve([kibanaIndex]);
 
   beforeAll(() =>
-    registerSavedObjectsCountUsageCollector(usageCollectionMock, kibanaIndex, () =>
+    registerSavedObjectsCountUsageCollector(usageCollectionMock, getAllIndices, () =>
       Promise.resolve(['type_one', 'type_two', 'type-three', 'type-four'])
     )
   );
@@ -82,7 +83,7 @@ describe('saved_objects_count_collector', () => {
 
     expect(getSavedObjectsCountsMock).toHaveBeenCalledWith(
       fetchContextMock.esClient,
-      kibanaIndex,
+      [kibanaIndex],
       ['type_one', 'type_two', 'type-three', 'type-four'],
       false
     );

diff --git a/x-pack/plugins/spaces/server/usage_collection/spaces_usage_collector.test.ts b/x-pack/plugins/spaces/server/usage_collection/spaces_usage_collector.test.ts
@@ -45,6 +45,7 @@ const MOCK_USAGE_STATS: UsageStats = {
 };
 
 const kibanaIndex = '.kibana-tests';
+const getIndexForType = () => Promise.resolve(kibanaIndex);
 
 function setup({
   license = { isAvailable: true },
@@ -122,7 +123,7 @@ describe('error handling', () => {
       license: { isAvailable: true, type: 'basic' },
     });
     const collector = getSpacesUsageCollector(usageCollection as any, {
-      kibanaIndex,
+      getIndexForType,
       features,
       licensing,
       usageStatsServicePromise: Promise.resolve(usageStatsService),
@@ -146,7 +147,7 @@ describe('with a basic license', () => {
 
   beforeAll(async () => {
     const collector = getSpacesUsageCollector(usageCollection as any, {
-      kibanaIndex,
+      getIndexForType,
       features,
       licensing,
       usageStatsServicePromise: Promise.resolve(usageStatsService),
@@ -205,7 +206,7 @@ describe('with no license', () => {
 
   beforeAll(async () => {
     const collector = getSpacesUsageCollector(usageCollection as any, {
-      kibanaIndex,
+      getIndexForType,
       features,
       licensing,
       usageStatsServicePromise: Promise.resolve(usageStatsService),
@@ -246,7 +247,7 @@ describe('with platinum license', () => {
 
   beforeAll(async () => {
     const collector = getSpacesUsageCollector(usageCollection as any, {
-      kibanaIndex,
+      getIndexForType,
       features,
       licensing,
       usageStatsServicePromise: Promise.resolve(usageStatsService),