PromiseTests.java

package org.springframework.async;

import java.util.concurrent.CountDownLatch;

import org.junit.Test;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * @author Jon Brisbin <jon@jbrisbin.com>
 */
@SuppressWarnings({"unchecked"})
public class PromiseTests {

	private static final Logger log = LoggerFactory.getLogger(PromiseTests.class);

	private static enum StringHandler implements CompletionHandler<String, CountDownLatch> {
		INSTANCE;

		@Override public void completed(String result, CountDownLatch latch) {
			log.info("Got result: " + result + " with attachment: " + latch);
			latch.countDown();
		}

		@Override public void failed(Throwable t, CountDownLatch latch) {
			log.error(t.getMessage(), t);
			latch.countDown();
		}

	}

	private static enum VoidHandler implements CompletionHandler<Void, CountDownLatch> {
		INSTANCE;

		@Override public void completed(Void empty, CountDownLatch latch) {
			log.info("Got void result with attachment: " + latch);
			latch.countDown();
		}

		@Override public void failed(Throwable t, CountDownLatch latch) {
			log.error(t.getMessage(), t);
			latch.countDown();
		}

	}

	@Test
	public void testPromiseCompletionHandler() throws InterruptedException {
		final CountDownLatch latch = new CountDownLatch(1);

		LongRunningAction action = new LongRunningAction();
		action.executeLongRunningAction(latch)
					.setCompletionHandler(StringHandler.INSTANCE);

		latch.await();
	}

	@Test
	public void testPromiseFailure() throws InterruptedException {
		final CountDownLatch latch = new CountDownLatch(1);

		LongRunningAction action = new LongRunningAction();
		action.executeFailedAction(latch)
					.setCompletionHandler(StringHandler.INSTANCE);

		latch.await();
	}

	@Test
	public void testPromiseVoid() throws InterruptedException {
		final CountDownLatch latch = new CountDownLatch(1);

		LongRunningAction action = new LongRunningAction();
		action.executeVoidAction(latch)
					.setCompletionHandler(VoidHandler.INSTANCE);

		latch.await();
	}

}