Apache
/
activemq-artemis
mirror of https://github.com/apache/activemq-artemis.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

ARTEMIS-3604 Small test fix on ThresholdActorTest

This commit is contained in:
Clebert Suconic 2021-12-10 22:24:07 -05:00
parent 1e62979577
commit af13d90c57
1 changed files with 27 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
artemis-commons/src/test/java/org/apache/activemq/artemis/utils/actors/ThresholdActorTest.java
View File

@ -17,9 +17,11 @@
package org.apache.activemq.artemis.utils.actors;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.Semaphore;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.concurrent.atomic.AtomicInteger;
@ -119,7 +121,7 @@ public class ThresholdActorTest {
public void block() {
try {
if (!semaphore.tryAcquire()) {
if (!semaphore.tryAcquire(10, TimeUnit.SECONDS)) {
errors.incrementAndGet();
System.err.println("acquire failed");
}
@ -128,18 +130,36 @@ public class ThresholdActorTest {
}
}
public void unblock() {
semaphore.release();
}
@Test
public void testFlow() throws Exception {
final ExecutorService executorService = Executors.newSingleThreadExecutor();
final ExecutorService executorService = Executors.newFixedThreadPool(2);
try {
ThresholdActor<Element> actor = new ThresholdActor<>(executorService, this::process, 20, (e) -> e.size, this::block, semaphore::release);
ThresholdActor<Element> actor = new ThresholdActor<>(executorService, this::process, 20, (e) -> e.size, this::block, this::unblock);
final int LAST_ELEMENT = 1000;
final int LAST_ELEMENT = 1111;
final CountDownLatch latchDone = new CountDownLatch(1);
executorService.execute(() -> {
for (int i = 0; i <= LAST_ELEMENT; i++) {
try {
semaphore.acquire();
semaphore.release();
actor.act(new Element(i, i % 2 == 0 ? 20 : 1));
} catch (Exception e) {
e.printStackTrace();
errors.incrementAndGet();
}
}
latchDone.countDown();
});
Assert.assertTrue(latchDone.await(10, TimeUnit.SECONDS));
Wait.assertEquals(LAST_ELEMENT, lastProcessed::get);
Assert.assertEquals(0, errors.get());