Class FuturePoolTest

future-java/src/test/java/io/trane/future/FuturePoolTest.java:19–154

import org.junit.Test;

public class FuturePoolTest {

  private <T> T get(Future<T> future) throws CheckedFutureException {
    return future.get(Duration.ofMillis(100));
  }

  @Test
  public void isolate() throws CheckedFutureException {
    ExecutorService es = Executors.newCachedThreadPool();
    try {
      FuturePool pool = FuturePool.apply(es);
      Thread originalThread = Thread.currentThread();

      Future<Integer> future = pool.isolate(() -> {
        assertNotEquals(originalThread, Thread.currentThread());
        return Future.value(1);
      });

      assertEquals(new Integer(1), get(future));
    } finally {
      es.shutdown();
    }
  }

  @Test
  public void async() throws CheckedFutureException {
    ExecutorService es = Executors.newCachedThreadPool();
    try {
      FuturePool pool = FuturePool.apply(es);
      Thread originalThread = Thread.currentThread();

      Future<Integer> future = pool.async(() -> {
        assertNotEquals(originalThread, Thread.currentThread());
        return 1;
      });

      assertEquals(new Integer(1), get(future));
    } finally {
      es.shutdown();
    }
  }

  @Test(expected = IllegalStateException.class)
  public void asyncFailure() throws CheckedFutureException {
    ExecutorService es = Executors.newCachedThreadPool();
    try {
      FuturePool pool = FuturePool.apply(es);
      Thread originalThread = Thread.currentThread();

      Future<Integer> future = pool.async(() -> {
        assertNotEquals(originalThread, Thread.currentThread());
        throw new IllegalStateException();
      });

      get(future);
    } finally {
      es.shutdown();
    }