TestIteration() { threadPool = new ThreadPool("transport-client-nodes-service-tests"); transport = new FailAndRetryMockTransport<TestResponse>(getRandom()) { @Override protected TestResponse newResponse() { return new TestResponse(); } }; transportService = new TransportService(Settings.EMPTY, transport, threadPool); transportService.start(); transportClientNodesService = new TransportClientNodesService( Settings.EMPTY, ClusterName.DEFAULT, transportService, threadPool, Headers.EMPTY, Version.CURRENT); nodesCount = randomIntBetween(1, 10); for (int i = 0; i < nodesCount; i++) { transportClientNodesService.addTransportAddresses(new LocalTransportAddress("node" + i)); } transport.endConnectMode(); }
TestIteration() { ClusterName clusterName = new ClusterName("test"); threadPool = new TestThreadPool("transport-client-nodes-service-tests"); transport = new FailAndRetryMockTransport<TestResponse>(random(), clusterName) { @Override public List<String> getLocalAddresses() { return Collections.emptyList(); } @Override protected TestResponse newResponse() { return new TestResponse(); } }; transportService = new TransportService(Settings.EMPTY, transport, threadPool, clusterName) { @Override public <T extends TransportResponse> void sendRequest( DiscoveryNode node, String action, TransportRequest request, final TransportResponseHandler<T> handler) { if (TransportLivenessAction.NAME.equals(action)) { super.sendRequest( node, action, request, wrapLivenessResponseHandler(handler, node, clusterName)); } else { super.sendRequest(node, action, request, handler); } } @Override public <T extends TransportResponse> void sendRequest( DiscoveryNode node, String action, TransportRequest request, TransportRequestOptions options, TransportResponseHandler<T> handler) { if (TransportLivenessAction.NAME.equals(action)) { super.sendRequest( node, action, request, options, wrapLivenessResponseHandler(handler, node, clusterName)); } else { super.sendRequest(node, action, request, options, handler); } } }; transportService.start(); transportService.acceptIncomingRequests(); transportClientNodesService = new TransportClientNodesService( Settings.EMPTY, clusterName, transportService, threadPool, Version.CURRENT); this.nodesCount = randomIntBetween(1, 10); for (int i = 0; i < nodesCount; i++) { transportClientNodesService.addTransportAddresses(new LocalTransportAddress("node" + i)); } transport.endConnectMode(); }