[TEST] First decrement in-flight requests before releasing the latch

This commit is contained in:
Simon Willnauer 2016-07-12 22:58:03 +02:00
parent c493774093
commit eba69ffade

View File

@ -563,9 +563,9 @@ public abstract class AbstractSimpleTransportTestCase extends ESTestCase {
new TransportRequestHandler<StringMessageRequest>() {
@Override
public void messageReceived(StringMessageRequest request, TransportChannel channel) throws InterruptedException {
String message = request.message;
inFlight.incrementAndGet();
try {
String message = request.message;
if ("forever".equals(message)) {
waitForever.await();
} else {
@ -577,13 +577,12 @@ public abstract class AbstractSimpleTransportTestCase extends ESTestCase {
} catch (IOException e) {
logger.error("Unexpected failure", e);
fail(e.getMessage());
} finally {
if ("forever".equals(message)) {
doneWaitingForever.countDown();
}
}
} finally {
inFlight.decrementAndGet();
if ("forever".equals(message)) {
doneWaitingForever.countDown();
}
}
}
});