[Test] testHostOnMessages - only decrease latch after setting transport addresses

This commit is contained in:
Boaz Leskes 2014-07-12 09:11:27 +02:00
parent 3689f67a76
commit 5e3742762a
1 changed files with 2 additions and 2 deletions

View File

@ -1065,9 +1065,9 @@ public abstract class AbstractSimpleTransportTests extends ElasticsearchTestCase
@Override
public void messageReceived(TestRequest request, TransportChannel channel) throws Exception {
latch.countDown();
addressA.set(request.remoteAddress());
channel.sendResponse(new TestResponse());
latch.countDown();
}
@Override
@ -1088,8 +1088,8 @@ public abstract class AbstractSimpleTransportTests extends ElasticsearchTestCase
@Override
public void handleResponse(TestResponse response) {
latch.countDown();
addressB.set(response.remoteAddress());
latch.countDown();
}
@Override