From a98550b4f1d741126cd57a3d43c48fc4156013ce Mon Sep 17 00:00:00 2001
From: "Christiane (Tina) Heiligers" <christiane.heiligers@elastic.co>
Date: Tue, 22 Dec 2020 11:23:54 -0700
Subject: [PATCH] Reporting usage collector fetch migration (#86675)

---
 .../reporting/server/usage/get_reporting_usage.ts | 15 +++++++++------
 .../usage/reporting_usage_collector.test.ts       |  2 +-
 .../server/usage/reporting_usage_collector.ts     |  4 ++--
 x-pack/plugins/reporting/server/usage/types.ts    |  2 +-
 4 files changed, 13 insertions(+), 10 deletions(-)

diff --git a/x-pack/plugins/reporting/server/usage/get_reporting_usage.ts b/x-pack/plugins/reporting/server/usage/get_reporting_usage.ts
index 1211d4c2cf1c33..6d7f5c2e367fa0 100644
--- a/x-pack/plugins/reporting/server/usage/get_reporting_usage.ts
+++ b/x-pack/plugins/reporting/server/usage/get_reporting_usage.ts
@@ -5,7 +5,7 @@
  */
 
 import { get } from 'lodash';
-import { LegacyAPICaller } from 'kibana/server';
+import { ElasticsearchClient, SearchResponse } from 'kibana/server';
 import { ReportingConfig } from '../';
 import { ExportTypesRegistry } from '../lib/export_types_registry';
 import { GetLicense } from './';
@@ -18,7 +18,7 @@ import {
   KeyCountBucket,
   RangeStats,
   ReportingUsageType,
-  SearchResponse,
+  ReportingUsageSearchResponse,
   StatusByAppBucket,
 } from './types';
 
@@ -99,7 +99,9 @@ type RangeStatSets = Partial<RangeStats> & {
   last7Days: Partial<RangeStats>;
 };
 
-async function handleResponse(response: SearchResponse): Promise<Partial<RangeStatSets>> {
+type ESResponse = Partial<SearchResponse<ReportingUsageSearchResponse>>;
+
+async function handleResponse(response: ESResponse): Promise<Partial<RangeStatSets>> {
   const buckets = get(response, 'aggregations.ranges.buckets');
   if (!buckets) {
     return {};
@@ -118,7 +120,7 @@ async function handleResponse(response: SearchResponse): Promise<Partial<RangeSt
 export async function getReportingUsage(
   config: ReportingConfig,
   getLicense: GetLicense,
-  callCluster: LegacyAPICaller,
+  esClient: ElasticsearchClient,
   exportTypesRegistry: ExportTypesRegistry
 ): Promise<ReportingUsageType> {
   const reportingIndex = config.get('index');
@@ -165,8 +167,9 @@ export async function getReportingUsage(
   };
 
   const featureAvailability = await getLicense();
-  return callCluster('search', params)
-    .then((response: SearchResponse) => handleResponse(response))
+  return esClient
+    .search(params)
+    .then(({ body: response }) => handleResponse(response))
     .then(
       (usage: Partial<RangeStatSets>): ReportingUsageType => {
         // Allow this to explicitly throw an exception if/when this config is deprecated,
diff --git a/x-pack/plugins/reporting/server/usage/reporting_usage_collector.test.ts b/x-pack/plugins/reporting/server/usage/reporting_usage_collector.test.ts
index 7cae5e9b6f9563..8b0c442c12b977 100644
--- a/x-pack/plugins/reporting/server/usage/reporting_usage_collector.test.ts
+++ b/x-pack/plugins/reporting/server/usage/reporting_usage_collector.test.ts
@@ -59,7 +59,7 @@ const getResponseMock = (base = {}) => base;
 
 const getMockFetchClients = (resp: any) => {
   const fetchParamsMock = createCollectorFetchContextMock();
-  fetchParamsMock.callCluster.mockResolvedValue(resp);
+  fetchParamsMock.esClient.search = jest.fn().mockResolvedValue({ body: resp });
   return fetchParamsMock;
 };
 describe('license checks', () => {
diff --git a/x-pack/plugins/reporting/server/usage/reporting_usage_collector.ts b/x-pack/plugins/reporting/server/usage/reporting_usage_collector.ts
index f4209730b68ce9..547c331784c5f4 100644
--- a/x-pack/plugins/reporting/server/usage/reporting_usage_collector.ts
+++ b/x-pack/plugins/reporting/server/usage/reporting_usage_collector.ts
@@ -26,9 +26,9 @@ export function getReportingUsageCollector(
 ) {
   return usageCollection.makeUsageCollector<ReportingUsageType>({
     type: 'reporting',
-    fetch: ({ callCluster }: CollectorFetchContext) => {
+    fetch: ({ esClient }: CollectorFetchContext) => {
       const config = reporting.getConfig();
-      return getReportingUsage(config, getLicense, callCluster, exportTypesRegistry);
+      return getReportingUsage(config, getLicense, esClient, exportTypesRegistry);
     },
     isReady,
     schema: reportingSchema,
diff --git a/x-pack/plugins/reporting/server/usage/types.ts b/x-pack/plugins/reporting/server/usage/types.ts
index 1ff680eff8eafb..fe7838240f2fa2 100644
--- a/x-pack/plugins/reporting/server/usage/types.ts
+++ b/x-pack/plugins/reporting/server/usage/types.ts
@@ -152,7 +152,7 @@ export interface AggregationResultBuckets {
   doc_count: number;
 }
 
-export interface SearchResponse {
+export interface ReportingUsageSearchResponse {
   aggregations: {
     ranges: {
       buckets: {