honeymoose
/
OpenSearch
mirror of https://github.com/honeymoose/OpenSearch.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[TEST] Use a semaphore to block unitl all in-flight requests are released

This commit is contained in:
Simon Willnauer 2016-07-13 10:31:05 +02:00
parent 814c7224f9
commit 067ca1f996
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
test/framework/src/main/java/org/elasticsearch/transport/AbstractSimpleTransportTestCase.java
View File

@ -558,13 +558,13 @@ public abstract class AbstractSimpleTransportTestCase extends ESTestCase {
public void testTimeoutSendExceptionWithDelayedResponse() throws Exception {
CountDownLatch waitForever = new CountDownLatch(1);
CountDownLatch doneWaitingForever = new CountDownLatch(1);
AtomicInteger inFlight = new AtomicInteger(0);
Semaphore inFlight = new Semaphore(Integer.MAX_VALUE);
serviceA.registerRequestHandler("sayHelloTimeoutDelayedResponse", StringMessageRequest::new, ThreadPool.Names.GENERIC,
new TransportRequestHandler<StringMessageRequest>() {
@Override
public void messageReceived(StringMessageRequest request, TransportChannel channel) throws InterruptedException {
String message = request.message;
inFlight.incrementAndGet();
inFlight.acquireUninterruptibly();
try {
if ("forever".equals(message)) {
waitForever.await();
@ -579,7 +579,7 @@ public abstract class AbstractSimpleTransportTestCase extends ESTestCase {
fail(e.getMessage());
}
} finally {
inFlight.decrementAndGet();
inFlight.release();
if ("forever".equals(message)) {
doneWaitingForever.countDown();
}
@ -661,7 +661,7 @@ public abstract class AbstractSimpleTransportTestCase extends ESTestCase {
serviceA.removeHandler("sayHelloTimeoutDelayedResponse");
waitForever.countDown();
doneWaitingForever.await();
assertEquals(0, inFlight.get());
assertTrue(inFlight.tryAcquire(Integer.MAX_VALUE, 10, TimeUnit.SECONDS));
}
@TestLogging(value = "test. transport.tracer:TRACE")