move response mocks to mocks.ts

src/core/server/elasticsearch/client/client_facade.mock.ts

@@ -17,30 +17,8 @@
  * under the License.
  */
 
-import { ApiResponse } from '@elastic/elasticsearch';
-import { TransportRequestPromise } from '@elastic/elasticsearch/lib/Transport';
 import { ClientFacade } from './client_facade';
 
-const createApiResponse = <T>(body: T): TransportRequestPromise<ApiResponse<T>> => {
-  const response: ApiResponse<T> = {
-    body,
-    statusCode: 200,
-    warnings: [],
-    headers: {},
-    meta: {} as any,
-  };
-  const promise = Promise.resolve(response);
-  (promise as TransportRequestPromise<ApiResponse<T>>).abort = () => undefined;
-
-  return promise as TransportRequestPromise<ApiResponse<T>>;
-};
-
-const createApiError = (err: any): TransportRequestPromise<never> => {
-  const promise = Promise.reject(err);
-  (promise as TransportRequestPromise<never>).abort = () => undefined;
-  return promise as TransportRequestPromise<never>;
-};
-
 const createFacadeMock = () => {
   const mock: DeeplyMockedKeys<ClientFacade> = {
     transport: {
@@ -435,6 +413,4 @@ const createFacadeMock = () => {
 
 export const clientFacadeMock = {
   create: createFacadeMock,
-  createApiResponse,
-  createApiError,
 };

src/core/server/elasticsearch/client/mocks.ts

@@ -17,6 +17,8 @@
  * under the License.
  */
 
+import { ApiResponse } from '@elastic/elasticsearch';
+import { TransportRequestPromise } from '@elastic/elasticsearch/lib/Transport';
 import { clientFacadeMock } from './client_facade.mock';
 import { IScopedClusterClient } from './scoped_cluster_client';
 import { IClusterClient, ICustomClusterClient } from './cluster_client';
@@ -58,11 +60,35 @@ const createCustomClusterClientMock = () => {
   return mock;
 };
 
+type MockedTransportRequestPromise<T> = TransportRequestPromise<T> & {
+  abort: jest.MockedFunction<() => undefined>;
+};
+
+const createMockedClientResponse = <T>(body: T): MockedTransportRequestPromise<ApiResponse<T>> => {
+  const response: ApiResponse<T> = {
+    body,
+    statusCode: 200,
+    warnings: [],
+    headers: {},
+    meta: {} as any,
+  };
+  const promise = Promise.resolve(response);
+  (promise as MockedTransportRequestPromise<ApiResponse<T>>).abort = jest.fn();
+
+  return promise as MockedTransportRequestPromise<ApiResponse<T>>;
+};
+
+const createMockedClientError = (err: any): MockedTransportRequestPromise<never> => {
+  const promise = Promise.reject(err);
+  (promise as MockedTransportRequestPromise<never>).abort = jest.fn();
+  return promise as MockedTransportRequestPromise<never>;
+};
+
 export const elasticsearchClientMock = {
   createClusterClient: createClusterClientMock,
   createCustomClusterClient: createCustomClusterClientMock,
   createScopedClusterClient: createScopedClusterClientMock,
   createFacade: clientFacadeMock.create,
-  createClientResponse: clientFacadeMock.createApiResponse,
-  createClientError: clientFacadeMock.createApiError,
+  createClientResponse: createMockedClientResponse,
+  createClientError: createMockedClientError,
 };