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); |
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); } |
60 |
|
} |
61 |
|
|
62 |
|
<T> void checkCompletedNormally(CompletableFuture<T> f, T value) { |
63 |
< |
try { |
64 |
< |
assertEquals(value, f.get(LONG_DELAY_MS, MILLISECONDS)); |
62 |
< |
} catch (Throwable fail) { threadUnexpectedException(fail); } |
63 |
> |
checkTimedGet(f, value); |
64 |
> |
|
65 |
|
try { |
66 |
|
assertEquals(value, f.join()); |
67 |
|
} catch (Throwable fail) { threadUnexpectedException(fail); } |
78 |
|
} |
79 |
|
|
80 |
|
void checkCompletedWithWrappedCFException(CompletableFuture<?> f) { |
81 |
+ |
long startTime = System.nanoTime(); |
82 |
+ |
long timeoutMillis = LONG_DELAY_MS; |
83 |
|
try { |
84 |
< |
f.get(LONG_DELAY_MS, MILLISECONDS); |
84 |
> |
f.get(timeoutMillis, MILLISECONDS); |
85 |
|
shouldThrow(); |
86 |
|
} catch (ExecutionException success) { |
87 |
|
assertTrue(success.getCause() instanceof CFException); |
88 |
|
} catch (Throwable fail) { threadUnexpectedException(fail); } |
89 |
+ |
assertTrue(millisElapsedSince(startTime) < timeoutMillis/2); |
90 |
+ |
|
91 |
|
try { |
92 |
|
f.join(); |
93 |
|
shouldThrow(); |
113 |
|
|
114 |
|
<U> void checkCompletedExceptionallyWithRootCause(CompletableFuture<U> f, |
115 |
|
Throwable ex) { |
116 |
+ |
long startTime = System.nanoTime(); |
117 |
+ |
long timeoutMillis = LONG_DELAY_MS; |
118 |
|
try { |
119 |
< |
f.get(LONG_DELAY_MS, MILLISECONDS); |
119 |
> |
f.get(timeoutMillis, MILLISECONDS); |
120 |
|
shouldThrow(); |
121 |
|
} catch (ExecutionException success) { |
122 |
|
assertSame(ex, success.getCause()); |
123 |
|
} catch (Throwable fail) { threadUnexpectedException(fail); } |
124 |
+ |
assertTrue(millisElapsedSince(startTime) < timeoutMillis/2); |
125 |
+ |
|
126 |
|
try { |
127 |
|
f.join(); |
128 |
|
shouldThrow(); |
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); |
205 |
|
} |
206 |
|
|
207 |
|
void checkCancelled(CompletableFuture<?> f) { |
208 |
+ |
long startTime = System.nanoTime(); |
209 |
+ |
long timeoutMillis = LONG_DELAY_MS; |
210 |
|
try { |
211 |
< |
f.get(LONG_DELAY_MS, MILLISECONDS); |
211 |
> |
f.get(timeoutMillis, MILLISECONDS); |
212 |
|
shouldThrow(); |
213 |
|
} catch (CancellationException success) { |
214 |
|
} catch (Throwable fail) { threadUnexpectedException(fail); } |
215 |
+ |
assertTrue(millisElapsedSince(startTime) < timeoutMillis/2); |
216 |
+ |
|
217 |
|
try { |
218 |
|
f.join(); |
219 |
|
shouldThrow(); |
234 |
|
} |
235 |
|
|
236 |
|
void checkCompletedWithWrappedCancellationException(CompletableFuture<?> f) { |
237 |
+ |
long startTime = System.nanoTime(); |
238 |
+ |
long timeoutMillis = LONG_DELAY_MS; |
239 |
|
try { |
240 |
< |
f.get(LONG_DELAY_MS, MILLISECONDS); |
240 |
> |
f.get(timeoutMillis, MILLISECONDS); |
241 |
|
shouldThrow(); |
242 |
|
} catch (ExecutionException success) { |
243 |
|
assertTrue(success.getCause() instanceof CancellationException); |
244 |
|
} catch (Throwable fail) { threadUnexpectedException(fail); } |
245 |
+ |
assertTrue(millisElapsedSince(startTime) < timeoutMillis/2); |
246 |
+ |
|
247 |
|
try { |
248 |
|
f.join(); |
249 |
|
shouldThrow(); |
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 |
|
|
586 |
|
} |
587 |
|
} |
588 |
|
|
533 |
– |
|
589 |
|
class CompletableFutureInc extends CheckedIntegerAction |
590 |
|
implements Function<Integer, CompletableFuture<Integer>> |
591 |
|
{ |
624 |
|
} |
625 |
|
} |
626 |
|
|
627 |
+ |
static final boolean defaultExecutorIsCommonPool |
628 |
+ |
= ForkJoinPool.getCommonPoolParallelism() > 1; |
629 |
+ |
|
630 |
|
/** |
631 |
|
* Permits the testing of parallel code for the 3 different |
632 |
|
* execution modes without copy/pasting all the test methods. |
633 |
|
*/ |
634 |
|
enum ExecutionMode { |
635 |
< |
DEFAULT { |
635 |
> |
SYNC { |
636 |
|
public void checkExecutionMode() { |
637 |
|
assertFalse(ThreadExecutor.startedCurrentThread()); |
638 |
|
assertNull(ForkJoinTask.getPool()); |
708 |
|
|
709 |
|
ASYNC { |
710 |
|
public void checkExecutionMode() { |
711 |
< |
assertSame(ForkJoinPool.commonPool(), |
712 |
< |
ForkJoinTask.getPool()); |
711 |
> |
assertEquals(defaultExecutorIsCommonPool, |
712 |
> |
(ForkJoinPool.commonPool() == ForkJoinTask.getPool())); |
713 |
|
} |
714 |
|
public CompletableFuture<Void> runAsync(Runnable a) { |
715 |
|
return CompletableFuture.runAsync(a); |
933 |
|
if (!createIncomplete) f.completeExceptionally(ex); |
934 |
|
final CompletableFuture<Integer> g = f.exceptionally |
935 |
|
((Throwable t) -> { |
936 |
< |
ExecutionMode.DEFAULT.checkExecutionMode(); |
936 |
> |
ExecutionMode.SYNC.checkExecutionMode(); |
937 |
|
threadAssertSame(t, ex); |
938 |
|
a.getAndIncrement(); |
939 |
|
return v1; |
946 |
|
|
947 |
|
public void testExceptionally_exceptionalCompletionActionFailed() { |
948 |
|
for (boolean createIncomplete : new boolean[] { true, false }) |
891 |
– |
for (Integer v1 : new Integer[] { 1, null }) |
949 |
|
{ |
950 |
|
final AtomicInteger a = new AtomicInteger(0); |
951 |
|
final CFException ex1 = new CFException(); |
954 |
|
if (!createIncomplete) f.completeExceptionally(ex1); |
955 |
|
final CompletableFuture<Integer> g = f.exceptionally |
956 |
|
((Throwable t) -> { |
957 |
< |
ExecutionMode.DEFAULT.checkExecutionMode(); |
957 |
> |
ExecutionMode.SYNC.checkExecutionMode(); |
958 |
|
threadAssertSame(t, ex1); |
959 |
|
a.getAndIncrement(); |
960 |
|
throw ex2; |
999 |
|
public void testWhenComplete_exceptionalCompletion() { |
1000 |
|
for (ExecutionMode m : ExecutionMode.values()) |
1001 |
|
for (boolean createIncomplete : new boolean[] { true, false }) |
945 |
– |
for (Integer v1 : new Integer[] { 1, null }) |
1002 |
|
{ |
1003 |
|
final AtomicInteger a = new AtomicInteger(0); |
1004 |
|
final CFException ex = new CFException(); |
1083 |
|
public void testWhenComplete_actionFailedSourceFailed() { |
1084 |
|
for (boolean createIncomplete : new boolean[] { true, false }) |
1085 |
|
for (ExecutionMode m : ExecutionMode.values()) |
1030 |
– |
for (Integer v1 : new Integer[] { 1, null }) |
1086 |
|
{ |
1087 |
|
final AtomicInteger a = new AtomicInteger(0); |
1088 |
|
final CFException ex1 = new CFException(); |
1683 |
|
rs[0].assertValue(subtract(v1, v2)); |
1684 |
|
rs[2].assertValue(subtract(v1, v2)); |
1685 |
|
rs[4].assertValue(subtract(v1, v2)); |
1686 |
< |
|
1686 |
> |
|
1687 |
|
checkCompletedNormally(f, v1); |
1688 |
|
checkCompletedNormally(g, v2); |
1689 |
|
}} |
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 |
|
/** |
3249 |
|
CompletableFuture<Integer> f = new CompletableFuture<>(); |
3250 |
|
CompletableFuture<Integer> g = new CompletableFuture<>(); |
3251 |
|
CompletableFuture<Integer> nullFuture = (CompletableFuture<Integer>)null; |
3145 |
– |
CompletableFuture<?> h; |
3252 |
|
ThreadExecutor exec = new ThreadExecutor(); |
3253 |
|
|
3254 |
|
Runnable[] throwingActions = { |
3255 |
|
() -> CompletableFuture.supplyAsync(null), |
3256 |
|
() -> CompletableFuture.supplyAsync(null, exec), |
3257 |
< |
() -> CompletableFuture.supplyAsync(new IntegerSupplier(ExecutionMode.DEFAULT, 42), null), |
3257 |
> |
() -> CompletableFuture.supplyAsync(new IntegerSupplier(ExecutionMode.SYNC, 42), null), |
3258 |
|
|
3259 |
|
() -> CompletableFuture.runAsync(null), |
3260 |
|
() -> CompletableFuture.runAsync(null, exec), |
3359 |
|
assertSame(f, f.toCompletableFuture()); |
3360 |
|
} |
3361 |
|
|
3362 |
+ |
// jdk9 |
3363 |
+ |
|
3364 |
+ |
/** |
3365 |
+ |
* newIncompleteFuture returns an incomplete CompletableFuture |
3366 |
+ |
*/ |
3367 |
+ |
public void testNewIncompleteFuture() { |
3368 |
+ |
CompletableFuture<Integer> f = new CompletableFuture<>(); |
3369 |
+ |
CompletableFuture<Integer> g = f.newIncompleteFuture(); |
3370 |
+ |
checkIncomplete(f); |
3371 |
+ |
checkIncomplete(g); |
3372 |
+ |
} |
3373 |
+ |
|
3374 |
+ |
/** |
3375 |
+ |
* completedStage returns a completed CompletionStage |
3376 |
+ |
*/ |
3377 |
+ |
public void testCompletedStage() { |
3378 |
+ |
AtomicInteger x = new AtomicInteger(); |
3379 |
+ |
AtomicReference<Throwable> r = new AtomicReference<Throwable>(); |
3380 |
+ |
CompletionStage<Integer> f = CompletableFuture.completedStage(1); |
3381 |
+ |
f.whenComplete((v, e) -> {if (e != null) r.set(e); else x.set(v);}); |
3382 |
+ |
assertEquals(x.get(), 1); |
3383 |
+ |
assertNull(r.get()); |
3384 |
+ |
} |
3385 |
+ |
|
3386 |
+ |
/** |
3387 |
+ |
* defaultExecutor by default returns the commonPool if |
3388 |
+ |
* it supports more than one thread. |
3389 |
+ |
*/ |
3390 |
+ |
public void testDefaultExecutor() { |
3391 |
+ |
CompletableFuture<Integer> f = new CompletableFuture<>(); |
3392 |
+ |
Executor e = f.defaultExecutor(); |
3393 |
+ |
Executor c = ForkJoinPool.commonPool(); |
3394 |
+ |
if (ForkJoinPool.getCommonPoolParallelism() > 1) |
3395 |
+ |
assertSame(e, c); |
3396 |
+ |
else |
3397 |
+ |
assertNotSame(e, c); |
3398 |
+ |
} |
3399 |
+ |
|
3400 |
+ |
/** |
3401 |
+ |
* failedFuture returns a CompletableFuture completed |
3402 |
+ |
* exceptionally with the given Exception |
3403 |
+ |
*/ |
3404 |
+ |
public void testFailedFuture() { |
3405 |
+ |
CFException ex = new CFException(); |
3406 |
+ |
CompletableFuture<Integer> f = CompletableFuture.failedFuture(ex); |
3407 |
+ |
checkCompletedExceptionallyWithRootCause(f, ex); |
3408 |
+ |
} |
3409 |
+ |
|
3410 |
+ |
/** |
3411 |
+ |
* failedFuture(null) throws NPE |
3412 |
+ |
*/ |
3413 |
+ |
public void testFailedFuture2() { |
3414 |
+ |
try { |
3415 |
+ |
CompletableFuture<Integer> f = CompletableFuture.failedFuture(null); |
3416 |
+ |
shouldThrow(); |
3417 |
+ |
} catch (NullPointerException success) {} |
3418 |
+ |
} |
3419 |
+ |
|
3420 |
+ |
/** |
3421 |
+ |
* copy returns a CompletableFuture that is completed normally, |
3422 |
+ |
* with the same value, when source is. |
3423 |
+ |
*/ |
3424 |
+ |
public void testCopy() { |
3425 |
+ |
CompletableFuture<Integer> f = new CompletableFuture<>(); |
3426 |
+ |
CompletableFuture<Integer> g = f.copy(); |
3427 |
+ |
checkIncomplete(f); |
3428 |
+ |
checkIncomplete(g); |
3429 |
+ |
f.complete(1); |
3430 |
+ |
checkCompletedNormally(f, 1); |
3431 |
+ |
checkCompletedNormally(g, 1); |
3432 |
+ |
} |
3433 |
+ |
|
3434 |
+ |
/** |
3435 |
+ |
* copy returns a CompletableFuture that is completed exceptionally |
3436 |
+ |
* when source is. |
3437 |
+ |
*/ |
3438 |
+ |
public void testCopy2() { |
3439 |
+ |
CompletableFuture<Integer> f = new CompletableFuture<>(); |
3440 |
+ |
CompletableFuture<Integer> g = f.copy(); |
3441 |
+ |
checkIncomplete(f); |
3442 |
+ |
checkIncomplete(g); |
3443 |
+ |
CFException ex = new CFException(); |
3444 |
+ |
f.completeExceptionally(ex); |
3445 |
+ |
checkCompletedExceptionally(f, ex); |
3446 |
+ |
checkCompletedWithWrappedCFException(g); |
3447 |
+ |
} |
3448 |
+ |
|
3449 |
+ |
/** |
3450 |
+ |
* minimalCompletionStage returns a CompletableFuture that is |
3451 |
+ |
* completed normally, with the same value, when source is. |
3452 |
+ |
*/ |
3453 |
+ |
public void testMinimalCompletionStage() { |
3454 |
+ |
CompletableFuture<Integer> f = new CompletableFuture<>(); |
3455 |
+ |
CompletionStage<Integer> g = f.minimalCompletionStage(); |
3456 |
+ |
AtomicInteger x = new AtomicInteger(); |
3457 |
+ |
AtomicReference<Throwable> r = new AtomicReference<Throwable>(); |
3458 |
+ |
checkIncomplete(f); |
3459 |
+ |
g.whenComplete((v, e) -> {if (e != null) r.set(e); else x.set(v);}); |
3460 |
+ |
f.complete(1); |
3461 |
+ |
checkCompletedNormally(f, 1); |
3462 |
+ |
assertEquals(x.get(), 1); |
3463 |
+ |
assertNull(r.get()); |
3464 |
+ |
} |
3465 |
+ |
|
3466 |
+ |
/** |
3467 |
+ |
* minimalCompletionStage returns a CompletableFuture that is |
3468 |
+ |
* completed exceptionally when source is. |
3469 |
+ |
*/ |
3470 |
+ |
public void testMinimalCompletionStage2() { |
3471 |
+ |
CompletableFuture<Integer> f = new CompletableFuture<>(); |
3472 |
+ |
CompletionStage<Integer> g = f.minimalCompletionStage(); |
3473 |
+ |
AtomicInteger x = new AtomicInteger(); |
3474 |
+ |
AtomicReference<Throwable> r = new AtomicReference<Throwable>(); |
3475 |
+ |
g.whenComplete((v, e) -> {if (e != null) r.set(e); else x.set(v);}); |
3476 |
+ |
checkIncomplete(f); |
3477 |
+ |
CFException ex = new CFException(); |
3478 |
+ |
f.completeExceptionally(ex); |
3479 |
+ |
checkCompletedExceptionally(f, ex); |
3480 |
+ |
assertEquals(x.get(), 0); |
3481 |
+ |
assertEquals(r.get().getCause(), ex); |
3482 |
+ |
} |
3483 |
+ |
|
3484 |
+ |
/** |
3485 |
+ |
* failedStage returns a CompletionStage completed |
3486 |
+ |
* exceptionally with the given Exception |
3487 |
+ |
*/ |
3488 |
+ |
public void testFailedStage() { |
3489 |
+ |
CFException ex = new CFException(); |
3490 |
+ |
CompletionStage<Integer> f = CompletableFuture.failedStage(ex); |
3491 |
+ |
AtomicInteger x = new AtomicInteger(); |
3492 |
+ |
AtomicReference<Throwable> r = new AtomicReference<Throwable>(); |
3493 |
+ |
f.whenComplete((v, e) -> {if (e != null) r.set(e); else x.set(v);}); |
3494 |
+ |
assertEquals(x.get(), 0); |
3495 |
+ |
assertEquals(r.get().getCause(), ex); |
3496 |
+ |
} |
3497 |
+ |
|
3498 |
+ |
/** |
3499 |
+ |
* completeAsync completes with value of given supplier |
3500 |
+ |
*/ |
3501 |
+ |
public void testCompleteAsync() { |
3502 |
+ |
CompletableFuture<Integer> f = new CompletableFuture<>(); |
3503 |
+ |
f.completeAsync(() -> 1); |
3504 |
+ |
f.join(); |
3505 |
+ |
checkCompletedNormally(f, 1); |
3506 |
+ |
} |
3507 |
+ |
|
3508 |
+ |
/** |
3509 |
+ |
* completeAsync completes exceptionally if given supplier throws |
3510 |
+ |
*/ |
3511 |
+ |
public void testCompleteAsync2() { |
3512 |
+ |
CompletableFuture<Integer> f = new CompletableFuture<>(); |
3513 |
+ |
CFException ex = new CFException(); |
3514 |
+ |
f.completeAsync(() -> {if (true) throw ex; return 1;}); |
3515 |
+ |
try { |
3516 |
+ |
f.join(); |
3517 |
+ |
shouldThrow(); |
3518 |
+ |
} catch (Exception success) {} |
3519 |
+ |
checkCompletedWithWrappedCFException(f); |
3520 |
+ |
} |
3521 |
+ |
|
3522 |
+ |
/** |
3523 |
+ |
* completeAsync with given executor completes with value of given supplier |
3524 |
+ |
*/ |
3525 |
+ |
public void testCompleteAsync3() { |
3526 |
+ |
CompletableFuture<Integer> f = new CompletableFuture<>(); |
3527 |
+ |
f.completeAsync(() -> 1, new ThreadExecutor()); |
3528 |
+ |
f.join(); |
3529 |
+ |
checkCompletedNormally(f, 1); |
3530 |
+ |
} |
3531 |
+ |
|
3532 |
+ |
/** |
3533 |
+ |
* completeAsync with given executor completes exceptionally if |
3534 |
+ |
* given supplier throws |
3535 |
+ |
*/ |
3536 |
+ |
public void testCompleteAsync4() { |
3537 |
+ |
CompletableFuture<Integer> f = new CompletableFuture<>(); |
3538 |
+ |
CFException ex = new CFException(); |
3539 |
+ |
f.completeAsync(() -> {if (true) throw ex; return 1;}, new ThreadExecutor()); |
3540 |
+ |
try { |
3541 |
+ |
f.join(); |
3542 |
+ |
shouldThrow(); |
3543 |
+ |
} catch (Exception success) {} |
3544 |
+ |
checkCompletedWithWrappedCFException(f); |
3545 |
+ |
} |
3546 |
+ |
|
3547 |
+ |
/** |
3548 |
+ |
* orTimeout completes with TimeoutException if not complete |
3549 |
+ |
*/ |
3550 |
+ |
public void testOrTimeout() { |
3551 |
+ |
CompletableFuture<Integer> f = new CompletableFuture<>(); |
3552 |
+ |
f.orTimeout(SHORT_DELAY_MS, MILLISECONDS); |
3553 |
+ |
checkCompletedExceptionallyWithTimeout(f); |
3554 |
+ |
} |
3555 |
+ |
|
3556 |
+ |
/** |
3557 |
+ |
* orTimeout completes normally if completed before timeout |
3558 |
+ |
*/ |
3559 |
+ |
public void testOrTimeout2() { |
3560 |
+ |
CompletableFuture<Integer> f = new CompletableFuture<>(); |
3561 |
+ |
f.complete(1); |
3562 |
+ |
f.orTimeout(SHORT_DELAY_MS, MILLISECONDS); |
3563 |
+ |
checkCompletedNormally(f, 1); |
3564 |
+ |
} |
3565 |
+ |
|
3566 |
+ |
/** |
3567 |
+ |
* completeOnTimeout completes with given value if not complete |
3568 |
+ |
*/ |
3569 |
+ |
public void testCompleteOnTimeout() { |
3570 |
+ |
CompletableFuture<Integer> f = new CompletableFuture<>(); |
3571 |
+ |
f.completeOnTimeout(-1, SHORT_DELAY_MS, MILLISECONDS); |
3572 |
+ |
f.join(); |
3573 |
+ |
checkCompletedNormally(f, -1); |
3574 |
+ |
} |
3575 |
+ |
|
3576 |
+ |
/** |
3577 |
+ |
* completeOnTimeout has no effect if completed within timeout |
3578 |
+ |
*/ |
3579 |
+ |
public void testCompleteOnTimeout2() { |
3580 |
+ |
CompletableFuture<Integer> f = new CompletableFuture<>(); |
3581 |
+ |
f.complete(1); |
3582 |
+ |
f.completeOnTimeout(-1, SHORT_DELAY_MS, MILLISECONDS); |
3583 |
+ |
checkCompletedNormally(f, 1); |
3584 |
+ |
} |
3585 |
+ |
|
3586 |
+ |
/** |
3587 |
+ |
* delayedExecutor returns an executor that delays submission |
3588 |
+ |
*/ |
3589 |
+ |
public void testDelayedExecutor() { |
3590 |
+ |
long timeoutMillis = SMALL_DELAY_MS; |
3591 |
+ |
Executor d = CompletableFuture.delayedExecutor(timeoutMillis, |
3592 |
+ |
MILLISECONDS); |
3593 |
+ |
long startTime = System.nanoTime(); |
3594 |
+ |
CompletableFuture<Integer> f = CompletableFuture.supplyAsync(() -> 1, d); |
3595 |
+ |
assertNull(f.getNow(null)); |
3596 |
+ |
try { |
3597 |
+ |
assertEquals(1, (int) f.get(LONG_DELAY_MS, MILLISECONDS)); |
3598 |
+ |
} catch (Throwable fail) { threadUnexpectedException(fail); } |
3599 |
+ |
assertTrue(millisElapsedSince(startTime) > timeoutMillis/2); |
3600 |
+ |
checkCompletedNormally(f, 1); |
3601 |
+ |
} |
3602 |
+ |
|
3603 |
+ |
/** |
3604 |
+ |
* delayedExecutor for a given executor returns an executor that |
3605 |
+ |
* delays submission |
3606 |
+ |
*/ |
3607 |
+ |
public void testDelayedExecutor2() { |
3608 |
+ |
long timeoutMillis = SMALL_DELAY_MS; |
3609 |
+ |
Executor d = CompletableFuture.delayedExecutor(timeoutMillis, |
3610 |
+ |
MILLISECONDS, |
3611 |
+ |
new ThreadExecutor()); |
3612 |
+ |
long startTime = System.nanoTime(); |
3613 |
+ |
CompletableFuture<Integer> f = CompletableFuture.supplyAsync(() -> 1, d); |
3614 |
+ |
assertNull(f.getNow(null)); |
3615 |
+ |
try { |
3616 |
+ |
assertEquals(1, (int) f.get(LONG_DELAY_MS, MILLISECONDS)); |
3617 |
+ |
} catch (Throwable fail) { threadUnexpectedException(fail); } |
3618 |
+ |
assertTrue(millisElapsedSince(startTime) > timeoutMillis/2); |
3619 |
+ |
checkCompletedNormally(f, 1); |
3620 |
+ |
} |
3621 |
+ |
|
3622 |
|
//--- tests of implementation details; not part of official tck --- |
3623 |
|
|
3624 |
|
Object resultOf(CompletableFuture<?> f) { |