|
26 | 26 | import org.elasticsearch.common.util.BigArrays;
|
27 | 27 | import org.elasticsearch.indices.breaker.CircuitBreakerService;
|
28 | 28 | import org.elasticsearch.indices.breaker.NoneCircuitBreakerService;
|
29 |
| -import org.elasticsearch.tasks.Task; |
30 | 29 | import org.elasticsearch.test.ESTestCase;
|
31 | 30 | import org.elasticsearch.test.transport.MockTransportService;
|
32 | 31 | import org.elasticsearch.threadpool.TestThreadPool;
|
33 | 32 | import org.elasticsearch.threadpool.ThreadPool;
|
34 | 33 | import org.elasticsearch.transport.TcpTransport;
|
35 |
| -import org.elasticsearch.transport.TransportChannel; |
36 | 34 | import org.elasticsearch.transport.TransportException;
|
37 | 35 | import org.elasticsearch.transport.TransportRequest;
|
38 |
| -import org.elasticsearch.transport.TransportRequestHandler; |
39 | 36 | import org.elasticsearch.transport.TransportRequestOptions;
|
40 | 37 | import org.elasticsearch.transport.TransportResponse;
|
41 | 38 | import org.elasticsearch.transport.TransportResponseHandler;
|
@@ -83,22 +80,19 @@ public void testScheduledPing() throws Exception {
|
83 | 80 | serviceB.connectToNode(nodeA);
|
84 | 81 |
|
85 | 82 | assertBusy(() -> {
|
86 |
| - assertThat(nettyA.getPing().getSuccessfulPings(), greaterThan(100L)); |
87 |
| - assertThat(nettyB.getPing().getSuccessfulPings(), greaterThan(100L)); |
| 83 | + assertThat(nettyA.successfulPingCount(), greaterThan(100L)); |
| 84 | + assertThat(nettyB.successfulPingCount(), greaterThan(100L)); |
88 | 85 | });
|
89 |
| - assertThat(nettyA.getPing().getFailedPings(), equalTo(0L)); |
90 |
| - assertThat(nettyB.getPing().getFailedPings(), equalTo(0L)); |
| 86 | + assertThat(nettyA.failedPingCount(), equalTo(0L)); |
| 87 | + assertThat(nettyB.failedPingCount(), equalTo(0L)); |
91 | 88 |
|
92 | 89 | serviceA.registerRequestHandler("internal:sayHello", TransportRequest.Empty::new, ThreadPool.Names.GENERIC,
|
93 |
| - new TransportRequestHandler<TransportRequest.Empty>() { |
94 |
| - @Override |
95 |
| - public void messageReceived(TransportRequest.Empty request, TransportChannel channel, Task task) { |
96 |
| - try { |
97 |
| - channel.sendResponse(TransportResponse.Empty.INSTANCE, TransportResponseOptions.EMPTY); |
98 |
| - } catch (IOException e) { |
99 |
| - logger.error("Unexpected failure", e); |
100 |
| - fail(e.getMessage()); |
101 |
| - } |
| 90 | + (request, channel, task) -> { |
| 91 | + try { |
| 92 | + channel.sendResponse(TransportResponse.Empty.INSTANCE, TransportResponseOptions.EMPTY); |
| 93 | + } catch (IOException e) { |
| 94 | + logger.error("Unexpected failure", e); |
| 95 | + fail(e.getMessage()); |
102 | 96 | }
|
103 | 97 | });
|
104 | 98 |
|
@@ -130,11 +124,11 @@ public void handleException(TransportException exp) {
|
130 | 124 | }
|
131 | 125 |
|
132 | 126 | assertBusy(() -> {
|
133 |
| - assertThat(nettyA.getPing().getSuccessfulPings(), greaterThan(200L)); |
134 |
| - assertThat(nettyB.getPing().getSuccessfulPings(), greaterThan(200L)); |
| 127 | + assertThat(nettyA.successfulPingCount(), greaterThan(200L)); |
| 128 | + assertThat(nettyB.successfulPingCount(), greaterThan(200L)); |
135 | 129 | });
|
136 |
| - assertThat(nettyA.getPing().getFailedPings(), equalTo(0L)); |
137 |
| - assertThat(nettyB.getPing().getFailedPings(), equalTo(0L)); |
| 130 | + assertThat(nettyA.failedPingCount(), equalTo(0L)); |
| 131 | + assertThat(nettyB.failedPingCount(), equalTo(0L)); |
138 | 132 |
|
139 | 133 | Releasables.close(serviceA, serviceB);
|
140 | 134 | terminate(threadPool);
|
|
0 commit comments