1818 */
1919
2020import { BehaviorSubject } from 'rxjs' ;
21- import { Client } from 'elasticsearch' ;
22- import {
23- ILegacyClusterClient ,
24- ILegacyCustomClusterClient ,
25- ILegacyScopedClusterClient ,
26- } from './legacy' ;
21+ import { ILegacyClusterClient , ILegacyCustomClusterClient } from './legacy' ;
22+ import { legacyClientMock } from './legacy/mocks' ;
2723import { ElasticsearchConfig } from './elasticsearch_config' ;
2824import { ElasticsearchService } from './elasticsearch_service' ;
2925import { InternalElasticsearchServiceSetup , ElasticsearchStatusMeta } from './types' ;
3026import { NodesVersionCompatibility } from './version_check/ensure_es_version' ;
3127import { ServiceStatus , ServiceStatusLevels } from '../status' ;
3228
33- const createScopedClusterClientMock = ( ) : jest . Mocked < ILegacyScopedClusterClient > => ( {
34- callAsInternalUser : jest . fn ( ) ,
35- callAsCurrentUser : jest . fn ( ) ,
36- } ) ;
37-
38- const createCustomClusterClientMock = ( ) : jest . Mocked < ILegacyCustomClusterClient > => ( {
39- ...createClusterClientMock ( ) ,
40- close : jest . fn ( ) ,
41- } ) ;
42-
43- function createClusterClientMock ( ) {
44- const client : jest . Mocked < ILegacyClusterClient > = {
45- callAsInternalUser : jest . fn ( ) ,
46- asScoped : jest . fn ( ) ,
47- } ;
48- client . asScoped . mockReturnValue ( createScopedClusterClientMock ( ) ) ;
49- return client ;
50- }
51-
5229interface MockedElasticSearchServiceSetup {
5330 legacy : {
5431 createClient : jest . Mock < ILegacyCustomClusterClient , any > ;
@@ -60,11 +37,13 @@ const createSetupContractMock = () => {
6037 const setupContract : MockedElasticSearchServiceSetup = {
6138 legacy : {
6239 createClient : jest . fn ( ) ,
63- client : createClusterClientMock ( ) ,
40+ client : legacyClientMock . createClusterClient ( ) ,
6441 } ,
6542 } ;
66- setupContract . legacy . createClient . mockReturnValue ( createCustomClusterClientMock ( ) ) ;
67- setupContract . legacy . client . asScoped . mockReturnValue ( createScopedClusterClientMock ( ) ) ;
43+ setupContract . legacy . createClient . mockReturnValue ( legacyClientMock . createCustomClusterClient ( ) ) ;
44+ setupContract . legacy . client . asScoped . mockReturnValue (
45+ legacyClientMock . createScopedClusterClient ( )
46+ ) ;
6847 return setupContract ;
6948} ;
7049
@@ -74,11 +53,14 @@ const createStartContractMock = () => {
7453 const startContract : MockedElasticSearchServiceStart = {
7554 legacy : {
7655 createClient : jest . fn ( ) ,
77- client : createClusterClientMock ( ) ,
56+ client : legacyClientMock . createClusterClient ( ) ,
7857 } ,
7958 } ;
80- startContract . legacy . createClient . mockReturnValue ( createCustomClusterClientMock ( ) ) ;
81- startContract . legacy . client . asScoped . mockReturnValue ( createScopedClusterClientMock ( ) ) ;
59+ startContract . legacy . createClient . mockReturnValue ( legacyClientMock . createCustomClusterClient ( ) ) ;
60+ startContract . legacy . client . asScoped . mockReturnValue (
61+ legacyClientMock . createScopedClusterClient ( )
62+ ) ;
63+
8264 return startContract ;
8365} ;
8466
@@ -104,7 +86,9 @@ const createInternalSetupContractMock = () => {
10486 ...createSetupContractMock ( ) . legacy ,
10587 } ,
10688 } ;
107- setupContract . legacy . client . asScoped . mockReturnValue ( createScopedClusterClientMock ( ) ) ;
89+ setupContract . legacy . client . asScoped . mockReturnValue (
90+ legacyClientMock . createScopedClusterClient ( )
91+ ) ;
10892 return setupContract ;
10993} ;
11094
@@ -121,62 +105,13 @@ const createMock = () => {
121105 return mocked ;
122106} ;
123107
124- const createElasticsearchClientMock = ( ) => {
125- const mocked : jest . Mocked < Client > = {
126- cat : { } as any ,
127- cluster : { } as any ,
128- indices : { } as any ,
129- ingest : { } as any ,
130- nodes : { } as any ,
131- snapshot : { } as any ,
132- tasks : { } as any ,
133- bulk : jest . fn ( ) ,
134- clearScroll : jest . fn ( ) ,
135- count : jest . fn ( ) ,
136- create : jest . fn ( ) ,
137- delete : jest . fn ( ) ,
138- deleteByQuery : jest . fn ( ) ,
139- deleteScript : jest . fn ( ) ,
140- deleteTemplate : jest . fn ( ) ,
141- exists : jest . fn ( ) ,
142- explain : jest . fn ( ) ,
143- fieldStats : jest . fn ( ) ,
144- get : jest . fn ( ) ,
145- getScript : jest . fn ( ) ,
146- getSource : jest . fn ( ) ,
147- getTemplate : jest . fn ( ) ,
148- index : jest . fn ( ) ,
149- info : jest . fn ( ) ,
150- mget : jest . fn ( ) ,
151- msearch : jest . fn ( ) ,
152- msearchTemplate : jest . fn ( ) ,
153- mtermvectors : jest . fn ( ) ,
154- ping : jest . fn ( ) ,
155- putScript : jest . fn ( ) ,
156- putTemplate : jest . fn ( ) ,
157- reindex : jest . fn ( ) ,
158- reindexRethrottle : jest . fn ( ) ,
159- renderSearchTemplate : jest . fn ( ) ,
160- scroll : jest . fn ( ) ,
161- search : jest . fn ( ) ,
162- searchShards : jest . fn ( ) ,
163- searchTemplate : jest . fn ( ) ,
164- suggest : jest . fn ( ) ,
165- termvectors : jest . fn ( ) ,
166- update : jest . fn ( ) ,
167- updateByQuery : jest . fn ( ) ,
168- close : jest . fn ( ) ,
169- } ;
170- return mocked ;
171- } ;
172-
173108export const elasticsearchServiceMock = {
174109 create : createMock ,
175110 createInternalSetup : createInternalSetupContractMock ,
176111 createSetup : createSetupContractMock ,
177112 createStart : createStartContractMock ,
178- createClusterClient : createClusterClientMock ,
179- createCustomClusterClient : createCustomClusterClientMock ,
180- createScopedClusterClient : createScopedClusterClientMock ,
181- createElasticsearchClient : createElasticsearchClientMock ,
113+ createLegacyClusterClient : legacyClientMock . createClusterClient ,
114+ createLegacyCustomClusterClient : legacyClientMock . createCustomClusterClient ,
115+ createLegacyScopedClusterClient : legacyClientMock . createScopedClusterClient ,
116+ createLegacyElasticsearchClient : legacyClientMock . createElasticsearchClient ,
182117} ;
0 commit comments