uses product response header in createApiResponse

elastic · Aug 17, 2021 · 810fbdc · 810fbdc
1 parent 0cb74e0
commit 810fbdc
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/core/server/elasticsearch/client/mocks.ts b/src/core/server/elasticsearch/client/mocks.ts
@@ -11,6 +11,7 @@ import { TransportRequestPromise } from '@elastic/elasticsearch/lib/Transport';
 import type { DeeplyMockedKeys } from '@kbn/utility-types/jest';
 import { ElasticsearchClient } from './types';
 import { ICustomClusterClient } from './cluster_client';
+import { PRODUCT_RESPONSE_HEADER } from '../supported_server_response_check';
 
 const createInternalClientMock = (
   res?: MockedTransportRequestPromise<unknown>
@@ -142,7 +143,7 @@ export type MockedTransportRequestPromise<T> = TransportRequestPromise<T> & {
 const createSuccessTransportRequestPromise = <T>(
   body: T,
   { statusCode = 200 }: { statusCode?: number } = {},
-  headers: Record<string, string | string[]> = { 'x-elastic-product': 'Elasticsearch' }
+  headers: Record<string, string | string[]> = { [PRODUCT_RESPONSE_HEADER]: 'Elasticsearch' }
 ): MockedTransportRequestPromise<ApiResponse<T>> => {
   const response = createApiResponse({ body, statusCode, headers });
   const promise = Promise.resolve(response);
@@ -163,7 +164,7 @@ function createApiResponse<TResponse = Record<string, any>>(
   return {
     body: {} as any,
     statusCode: 200,
-    headers: { 'x-elastic-product': 'Elasticsearch' },
+    headers: { [PRODUCT_RESPONSE_HEADER]: 'Elasticsearch' },
     warnings: [],
     meta: {} as any,
     ...opts,

diff --git a/src/core/server/elasticsearch/index.ts b/src/core/server/elasticsearch/index.ts
@@ -41,4 +41,5 @@ export { getRequestDebugMeta, getErrorMessage } from './client';
 export {
   isSupportedEsServer,
   isNotFoundFromUnsupportedServer,
+  PRODUCT_RESPONSE_HEADER,
 } from './supported_server_response_check';