1818import org .elasticsearch .cluster .node .DiscoveryNode ;
1919import org .elasticsearch .cluster .node .DiscoveryNodeUtils ;
2020import org .elasticsearch .cluster .node .VersionInformation ;
21+ import org .elasticsearch .cluster .project .DefaultProjectResolver ;
22+ import org .elasticsearch .cluster .project .ProjectResolver ;
2123import org .elasticsearch .common .UUIDs ;
2224import org .elasticsearch .common .io .stream .BytesStreamOutput ;
2325import org .elasticsearch .common .io .stream .NamedWriteableRegistry ;
5153import org .elasticsearch .test .tasks .MockTaskManager ;
5254import org .elasticsearch .threadpool .ThreadPool ;
5355import org .elasticsearch .transport .ClusterConnectionManager ;
56+ import org .elasticsearch .transport .ClusterSettingsLinkedProjectConfigService ;
5457import org .elasticsearch .transport .ConnectTransportException ;
5558import org .elasticsearch .transport .ConnectionProfile ;
59+ import org .elasticsearch .transport .LinkedProjectConfigService ;
5660import org .elasticsearch .transport .NodeNotConnectedException ;
5761import org .elasticsearch .transport .RequestHandlerRegistry ;
5862import org .elasticsearch .transport .TcpTransport ;
@@ -192,7 +196,7 @@ public static MockTransportService createNewService(
192196 String nodeId = UUIDs .randomBase64UUID ();
193197 return new MockTransportService (
194198 settings ,
195- new StubbableTransport ( transport ) ,
199+ transport ,
196200 threadPool ,
197201 interceptor ,
198202 boundAddress -> DiscoveryNodeUtils .builder (nodeId )
@@ -203,7 +207,8 @@ public static MockTransportService createNewService(
203207 .version (version )
204208 .build (),
205209 clusterSettings ,
206- createTaskManager (settings , threadPool , taskHeaders , Tracer .NOOP , nodeId )
210+ taskHeaders ,
211+ nodeId
207212 );
208213 }
209214
@@ -223,12 +228,13 @@ public static MockTransportService createMockTransportService(Transport transpor
223228 String nodeId = UUIDs .randomBase64UUID ();
224229 return new MockTransportService (
225230 Settings .EMPTY ,
226- new StubbableTransport ( transport ) ,
231+ transport ,
227232 threadPool ,
228233 TransportService .NOOP_TRANSPORT_INTERCEPTOR ,
229234 (boundAddress ) -> DiscoveryNodeUtils .builder (nodeId ).address (boundAddress .publishAddress ()).build (),
230235 null , // clusterSettings
231- createTaskManager (Settings .EMPTY , threadPool , Set .of (), Tracer .NOOP , nodeId )
236+ Set .of (),
237+ nodeId
232238 );
233239 }
234240
@@ -256,38 +262,22 @@ public MockTransportService(
256262 interceptor ,
257263 localNodeFactory ,
258264 clusterSettings ,
259- createTaskManager (settings , threadPool , taskHeaders , Tracer .NOOP , nodeId )
265+ createTaskManager (settings , threadPool , taskHeaders , Tracer .NOOP , nodeId ),
266+ new ClusterSettingsLinkedProjectConfigService (settings , clusterSettings , DefaultProjectResolver .INSTANCE ),
267+ DefaultProjectResolver .INSTANCE
260268 );
261269 }
262270
263271 public MockTransportService (
264- Settings settings ,
265- Transport transport ,
266- ThreadPool threadPool ,
267- TransportInterceptor interceptor ,
268- Function <BoundTransportAddress , DiscoveryNode > localNodeFactory ,
269- @ Nullable ClusterSettings clusterSettings ,
270- String nodeId
271- ) {
272- this (
273- settings ,
274- new StubbableTransport (transport ),
275- threadPool ,
276- interceptor ,
277- localNodeFactory ,
278- clusterSettings ,
279- createTaskManager (settings , threadPool , Set .of (), Tracer .NOOP , nodeId )
280- );
281- }
282-
283- private MockTransportService (
284272 Settings settings ,
285273 StubbableTransport transport ,
286274 ThreadPool threadPool ,
287275 TransportInterceptor interceptor ,
288276 Function <BoundTransportAddress , DiscoveryNode > localNodeFactory ,
289277 @ Nullable ClusterSettings clusterSettings ,
290- TaskManager taskManager
278+ TaskManager taskManager ,
279+ LinkedProjectConfigService linkedProjectConfigService ,
280+ ProjectResolver projectResolver
291281 ) {
292282 super (
293283 settings ,
@@ -297,7 +287,9 @@ private MockTransportService(
297287 localNodeFactory ,
298288 clusterSettings ,
299289 new StubbableConnectionManager (new ClusterConnectionManager (settings , transport , threadPool .getThreadContext ())),
300- taskManager
290+ taskManager ,
291+ linkedProjectConfigService ,
292+ projectResolver
301293 );
302294 this .original = transport .getDelegate ();
303295 this .testExecutor = EsExecutors .newScaling (
0 commit comments