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.atomic.AtomicInteger; |
24 |
– |
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; |
25 |
|
import java.util.function.BiConsumer; |
30 |
– |
import java.util.function.Function; |
26 |
|
import java.util.function.BiFunction; |
27 |
+ |
import java.util.function.Consumer; |
28 |
+ |
import java.util.function.Function; |
29 |
+ |
import java.util.function.Supplier; |
30 |
+ |
|
31 |
+ |
import junit.framework.Test; |
32 |
+ |
import junit.framework.TestSuite; |
33 |
|
|
34 |
|
public class CompletableFutureTest extends JSR166TestCase { |
35 |
|
|
273 |
|
{ |
274 |
|
CompletableFuture<Integer> f = new CompletableFuture<>(); |
275 |
|
checkIncomplete(f); |
276 |
< |
assertTrue(f.cancel(true)); |
277 |
< |
assertTrue(f.cancel(true)); |
276 |
> |
assertTrue(f.cancel(mayInterruptIfRunning)); |
277 |
> |
assertTrue(f.cancel(mayInterruptIfRunning)); |
278 |
> |
assertTrue(f.cancel(!mayInterruptIfRunning)); |
279 |
|
checkCancelled(f); |
280 |
|
}} |
281 |
|
|
547 |
|
} |
548 |
|
} |
549 |
|
|
548 |
– |
|
550 |
|
class CompletableFutureInc extends CheckedIntegerAction |
551 |
|
implements Function<Integer, CompletableFuture<Integer>> |
552 |
|
{ |
585 |
|
} |
586 |
|
} |
587 |
|
|
588 |
+ |
static final boolean defaultExecutorIsCommonPool |
589 |
+ |
= ForkJoinPool.getCommonPoolParallelism() > 1; |
590 |
+ |
|
591 |
|
/** |
592 |
|
* Permits the testing of parallel code for the 3 different |
593 |
|
* execution modes without copy/pasting all the test methods. |
669 |
|
|
670 |
|
ASYNC { |
671 |
|
public void checkExecutionMode() { |
672 |
< |
assertSame(ForkJoinPool.commonPool(), |
673 |
< |
ForkJoinTask.getPool()); |
672 |
> |
assertEquals(defaultExecutorIsCommonPool, |
673 |
> |
(ForkJoinPool.commonPool() == ForkJoinTask.getPool())); |
674 |
|
} |
675 |
|
public CompletableFuture<Void> runAsync(Runnable a) { |
676 |
|
return CompletableFuture.runAsync(a); |
907 |
|
|
908 |
|
public void testExceptionally_exceptionalCompletionActionFailed() { |
909 |
|
for (boolean createIncomplete : new boolean[] { true, false }) |
906 |
– |
for (Integer v1 : new Integer[] { 1, null }) |
910 |
|
{ |
911 |
|
final AtomicInteger a = new AtomicInteger(0); |
912 |
|
final CFException ex1 = new CFException(); |
960 |
|
public void testWhenComplete_exceptionalCompletion() { |
961 |
|
for (ExecutionMode m : ExecutionMode.values()) |
962 |
|
for (boolean createIncomplete : new boolean[] { true, false }) |
960 |
– |
for (Integer v1 : new Integer[] { 1, null }) |
963 |
|
{ |
964 |
|
final AtomicInteger a = new AtomicInteger(0); |
965 |
|
final CFException ex = new CFException(); |
1044 |
|
public void testWhenComplete_actionFailedSourceFailed() { |
1045 |
|
for (boolean createIncomplete : new boolean[] { true, false }) |
1046 |
|
for (ExecutionMode m : ExecutionMode.values()) |
1045 |
– |
for (Integer v1 : new Integer[] { 1, null }) |
1047 |
|
{ |
1048 |
|
final AtomicInteger a = new AtomicInteger(0); |
1049 |
|
final CFException ex1 = new CFException(); |
2991 |
|
checkCancelled(f); |
2992 |
|
}} |
2993 |
|
|
2994 |
+ |
/** |
2995 |
+ |
* thenCompose result completes exceptionally if the result of the action does |
2996 |
+ |
*/ |
2997 |
+ |
public void testThenCompose_actionReturnsFailingFuture() { |
2998 |
+ |
for (ExecutionMode m : ExecutionMode.values()) |
2999 |
+ |
for (int order = 0; order < 6; order++) |
3000 |
+ |
for (Integer v1 : new Integer[] { 1, null }) |
3001 |
+ |
{ |
3002 |
+ |
final CFException ex = new CFException(); |
3003 |
+ |
final CompletableFuture<Integer> f = new CompletableFuture<>(); |
3004 |
+ |
final CompletableFuture<Integer> g = new CompletableFuture<>(); |
3005 |
+ |
final CompletableFuture<Integer> h; |
3006 |
+ |
// Test all permutations of orders |
3007 |
+ |
switch (order) { |
3008 |
+ |
case 0: |
3009 |
+ |
assertTrue(f.complete(v1)); |
3010 |
+ |
assertTrue(g.completeExceptionally(ex)); |
3011 |
+ |
h = m.thenCompose(f, (x -> g)); |
3012 |
+ |
break; |
3013 |
+ |
case 1: |
3014 |
+ |
assertTrue(f.complete(v1)); |
3015 |
+ |
h = m.thenCompose(f, (x -> g)); |
3016 |
+ |
assertTrue(g.completeExceptionally(ex)); |
3017 |
+ |
break; |
3018 |
+ |
case 2: |
3019 |
+ |
assertTrue(g.completeExceptionally(ex)); |
3020 |
+ |
assertTrue(f.complete(v1)); |
3021 |
+ |
h = m.thenCompose(f, (x -> g)); |
3022 |
+ |
break; |
3023 |
+ |
case 3: |
3024 |
+ |
assertTrue(g.completeExceptionally(ex)); |
3025 |
+ |
h = m.thenCompose(f, (x -> g)); |
3026 |
+ |
assertTrue(f.complete(v1)); |
3027 |
+ |
break; |
3028 |
+ |
case 4: |
3029 |
+ |
h = m.thenCompose(f, (x -> g)); |
3030 |
+ |
assertTrue(f.complete(v1)); |
3031 |
+ |
assertTrue(g.completeExceptionally(ex)); |
3032 |
+ |
break; |
3033 |
+ |
case 5: |
3034 |
+ |
h = m.thenCompose(f, (x -> g)); |
3035 |
+ |
assertTrue(f.complete(v1)); |
3036 |
+ |
assertTrue(g.completeExceptionally(ex)); |
3037 |
+ |
break; |
3038 |
+ |
default: throw new AssertionError(); |
3039 |
+ |
} |
3040 |
+ |
|
3041 |
+ |
checkCompletedExceptionally(g, ex); |
3042 |
+ |
checkCompletedWithWrappedException(h, ex); |
3043 |
+ |
checkCompletedNormally(f, v1); |
3044 |
+ |
}} |
3045 |
+ |
|
3046 |
|
// other static methods |
3047 |
|
|
3048 |
|
/** |
3210 |
|
CompletableFuture<Integer> f = new CompletableFuture<>(); |
3211 |
|
CompletableFuture<Integer> g = new CompletableFuture<>(); |
3212 |
|
CompletableFuture<Integer> nullFuture = (CompletableFuture<Integer>)null; |
3160 |
– |
CompletableFuture<?> h; |
3213 |
|
ThreadExecutor exec = new ThreadExecutor(); |
3214 |
|
|
3215 |
|
Runnable[] throwingActions = { |