ConcurrentUnit

A simple, zero-dependency toolkit for testing multi-threaded code. Supports Java 1.6+.

Introduction

ConcurrentUnit was created to help developers test multi-threaded or asynchronous code. It allows you to perform assertions and wait for operations in any thread, with failures being properly reported back to the main test thread. If an assertion fails, your test fails, regardless of which thread the assertion came from.

Usage

Create a Waiter
Use Waiter.await to block the main test thread.
Use the Waiter.assert calls from any thread to perform assertions.
Once expected assertions are completed, use Waiter.resume call to unblock the main thread.

When your test runs, assertion failures will result in the main thread being interrupted and the failure thrown. If an await call times out before all expected resume calls occur, the test will fail with a TimeoutException.

Examples

Consider a test for a message bus that delivers messages asynchronously:

@Test
public void shouldDeliverMessage() throws Throwable {
  final Waiter waiter = new Waiter();

  messageBus.registerHandler(message -> {
    // Called on separate thread
    waiter.assertEquals(message, "foo");
    waiter.resume();
  };
  
  messageBus.send("foo");
  
  // Wait for resume() to be called
  waiter.await(1000);
}

We can also handle wait for multiple resume calls:

@Test
public void shouldDeliverMessages() throws Throwable {
  final Waiter waiter = new Waiter();

  messageBus.registerHandler(message -> {
    waiter.assertEquals(message, "foo");
    waiter.resume();
  };
  
  messageBus.send("foo");
  messageBus.send("foo");
  messageBus.send("foo");
  
  // Wait for resume() to be called 3 times
  waiter.await(1000, 3);
}

If an assertion fails in any thread, the test will fail as expected:

@Test(expected = AssertionError.class)
public void shouldFail() throws Throwable {
  final Waiter waiter = new Waiter();

  new Thread(() -> {
    waiter.assertTrue(false);
  }).start();
  
  waiter.await();
}

TimeoutException is thrown if resume is not called before the await time is exceeded:

@Test(expected = TimeoutException.class)
public void shouldTimeout() throws Throwable {
  new Waiter().await(1);
}

Alternatively

As a more concise alternative to using the Waiter class, you can extend the ConcurrentTestCase:

class SomeTest extends ConcurrentTestCase {
  @Test
  public void shouldSucceed() throws Throwable {
	new Thread(() -> {
      doSomeWork();
	  threadAssertTrue(true);
	  resume();
	}).start();
	  
    await(1000);
  }
}

Assertions

ConcurrentUnit supports the standard assertions along with Hamcrest Matcher assertions:

waiter.assertEquals(expected, result);
waiter.assertThat(result, is(equalTo(expected)));

Since Hamcrest is an optional dependency, users need to explicitly add it to their classpath (via Maven/Gradle/etc).

Other Examples

More example usages can be found in the WaiterTest or in the following projects:

Docs

JavaDocs are available here.

gkukurin/concurrentunit