ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/jsr166/jsr166/src/test/tck/CompletableFutureTest.java
(Generate patch)

Comparing jsr166/src/test/tck/CompletableFutureTest.java (file contents):
Revision 1.96 by jsr166, Sat Nov 1 14:50:26 2014 UTC vs.
Revision 1.103 by dl, Thu Sep 3 11:45:34 2015 UTC

# Line 5 | Line 5
5   * http://creativecommons.org/publicdomain/zero/1.0/
6   */
7  
8 < import junit.framework.*;
8 > import static java.util.concurrent.TimeUnit.MILLISECONDS;
9 > import static java.util.concurrent.TimeUnit.SECONDS;
10 >
11 > import java.util.ArrayList;
12 > import java.util.List;
13 > import java.util.Objects;
14   import java.util.concurrent.Callable;
10 import java.util.concurrent.Executor;
11 import java.util.concurrent.ExecutorService;
12 import java.util.concurrent.Executors;
15   import java.util.concurrent.CancellationException;
14 import java.util.concurrent.CountDownLatch;
15 import java.util.concurrent.ExecutionException;
16 import java.util.concurrent.Future;
16   import java.util.concurrent.CompletableFuture;
17   import java.util.concurrent.CompletionException;
18   import java.util.concurrent.CompletionStage;
19 + import java.util.concurrent.ExecutionException;
20 + import java.util.concurrent.Executor;
21   import java.util.concurrent.ForkJoinPool;
22   import java.util.concurrent.ForkJoinTask;
23   import java.util.concurrent.TimeoutException;
24 + import java.util.concurrent.TimeUnit;
25   import java.util.concurrent.atomic.AtomicInteger;
26 < import static java.util.concurrent.TimeUnit.MILLISECONDS;
25 < import static java.util.concurrent.TimeUnit.SECONDS;
26 < import java.util.*;
27 < import java.util.function.Supplier;
28 < import java.util.function.Consumer;
26 > import java.util.concurrent.atomic.AtomicReference;
27   import java.util.function.BiConsumer;
30 import java.util.function.Function;
28   import java.util.function.BiFunction;
29 + import java.util.function.Consumer;
30 + import java.util.function.Function;
31 + import java.util.function.Supplier;
32 +
33 + import junit.framework.Test;
34 + import junit.framework.TestSuite;
35  
36   public class CompletableFutureTest extends JSR166TestCase {
37  
38      public static void main(String[] args) {
39 <        junit.textui.TestRunner.run(suite());
39 >        main(suite(), args);
40      }
41      public static Test suite() {
42          return new TestSuite(CompletableFutureTest.class);
# Line 44 | Line 47 | public class CompletableFutureTest exten
47      void checkIncomplete(CompletableFuture<?> f) {
48          assertFalse(f.isDone());
49          assertFalse(f.isCancelled());
50 <        assertTrue(f.toString().contains("[Not completed]"));
50 >        assertTrue(f.toString().contains("Not completed"));
51          try {
52              assertNull(f.getNow(null));
53          } catch (Throwable fail) { threadUnexpectedException(fail); }
# Line 144 | Line 147 | public class CompletableFutureTest exten
147          assertTrue(f.toString().contains("[Completed exceptionally]"));
148      }
149  
150 +    <U> void checkCompletedExceptionallyWithTimeout(CompletableFuture<U> f) {
151 +        long startTime = System.nanoTime();
152 +        long timeoutMillis = LONG_DELAY_MS;
153 +        try {
154 +            f.get(timeoutMillis, MILLISECONDS);
155 +            shouldThrow();
156 +        } catch (ExecutionException ex) {
157 +            assertTrue(ex.getCause() instanceof TimeoutException);
158 +        } catch (Throwable fail) { threadUnexpectedException(fail); }
159 +        assertTrue(millisElapsedSince(startTime) < timeoutMillis/2);
160 +
161 +        try {
162 +            f.join();
163 +            shouldThrow();
164 +        } catch (Throwable ex) {
165 +            assertTrue(ex.getCause() instanceof TimeoutException);
166 +        }
167 +
168 +        try {
169 +            f.getNow(null);
170 +            shouldThrow();
171 +        } catch (Throwable ex) {
172 +            assertTrue(ex.getCause() instanceof TimeoutException);
173 +        }
174 +
175 +        try {
176 +            f.get();
177 +            shouldThrow();
178 +        } catch (ExecutionException ex) {
179 +            assertTrue(ex.getCause() instanceof TimeoutException);
180 +        } catch (Throwable fail) { threadUnexpectedException(fail); }
181 +
182 +        assertTrue(f.isDone());
183 +        assertFalse(f.isCancelled());
184 +        assertTrue(f.toString().contains("[Completed exceptionally]"));
185 +    }
186 +
187      <U> void checkCompletedWithWrappedException(CompletableFuture<U> f,
188                                                  Throwable ex) {
189          checkCompletedExceptionallyWithRootCause(f, ex);
# Line 272 | Line 312 | public class CompletableFutureTest exten
312      {
313          CompletableFuture<Integer> f = new CompletableFuture<>();
314          checkIncomplete(f);
315 <        assertTrue(f.cancel(true));
316 <        assertTrue(f.cancel(true));
315 >        assertTrue(f.cancel(mayInterruptIfRunning));
316 >        assertTrue(f.cancel(mayInterruptIfRunning));
317 >        assertTrue(f.cancel(!mayInterruptIfRunning));
318          checkCancelled(f);
319      }}
320  
# Line 545 | Line 586 | public class CompletableFutureTest exten
586          }
587      }
588  
548
589      class CompletableFutureInc extends CheckedIntegerAction
590          implements Function<Integer, CompletableFuture<Integer>>
591      {
# Line 906 | Line 946 | public class CompletableFutureTest exten
946  
947      public void testExceptionally_exceptionalCompletionActionFailed() {
948          for (boolean createIncomplete : new boolean[] { true, false })
909        for (Integer v1 : new Integer[] { 1, null })
949      {
950          final AtomicInteger a = new AtomicInteger(0);
951          final CFException ex1 = new CFException();
# Line 960 | Line 999 | public class CompletableFutureTest exten
999      public void testWhenComplete_exceptionalCompletion() {
1000          for (ExecutionMode m : ExecutionMode.values())
1001          for (boolean createIncomplete : new boolean[] { true, false })
963        for (Integer v1 : new Integer[] { 1, null })
1002      {
1003          final AtomicInteger a = new AtomicInteger(0);
1004          final CFException ex = new CFException();
# Line 1045 | Line 1083 | public class CompletableFutureTest exten
1083      public void testWhenComplete_actionFailedSourceFailed() {
1084          for (boolean createIncomplete : new boolean[] { true, false })
1085          for (ExecutionMode m : ExecutionMode.values())
1048        for (Integer v1 : new Integer[] { 1, null })
1086      {
1087          final AtomicInteger a = new AtomicInteger(0);
1088          final CFException ex1 = new CFException();
# Line 2993 | Line 3030 | public class CompletableFutureTest exten
3030          checkCancelled(f);
3031      }}
3032  
3033 +    /**
3034 +     * thenCompose result completes exceptionally if the result of the action does
3035 +     */
3036 +    public void testThenCompose_actionReturnsFailingFuture() {
3037 +        for (ExecutionMode m : ExecutionMode.values())
3038 +        for (int order = 0; order < 6; order++)
3039 +        for (Integer v1 : new Integer[] { 1, null })
3040 +    {
3041 +        final CFException ex = new CFException();
3042 +        final CompletableFuture<Integer> f = new CompletableFuture<>();
3043 +        final CompletableFuture<Integer> g = new CompletableFuture<>();
3044 +        final CompletableFuture<Integer> h;
3045 +        // Test all permutations of orders
3046 +        switch (order) {
3047 +        case 0:
3048 +            assertTrue(f.complete(v1));
3049 +            assertTrue(g.completeExceptionally(ex));
3050 +            h = m.thenCompose(f, (x -> g));
3051 +            break;
3052 +        case 1:
3053 +            assertTrue(f.complete(v1));
3054 +            h = m.thenCompose(f, (x -> g));
3055 +            assertTrue(g.completeExceptionally(ex));
3056 +            break;
3057 +        case 2:
3058 +            assertTrue(g.completeExceptionally(ex));
3059 +            assertTrue(f.complete(v1));
3060 +            h = m.thenCompose(f, (x -> g));
3061 +            break;
3062 +        case 3:
3063 +            assertTrue(g.completeExceptionally(ex));
3064 +            h = m.thenCompose(f, (x -> g));
3065 +            assertTrue(f.complete(v1));
3066 +            break;
3067 +        case 4:
3068 +            h = m.thenCompose(f, (x -> g));
3069 +            assertTrue(f.complete(v1));
3070 +            assertTrue(g.completeExceptionally(ex));
3071 +            break;
3072 +        case 5:
3073 +            h = m.thenCompose(f, (x -> g));
3074 +            assertTrue(f.complete(v1));
3075 +            assertTrue(g.completeExceptionally(ex));
3076 +            break;
3077 +        default: throw new AssertionError();
3078 +        }
3079 +
3080 +        checkCompletedExceptionally(g, ex);
3081 +        checkCompletedWithWrappedException(h, ex);
3082 +        checkCompletedNormally(f, v1);
3083 +    }}
3084 +
3085      // other static methods
3086  
3087      /**
# Line 3160 | Line 3249 | public class CompletableFutureTest exten
3249          CompletableFuture<Integer> f = new CompletableFuture<>();
3250          CompletableFuture<Integer> g = new CompletableFuture<>();
3251          CompletableFuture<Integer> nullFuture = (CompletableFuture<Integer>)null;
3163        CompletableFuture<?> h;
3252          ThreadExecutor exec = new ThreadExecutor();
3253  
3254          Runnable[] throwingActions = {
# Line 3245 | Line 3333 | public class CompletableFutureTest exten
3333              () -> f.exceptionally(null),
3334  
3335              () -> f.handle(null),
3248
3336              () -> CompletableFuture.allOf((CompletableFuture<?>)null),
3337              () -> CompletableFuture.allOf((CompletableFuture<?>[])null),
3338              () -> CompletableFuture.allOf(f, null),
# Line 3271 | Line 3358 | public class CompletableFutureTest exten
3358          assertSame(f, f.toCompletableFuture());
3359      }
3360  
3361 +    // jdk9
3362 +    
3363 +    /**
3364 +     * newIncompleteFuture returns an incomplete CompletableFuture
3365 +     */
3366 +    public void testNewIncompleteFuture() {
3367 +        CompletableFuture<Integer> f = new CompletableFuture<>();
3368 +        CompletableFuture<Integer> g = f.newIncompleteFuture();
3369 +        checkIncomplete(f);
3370 +        checkIncomplete(g);
3371 +    }
3372 +
3373 +    /**
3374 +     * completedStage returns a completed CompletionStage
3375 +     */
3376 +    public void testCompletedStage() {
3377 +        AtomicInteger x = new AtomicInteger();
3378 +        AtomicReference<Throwable> r = new AtomicReference<Throwable>();
3379 +        CompletionStage<Integer> f = CompletableFuture.completedStage(1);
3380 +        f.whenComplete((v, e) -> {if (e != null) r.set(e); else x.set(v);});
3381 +        assertEquals(x.get(), 1);
3382 +        assertNull(r.get());
3383 +    }
3384 +
3385 +    /**
3386 +     * defaultExecutor by default returns the commonPool if
3387 +     * it supports at least one thread.
3388 +     */
3389 +    public void testDefaultExecutor() {
3390 +        CompletableFuture<Integer> f = new CompletableFuture<>();
3391 +        Executor e = f.defaultExecutor();
3392 +        Executor c =  ForkJoinPool.commonPool();
3393 +        if (ForkJoinPool.getCommonPoolParallelism() > 0)
3394 +            assertSame(e, c);
3395 +    }
3396 +
3397 +    /**
3398 +     * failedFuture returns a CompletableFuture completed
3399 +     * exceptionally with the given Exception
3400 +     */
3401 +    public void testFailedFuture() {
3402 +        CFException ex = new CFException();
3403 +        CompletableFuture<Integer> f = CompletableFuture.failedFuture(ex);
3404 +        checkCompletedExceptionallyWithRootCause(f, ex);
3405 +    }
3406 +
3407 +    /**
3408 +     * failedFuture(null) throws NPE
3409 +     */
3410 +    public void testFailedFuture2() {
3411 +        try {
3412 +            CompletableFuture<Integer> f = CompletableFuture.failedFuture(null);
3413 +        } catch(NullPointerException success) {
3414 +        }
3415 +    }
3416 +
3417 +    /**
3418 +     * copy returns a CompletableFuture that is completed normally,
3419 +     * with the same value, when source is.
3420 +     */
3421 +    public void testCopy() {
3422 +        CompletableFuture<Integer> f = new CompletableFuture<>();
3423 +        CompletableFuture<Integer> g = f.copy();
3424 +        checkIncomplete(f);
3425 +        checkIncomplete(g);
3426 +        f.complete(1);
3427 +        checkCompletedNormally(f, 1);
3428 +        checkCompletedNormally(g, 1);
3429 +    }
3430 +
3431 +    /**
3432 +     * copy returns a CompletableFuture that is completed exceptionally
3433 +     * when source is.
3434 +     */
3435 +    public void testCopy2() {
3436 +        CompletableFuture<Integer> f = new CompletableFuture<>();
3437 +        CompletableFuture<Integer> g = f.copy();
3438 +        checkIncomplete(f);
3439 +        checkIncomplete(g);
3440 +        CFException ex = new CFException();
3441 +        f.completeExceptionally(ex);
3442 +        checkCompletedExceptionally(f, ex);
3443 +        checkCompletedWithWrappedCFException(g);
3444 +    }
3445 +
3446 +    /**
3447 +     * minimalCompletionStage returns a CompletableFuture that is
3448 +     * completed normally, with the same value, when source is.
3449 +     */
3450 +    public void testMinimalCompletionStage() {
3451 +        CompletableFuture<Integer> f = new CompletableFuture<>();
3452 +        CompletionStage<Integer> g = f.minimalCompletionStage();
3453 +        AtomicInteger x = new AtomicInteger();
3454 +        AtomicReference<Throwable> r = new AtomicReference<Throwable>();
3455 +        checkIncomplete(f);
3456 +        g.whenComplete((v, e) -> {if (e != null) r.set(e); else x.set(v);});
3457 +        f.complete(1);
3458 +        checkCompletedNormally(f, 1);
3459 +        assertEquals(x.get(), 1);
3460 +        assertNull(r.get());
3461 +    }
3462 +
3463 +    /**
3464 +     * minimalCompletionStage returns a CompletableFuture that is
3465 +     * completed exceptionally when source is.
3466 +     */
3467 +    public void testMinimalCompletionStage2() {
3468 +        CompletableFuture<Integer> f = new CompletableFuture<>();
3469 +        CompletionStage<Integer> g = f.minimalCompletionStage();
3470 +        AtomicInteger x = new AtomicInteger();
3471 +        AtomicReference<Throwable> r = new AtomicReference<Throwable>();
3472 +        g.whenComplete((v, e) -> {if (e != null) r.set(e); else x.set(v);});
3473 +        checkIncomplete(f);
3474 +        CFException ex = new CFException();
3475 +        f.completeExceptionally(ex);
3476 +        checkCompletedExceptionally(f, ex);
3477 +        assertEquals(x.get(), 0);
3478 +        assertEquals(r.get().getCause(), ex);
3479 +    }
3480 +
3481 +    /**
3482 +     * failedStage returns a Completionstage completed
3483 +     * exceptionally with the given Exception
3484 +     */
3485 +    public void testFailedStage() {
3486 +        CFException ex = new CFException();
3487 +        CompletionStage<Integer> f = CompletableFuture.failedStage(ex);
3488 +        AtomicInteger x = new AtomicInteger();
3489 +        AtomicReference<Throwable> r = new AtomicReference<Throwable>();
3490 +        f.whenComplete((v, e) -> {if (e != null) r.set(e); else x.set(v);});
3491 +        assertEquals(x.get(), 0);
3492 +        assertEquals(r.get().getCause(), ex);
3493 +    }
3494 +
3495 +    /**
3496 +     * completeAsync completes with value of given supplier
3497 +     */
3498 +    public void testCompleteAsync() {
3499 +        CompletableFuture<Integer> f = new CompletableFuture<>();
3500 +        f.completeAsync(() -> 1);
3501 +        f.join();
3502 +        checkCompletedNormally(f, 1);
3503 +    }
3504 +
3505 +    /**
3506 +     * completeAsync completes exceptionally if given supplier throws
3507 +     */
3508 +    public void testCompleteAsync2() {
3509 +        CompletableFuture<Integer> f = new CompletableFuture<>();
3510 +        CFException ex = new CFException();
3511 +        f.completeAsync(() -> {if (true) throw ex; return 1;});
3512 +        try {
3513 +            f.join();
3514 +            shouldThrow();
3515 +        } catch(Exception success) {
3516 +        }
3517 +        checkCompletedWithWrappedCFException(f);
3518 +    }
3519 +
3520 +    /**
3521 +     * completeAsync with given executor completes with value of given supplier
3522 +     */
3523 +    public void testCompleteAsync3() {
3524 +        CompletableFuture<Integer> f = new CompletableFuture<>();
3525 +        f.completeAsync(() -> 1, new ThreadExecutor());
3526 +        f.join();
3527 +        checkCompletedNormally(f, 1);
3528 +    }
3529 +
3530 +    /**
3531 +     * completeAsync with given executor completes exceptionally if
3532 +     * given supplier throws
3533 +     */
3534 +    public void testCompleteAsync4() {
3535 +        CompletableFuture<Integer> f = new CompletableFuture<>();
3536 +        CFException ex = new CFException();
3537 +        f.completeAsync(() -> {if (true) throw ex; return 1;}, new ThreadExecutor());
3538 +        try {
3539 +            f.join();
3540 +            shouldThrow();
3541 +        } catch(Exception success) {
3542 +        }
3543 +        checkCompletedWithWrappedCFException(f);
3544 +    }
3545 +
3546 +    /**
3547 +     *  orTimeout completes with TimeoutException if not complete
3548 +     */
3549 +    public void testOrTimeout() {
3550 +        CompletableFuture<Integer> f = new CompletableFuture<>();
3551 +        f.orTimeout(SHORT_DELAY_MS, TimeUnit.MILLISECONDS);
3552 +        checkCompletedExceptionallyWithTimeout(f);
3553 +    }
3554 +
3555 +    /**
3556 +     *  orTimeout completes normally if completed before timeout
3557 +     */
3558 +    public void testOrTimeout2() {
3559 +        CompletableFuture<Integer> f = new CompletableFuture<>();
3560 +        f.complete(1);
3561 +        f.orTimeout(SHORT_DELAY_MS, TimeUnit.MILLISECONDS);
3562 +        checkCompletedNormally(f, 1);
3563 +    }
3564 +
3565 +    /**
3566 +     *  completeOnTimeout completes with given value if not complete
3567 +     */
3568 +    public void testCompleteOnTimeout() {
3569 +        CompletableFuture<Integer> f = new CompletableFuture<>();
3570 +        f.completeOnTimeout(-1, SHORT_DELAY_MS, TimeUnit.MILLISECONDS);
3571 +        f.join();
3572 +        checkCompletedNormally(f, -1);
3573 +    }
3574 +
3575 +    /**
3576 +     *  completeOnTimeout has no effect if completed within timeout
3577 +     */
3578 +    public void testCompleteOnTimeout2() {
3579 +        CompletableFuture<Integer> f = new CompletableFuture<>();
3580 +        f.complete(1);
3581 +        f.completeOnTimeout(-1, SHORT_DELAY_MS, TimeUnit.MILLISECONDS);
3582 +        checkCompletedNormally(f, 1);
3583 +    }
3584 +
3585 +    /**
3586 +     * delayedExecutor returns an executor that delays submission
3587 +     */
3588 +    public void testDelayedExecutor() {
3589 +        long timeoutMillis = SMALL_DELAY_MS;
3590 +        Executor d = CompletableFuture.delayedExecutor(timeoutMillis,
3591 +                                                       MILLISECONDS);
3592 +        long startTime = System.nanoTime();
3593 +        CompletableFuture<Integer> f = CompletableFuture.supplyAsync(() -> 1, d);
3594 +        assertNull(f.getNow(null));
3595 +        try {
3596 +            f.get(LONG_DELAY_MS, MILLISECONDS);
3597 +        } catch (Throwable fail) { threadUnexpectedException(fail); }
3598 +        assertTrue(millisElapsedSince(startTime) > timeoutMillis/2);
3599 +        checkCompletedNormally(f, 1);
3600 +    }
3601 +
3602 +    /**
3603 +     * delayedExecutor for a given executor returns an executor that
3604 +     * delays submission
3605 +     */
3606 +    public void testDelayedExecutor2() {
3607 +        long timeoutMillis = SMALL_DELAY_MS;
3608 +        Executor d = CompletableFuture.delayedExecutor(timeoutMillis,
3609 +                                                       MILLISECONDS,
3610 +                                                       new ThreadExecutor());
3611 +        long startTime = System.nanoTime();
3612 +        CompletableFuture<Integer> f = CompletableFuture.supplyAsync(() -> 1, d);
3613 +        assertNull(f.getNow(null));
3614 +        try {
3615 +            f.get(LONG_DELAY_MS, MILLISECONDS);
3616 +        } catch (Throwable fail) { threadUnexpectedException(fail); }
3617 +        assertTrue(millisElapsedSince(startTime) > timeoutMillis/2);
3618 +        checkCompletedNormally(f, 1);
3619 +    }
3620 +
3621      //--- tests of implementation details; not part of official tck ---
3622  
3623      Object resultOf(CompletableFuture<?> f) {

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines