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.95 by jsr166, Wed Jun 25 15:32:10 2014 UTC vs.
Revision 1.153 by jsr166, Sun Jun 26 19:17:08 2016 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 > import static java.util.concurrent.CompletableFuture.completedFuture;
11 > import static java.util.concurrent.CompletableFuture.failedFuture;
12 >
13 > import java.lang.reflect.Method;
14 > import java.lang.reflect.Modifier;
15 >
16 > import java.util.stream.Collectors;
17 > import java.util.stream.Stream;
18 >
19 > import java.util.ArrayList;
20 > import java.util.Arrays;
21 > import java.util.List;
22 > import java.util.Objects;
23 > import java.util.Set;
24   import java.util.concurrent.Callable;
10 import java.util.concurrent.Executor;
11 import java.util.concurrent.ExecutorService;
12 import java.util.concurrent.Executors;
25   import java.util.concurrent.CancellationException;
14 import java.util.concurrent.CountDownLatch;
15 import java.util.concurrent.ExecutionException;
16 import java.util.concurrent.Future;
26   import java.util.concurrent.CompletableFuture;
27   import java.util.concurrent.CompletionException;
28   import java.util.concurrent.CompletionStage;
29 + import java.util.concurrent.ExecutionException;
30 + import java.util.concurrent.Executor;
31   import java.util.concurrent.ForkJoinPool;
32   import java.util.concurrent.ForkJoinTask;
33 + import java.util.concurrent.RejectedExecutionException;
34   import java.util.concurrent.TimeoutException;
35 + import java.util.concurrent.TimeUnit;
36   import java.util.concurrent.atomic.AtomicInteger;
37 < 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;
37 > import java.util.concurrent.atomic.AtomicReference;
38   import java.util.function.BiConsumer;
30 import java.util.function.Function;
39   import java.util.function.BiFunction;
40 + import java.util.function.Consumer;
41 + import java.util.function.Function;
42 + import java.util.function.Predicate;
43 + import java.util.function.Supplier;
44 +
45 + import junit.framework.AssertionFailedError;
46 + import junit.framework.Test;
47 + import junit.framework.TestSuite;
48  
49   public class CompletableFutureTest extends JSR166TestCase {
50  
51      public static void main(String[] args) {
52 <        junit.textui.TestRunner.run(suite());
52 >        main(suite(), args);
53      }
54      public static Test suite() {
55          return new TestSuite(CompletableFutureTest.class);
# Line 44 | Line 60 | public class CompletableFutureTest exten
60      void checkIncomplete(CompletableFuture<?> f) {
61          assertFalse(f.isDone());
62          assertFalse(f.isCancelled());
63 <        assertTrue(f.toString().contains("[Not completed]"));
63 >        assertTrue(f.toString().contains("Not completed"));
64          try {
65              assertNull(f.getNow(null));
66          } catch (Throwable fail) { threadUnexpectedException(fail); }
# Line 74 | Line 90 | public class CompletableFutureTest exten
90          assertTrue(f.toString().contains("[Completed normally]"));
91      }
92  
93 <    void checkCompletedWithWrappedCFException(CompletableFuture<?> f) {
93 >    /**
94 >     * Returns the "raw" internal exceptional completion of f,
95 >     * without any additional wrapping with CompletionException.
96 >     */
97 >    <U> Throwable exceptionalCompletion(CompletableFuture<U> f) {
98 >        // handle (and whenComplete) can distinguish between "direct"
99 >        // and "wrapped" exceptional completion
100 >        return f.handle((U u, Throwable t) -> t).join();
101 >    }
102 >
103 >    void checkCompletedExceptionally(CompletableFuture<?> f,
104 >                                     boolean wrapped,
105 >                                     Consumer<Throwable> checker) {
106 >        Throwable cause = exceptionalCompletion(f);
107 >        if (wrapped) {
108 >            assertTrue(cause instanceof CompletionException);
109 >            cause = cause.getCause();
110 >        }
111 >        checker.accept(cause);
112 >
113          long startTime = System.nanoTime();
79        long timeoutMillis = LONG_DELAY_MS;
114          try {
115 <            f.get(timeoutMillis, MILLISECONDS);
115 >            f.get(LONG_DELAY_MS, MILLISECONDS);
116              shouldThrow();
117          } catch (ExecutionException success) {
118 <            assertTrue(success.getCause() instanceof CFException);
118 >            assertSame(cause, success.getCause());
119          } catch (Throwable fail) { threadUnexpectedException(fail); }
120 <        assertTrue(millisElapsedSince(startTime) < timeoutMillis/2);
120 >        assertTrue(millisElapsedSince(startTime) < LONG_DELAY_MS / 2);
121  
122          try {
123              f.join();
124              shouldThrow();
125          } catch (CompletionException success) {
126 <            assertTrue(success.getCause() instanceof CFException);
127 <        }
126 >            assertSame(cause, success.getCause());
127 >        } catch (Throwable fail) { threadUnexpectedException(fail); }
128 >
129          try {
130              f.getNow(null);
131              shouldThrow();
132          } catch (CompletionException success) {
133 <            assertTrue(success.getCause() instanceof CFException);
134 <        }
133 >            assertSame(cause, success.getCause());
134 >        } catch (Throwable fail) { threadUnexpectedException(fail); }
135 >
136          try {
137              f.get();
138              shouldThrow();
139          } catch (ExecutionException success) {
140 <            assertTrue(success.getCause() instanceof CFException);
140 >            assertSame(cause, success.getCause());
141          } catch (Throwable fail) { threadUnexpectedException(fail); }
142 <        assertTrue(f.isDone());
142 >
143          assertFalse(f.isCancelled());
144 +        assertTrue(f.isDone());
145 +        assertTrue(f.isCompletedExceptionally());
146          assertTrue(f.toString().contains("[Completed exceptionally]"));
147      }
148  
149 <    <U> void checkCompletedExceptionallyWithRootCause(CompletableFuture<U> f,
150 <                                                      Throwable ex) {
151 <        long startTime = System.nanoTime();
152 <        long timeoutMillis = LONG_DELAY_MS;
115 <        try {
116 <            f.get(timeoutMillis, MILLISECONDS);
117 <            shouldThrow();
118 <        } catch (ExecutionException success) {
119 <            assertSame(ex, success.getCause());
120 <        } catch (Throwable fail) { threadUnexpectedException(fail); }
121 <        assertTrue(millisElapsedSince(startTime) < timeoutMillis/2);
149 >    void checkCompletedWithWrappedCFException(CompletableFuture<?> f) {
150 >        checkCompletedExceptionally(f, true,
151 >            (t) -> assertTrue(t instanceof CFException));
152 >    }
153  
154 <        try {
155 <            f.join();
156 <            shouldThrow();
157 <        } catch (CompletionException success) {
127 <            assertSame(ex, success.getCause());
128 <        }
129 <        try {
130 <            f.getNow(null);
131 <            shouldThrow();
132 <        } catch (CompletionException success) {
133 <            assertSame(ex, success.getCause());
134 <        }
135 <        try {
136 <            f.get();
137 <            shouldThrow();
138 <        } catch (ExecutionException success) {
139 <            assertSame(ex, success.getCause());
140 <        } catch (Throwable fail) { threadUnexpectedException(fail); }
154 >    void checkCompletedWithWrappedCancellationException(CompletableFuture<?> f) {
155 >        checkCompletedExceptionally(f, true,
156 >            (t) -> assertTrue(t instanceof CancellationException));
157 >    }
158  
159 <        assertTrue(f.isDone());
160 <        assertFalse(f.isCancelled());
161 <        assertTrue(f.toString().contains("[Completed exceptionally]"));
159 >    void checkCompletedWithTimeoutException(CompletableFuture<?> f) {
160 >        checkCompletedExceptionally(f, false,
161 >            (t) -> assertTrue(t instanceof TimeoutException));
162      }
163  
164 <    <U> void checkCompletedWithWrappedException(CompletableFuture<U> f,
165 <                                                Throwable ex) {
166 <        checkCompletedExceptionallyWithRootCause(f, ex);
150 <        try {
151 <            CompletableFuture<Throwable> spy = f.handle
152 <                ((U u, Throwable t) -> t);
153 <            assertTrue(spy.join() instanceof CompletionException);
154 <            assertSame(ex, spy.join().getCause());
155 <        } catch (Throwable fail) { threadUnexpectedException(fail); }
164 >    void checkCompletedWithWrappedException(CompletableFuture<?> f,
165 >                                            Throwable ex) {
166 >        checkCompletedExceptionally(f, true, (t) -> assertSame(t, ex));
167      }
168  
169 <    <U> void checkCompletedExceptionally(CompletableFuture<U> f, Throwable ex) {
170 <        checkCompletedExceptionallyWithRootCause(f, ex);
160 <        try {
161 <            CompletableFuture<Throwable> spy = f.handle
162 <                ((U u, Throwable t) -> t);
163 <            assertSame(ex, spy.join());
164 <        } catch (Throwable fail) { threadUnexpectedException(fail); }
169 >    void checkCompletedExceptionally(CompletableFuture<?> f, Throwable ex) {
170 >        checkCompletedExceptionally(f, false, (t) -> assertSame(t, ex));
171      }
172  
173      void checkCancelled(CompletableFuture<?> f) {
174          long startTime = System.nanoTime();
169        long timeoutMillis = LONG_DELAY_MS;
175          try {
176 <            f.get(timeoutMillis, MILLISECONDS);
176 >            f.get(LONG_DELAY_MS, MILLISECONDS);
177              shouldThrow();
178          } catch (CancellationException success) {
179          } catch (Throwable fail) { threadUnexpectedException(fail); }
180 <        assertTrue(millisElapsedSince(startTime) < timeoutMillis/2);
180 >        assertTrue(millisElapsedSince(startTime) < LONG_DELAY_MS / 2);
181  
182          try {
183              f.join();
# Line 187 | Line 192 | public class CompletableFutureTest exten
192              shouldThrow();
193          } catch (CancellationException success) {
194          } catch (Throwable fail) { threadUnexpectedException(fail); }
190        assertTrue(f.isDone());
191        assertTrue(f.isCompletedExceptionally());
192        assertTrue(f.isCancelled());
193        assertTrue(f.toString().contains("[Completed exceptionally]"));
194    }
195  
196 <    void checkCompletedWithWrappedCancellationException(CompletableFuture<?> f) {
197 <        long startTime = System.nanoTime();
198 <        long timeoutMillis = LONG_DELAY_MS;
199 <        try {
200 <            f.get(timeoutMillis, MILLISECONDS);
201 <            shouldThrow();
202 <        } catch (ExecutionException success) {
203 <            assertTrue(success.getCause() instanceof CancellationException);
204 <        } catch (Throwable fail) { threadUnexpectedException(fail); }
205 <        assertTrue(millisElapsedSince(startTime) < timeoutMillis/2);
196 >        assertTrue(exceptionalCompletion(f) instanceof CancellationException);
197  
207        try {
208            f.join();
209            shouldThrow();
210        } catch (CompletionException success) {
211            assertTrue(success.getCause() instanceof CancellationException);
212        }
213        try {
214            f.getNow(null);
215            shouldThrow();
216        } catch (CompletionException success) {
217            assertTrue(success.getCause() instanceof CancellationException);
218        }
219        try {
220            f.get();
221            shouldThrow();
222        } catch (ExecutionException success) {
223            assertTrue(success.getCause() instanceof CancellationException);
224        } catch (Throwable fail) { threadUnexpectedException(fail); }
198          assertTrue(f.isDone());
226        assertFalse(f.isCancelled());
199          assertTrue(f.isCompletedExceptionally());
200 +        assertTrue(f.isCancelled());
201          assertTrue(f.toString().contains("[Completed exceptionally]"));
202      }
203  
# Line 272 | Line 245 | public class CompletableFutureTest exten
245      {
246          CompletableFuture<Integer> f = new CompletableFuture<>();
247          checkIncomplete(f);
248 <        assertTrue(f.cancel(true));
249 <        assertTrue(f.cancel(true));
248 >        assertTrue(f.cancel(mayInterruptIfRunning));
249 >        assertTrue(f.cancel(mayInterruptIfRunning));
250 >        assertTrue(f.cancel(!mayInterruptIfRunning));
251          checkCancelled(f);
252      }}
253  
# Line 486 | Line 460 | public class CompletableFutureTest exten
460      class FailingSupplier extends CheckedAction
461          implements Supplier<Integer>
462      {
463 <        FailingSupplier(ExecutionMode m) { super(m); }
463 >        final CFException ex;
464 >        FailingSupplier(ExecutionMode m) { super(m); ex = new CFException(); }
465          public Integer get() {
466              invoked();
467 <            throw new CFException();
467 >            throw ex;
468          }
469      }
470  
471      class FailingConsumer extends CheckedIntegerAction
472          implements Consumer<Integer>
473      {
474 <        FailingConsumer(ExecutionMode m) { super(m); }
474 >        final CFException ex;
475 >        FailingConsumer(ExecutionMode m) { super(m); ex = new CFException(); }
476          public void accept(Integer x) {
477              invoked();
478              value = x;
479 <            throw new CFException();
479 >            throw ex;
480          }
481      }
482  
483      class FailingBiConsumer extends CheckedIntegerAction
484          implements BiConsumer<Integer, Integer>
485      {
486 <        FailingBiConsumer(ExecutionMode m) { super(m); }
486 >        final CFException ex;
487 >        FailingBiConsumer(ExecutionMode m) { super(m); ex = new CFException(); }
488          public void accept(Integer x, Integer y) {
489              invoked();
490              value = subtract(x, y);
491 <            throw new CFException();
491 >            throw ex;
492          }
493      }
494  
495      class FailingFunction extends CheckedIntegerAction
496          implements Function<Integer, Integer>
497      {
498 <        FailingFunction(ExecutionMode m) { super(m); }
498 >        final CFException ex;
499 >        FailingFunction(ExecutionMode m) { super(m); ex = new CFException(); }
500          public Integer apply(Integer x) {
501              invoked();
502              value = x;
503 <            throw new CFException();
503 >            throw ex;
504          }
505      }
506  
507      class FailingBiFunction extends CheckedIntegerAction
508          implements BiFunction<Integer, Integer, Integer>
509      {
510 <        FailingBiFunction(ExecutionMode m) { super(m); }
510 >        final CFException ex;
511 >        FailingBiFunction(ExecutionMode m) { super(m); ex = new CFException(); }
512          public Integer apply(Integer x, Integer y) {
513              invoked();
514              value = subtract(x, y);
515 <            throw new CFException();
515 >            throw ex;
516          }
517      }
518  
519      class FailingRunnable extends CheckedAction implements Runnable {
520 <        FailingRunnable(ExecutionMode m) { super(m); }
520 >        final CFException ex;
521 >        FailingRunnable(ExecutionMode m) { super(m); ex = new CFException(); }
522          public void run() {
523              invoked();
524 <            throw new CFException();
524 >            throw ex;
525          }
526      }
527  
548
528      class CompletableFutureInc extends CheckedIntegerAction
529          implements Function<Integer, CompletableFuture<Integer>>
530      {
# Line 562 | Line 541 | public class CompletableFutureTest exten
541      class FailingCompletableFutureFunction extends CheckedIntegerAction
542          implements Function<Integer, CompletableFuture<Integer>>
543      {
544 <        FailingCompletableFutureFunction(ExecutionMode m) { super(m); }
544 >        final CFException ex;
545 >        FailingCompletableFutureFunction(ExecutionMode m) { super(m); ex = new CFException(); }
546          public CompletableFuture<Integer> apply(Integer x) {
547              invoked();
548              value = x;
549 <            throw new CFException();
549 >            throw ex;
550          }
551      }
552  
# Line 584 | Line 564 | public class CompletableFutureTest exten
564          }
565      }
566  
567 +    static final boolean defaultExecutorIsCommonPool
568 +        = ForkJoinPool.getCommonPoolParallelism() > 1;
569 +
570      /**
571       * Permits the testing of parallel code for the 3 different
572       * execution modes without copy/pasting all the test methods.
# Line 665 | Line 648 | public class CompletableFutureTest exten
648  
649          ASYNC {
650              public void checkExecutionMode() {
651 <                assertSame(ForkJoinPool.commonPool(),
652 <                           ForkJoinTask.getPool());
651 >                assertEquals(defaultExecutorIsCommonPool,
652 >                             (ForkJoinPool.commonPool() == ForkJoinTask.getPool()));
653              }
654              public CompletableFuture<Void> runAsync(Runnable a) {
655                  return CompletableFuture.runAsync(a);
# Line 865 | Line 848 | public class CompletableFutureTest exten
848          if (!createIncomplete) assertTrue(f.complete(v1));
849          final CompletableFuture<Integer> g = f.exceptionally
850              ((Throwable t) -> {
868                // Should not be called
851                  a.getAndIncrement();
852 <                throw new AssertionError();
852 >                threadFail("should not be called");
853 >                return null;            // unreached
854              });
855          if (createIncomplete) assertTrue(f.complete(v1));
856  
# Line 901 | Line 884 | public class CompletableFutureTest exten
884          assertEquals(1, a.get());
885      }}
886  
887 +    /**
888 +     * If an "exceptionally action" throws an exception, it completes
889 +     * exceptionally with that exception
890 +     */
891      public void testExceptionally_exceptionalCompletionActionFailed() {
892          for (boolean createIncomplete : new boolean[] { true, false })
906        for (Integer v1 : new Integer[] { 1, null })
893      {
894          final AtomicInteger a = new AtomicInteger(0);
895          final CFException ex1 = new CFException();
# Line 920 | Line 906 | public class CompletableFutureTest exten
906          if (createIncomplete) f.completeExceptionally(ex1);
907  
908          checkCompletedWithWrappedException(g, ex2);
909 +        checkCompletedExceptionally(f, ex1);
910          assertEquals(1, a.get());
911      }}
912  
# Line 927 | Line 914 | public class CompletableFutureTest exten
914       * whenComplete action executes on normal completion, propagating
915       * source result.
916       */
917 <    public void testWhenComplete_normalCompletion1() {
917 >    public void testWhenComplete_normalCompletion() {
918          for (ExecutionMode m : ExecutionMode.values())
919          for (boolean createIncomplete : new boolean[] { true, false })
920          for (Integer v1 : new Integer[] { 1, null })
# Line 937 | Line 924 | public class CompletableFutureTest exten
924          if (!createIncomplete) assertTrue(f.complete(v1));
925          final CompletableFuture<Integer> g = m.whenComplete
926              (f,
927 <             (Integer x, Throwable t) -> {
927 >             (Integer result, Throwable t) -> {
928                  m.checkExecutionMode();
929 <                threadAssertSame(x, v1);
929 >                threadAssertSame(result, v1);
930                  threadAssertNull(t);
931                  a.getAndIncrement();
932              });
# Line 957 | Line 944 | public class CompletableFutureTest exten
944      public void testWhenComplete_exceptionalCompletion() {
945          for (ExecutionMode m : ExecutionMode.values())
946          for (boolean createIncomplete : new boolean[] { true, false })
960        for (Integer v1 : new Integer[] { 1, null })
947      {
948          final AtomicInteger a = new AtomicInteger(0);
949          final CFException ex = new CFException();
# Line 965 | Line 951 | public class CompletableFutureTest exten
951          if (!createIncomplete) f.completeExceptionally(ex);
952          final CompletableFuture<Integer> g = m.whenComplete
953              (f,
954 <             (Integer x, Throwable t) -> {
954 >             (Integer result, Throwable t) -> {
955                  m.checkExecutionMode();
956 <                threadAssertNull(x);
956 >                threadAssertNull(result);
957                  threadAssertSame(t, ex);
958                  a.getAndIncrement();
959              });
# Line 992 | Line 978 | public class CompletableFutureTest exten
978          if (!createIncomplete) assertTrue(f.cancel(mayInterruptIfRunning));
979          final CompletableFuture<Integer> g = m.whenComplete
980              (f,
981 <             (Integer x, Throwable t) -> {
981 >             (Integer result, Throwable t) -> {
982                  m.checkExecutionMode();
983 <                threadAssertNull(x);
983 >                threadAssertNull(result);
984                  threadAssertTrue(t instanceof CancellationException);
985                  a.getAndIncrement();
986              });
# Line 1009 | Line 995 | public class CompletableFutureTest exten
995       * If a whenComplete action throws an exception when triggered by
996       * a normal completion, it completes exceptionally
997       */
998 <    public void testWhenComplete_actionFailed() {
998 >    public void testWhenComplete_sourceCompletedNormallyActionFailed() {
999          for (boolean createIncomplete : new boolean[] { true, false })
1000          for (ExecutionMode m : ExecutionMode.values())
1001          for (Integer v1 : new Integer[] { 1, null })
# Line 1020 | Line 1006 | public class CompletableFutureTest exten
1006          if (!createIncomplete) assertTrue(f.complete(v1));
1007          final CompletableFuture<Integer> g = m.whenComplete
1008              (f,
1009 <             (Integer x, Throwable t) -> {
1009 >             (Integer result, Throwable t) -> {
1010                  m.checkExecutionMode();
1011 <                threadAssertSame(x, v1);
1011 >                threadAssertSame(result, v1);
1012                  threadAssertNull(t);
1013                  a.getAndIncrement();
1014                  throw ex;
# Line 1037 | Line 1023 | public class CompletableFutureTest exten
1023      /**
1024       * If a whenComplete action throws an exception when triggered by
1025       * a source completion that also throws an exception, the source
1026 <     * exception takes precedence.
1026 >     * exception takes precedence (unlike handle)
1027       */
1028 <    public void testWhenComplete_actionFailedSourceFailed() {
1028 >    public void testWhenComplete_sourceFailedActionFailed() {
1029          for (boolean createIncomplete : new boolean[] { true, false })
1030          for (ExecutionMode m : ExecutionMode.values())
1045        for (Integer v1 : new Integer[] { 1, null })
1031      {
1032          final AtomicInteger a = new AtomicInteger(0);
1033          final CFException ex1 = new CFException();
# Line 1052 | Line 1037 | public class CompletableFutureTest exten
1037          if (!createIncomplete) f.completeExceptionally(ex1);
1038          final CompletableFuture<Integer> g = m.whenComplete
1039              (f,
1040 <             (Integer x, Throwable t) -> {
1040 >             (Integer result, Throwable t) -> {
1041                  m.checkExecutionMode();
1042                  threadAssertSame(t, ex1);
1043 <                threadAssertNull(x);
1043 >                threadAssertNull(result);
1044                  a.getAndIncrement();
1045                  throw ex2;
1046              });
# Line 1063 | Line 1048 | public class CompletableFutureTest exten
1048  
1049          checkCompletedWithWrappedException(g, ex1);
1050          checkCompletedExceptionally(f, ex1);
1051 +        if (testImplementationDetails) {
1052 +            assertEquals(1, ex1.getSuppressed().length);
1053 +            assertSame(ex2, ex1.getSuppressed()[0]);
1054 +        }
1055          assertEquals(1, a.get());
1056      }}
1057  
# Line 1080 | Line 1069 | public class CompletableFutureTest exten
1069          if (!createIncomplete) assertTrue(f.complete(v1));
1070          final CompletableFuture<Integer> g = m.handle
1071              (f,
1072 <             (Integer x, Throwable t) -> {
1072 >             (Integer result, Throwable t) -> {
1073                  m.checkExecutionMode();
1074 <                threadAssertSame(x, v1);
1074 >                threadAssertSame(result, v1);
1075                  threadAssertNull(t);
1076                  a.getAndIncrement();
1077                  return inc(v1);
# Line 1109 | Line 1098 | public class CompletableFutureTest exten
1098          if (!createIncomplete) f.completeExceptionally(ex);
1099          final CompletableFuture<Integer> g = m.handle
1100              (f,
1101 <             (Integer x, Throwable t) -> {
1101 >             (Integer result, Throwable t) -> {
1102                  m.checkExecutionMode();
1103 <                threadAssertNull(x);
1103 >                threadAssertNull(result);
1104                  threadAssertSame(t, ex);
1105                  a.getAndIncrement();
1106                  return v1;
# Line 1138 | Line 1127 | public class CompletableFutureTest exten
1127          if (!createIncomplete) assertTrue(f.cancel(mayInterruptIfRunning));
1128          final CompletableFuture<Integer> g = m.handle
1129              (f,
1130 <             (Integer x, Throwable t) -> {
1130 >             (Integer result, Throwable t) -> {
1131                  m.checkExecutionMode();
1132 <                threadAssertNull(x);
1132 >                threadAssertNull(result);
1133                  threadAssertTrue(t instanceof CancellationException);
1134                  a.getAndIncrement();
1135                  return v1;
# Line 1153 | Line 1142 | public class CompletableFutureTest exten
1142      }}
1143  
1144      /**
1145 <     * handle result completes exceptionally if action does
1145 >     * If a "handle action" throws an exception when triggered by
1146 >     * a normal completion, it completes exceptionally
1147       */
1148 <    public void testHandle_sourceFailedActionFailed() {
1148 >    public void testHandle_sourceCompletedNormallyActionFailed() {
1149          for (ExecutionMode m : ExecutionMode.values())
1150          for (boolean createIncomplete : new boolean[] { true, false })
1151 +        for (Integer v1 : new Integer[] { 1, null })
1152      {
1153          final CompletableFuture<Integer> f = new CompletableFuture<>();
1154          final AtomicInteger a = new AtomicInteger(0);
1155 <        final CFException ex1 = new CFException();
1156 <        final CFException ex2 = new CFException();
1166 <        if (!createIncomplete) f.completeExceptionally(ex1);
1155 >        final CFException ex = new CFException();
1156 >        if (!createIncomplete) assertTrue(f.complete(v1));
1157          final CompletableFuture<Integer> g = m.handle
1158              (f,
1159 <             (Integer x, Throwable t) -> {
1159 >             (Integer result, Throwable t) -> {
1160                  m.checkExecutionMode();
1161 <                threadAssertNull(x);
1162 <                threadAssertSame(ex1, t);
1161 >                threadAssertSame(result, v1);
1162 >                threadAssertNull(t);
1163                  a.getAndIncrement();
1164 <                throw ex2;
1164 >                throw ex;
1165              });
1166 <        if (createIncomplete) f.completeExceptionally(ex1);
1166 >        if (createIncomplete) assertTrue(f.complete(v1));
1167  
1168 <        checkCompletedWithWrappedException(g, ex2);
1169 <        checkCompletedExceptionally(f, ex1);
1168 >        checkCompletedWithWrappedException(g, ex);
1169 >        checkCompletedNormally(f, v1);
1170          assertEquals(1, a.get());
1171      }}
1172  
1173 <    public void testHandle_sourceCompletedNormallyActionFailed() {
1174 <        for (ExecutionMode m : ExecutionMode.values())
1173 >    /**
1174 >     * If a "handle action" throws an exception when triggered by
1175 >     * a source completion that also throws an exception, the action
1176 >     * exception takes precedence (unlike whenComplete)
1177 >     */
1178 >    public void testHandle_sourceFailedActionFailed() {
1179          for (boolean createIncomplete : new boolean[] { true, false })
1180 <        for (Integer v1 : new Integer[] { 1, null })
1180 >        for (ExecutionMode m : ExecutionMode.values())
1181      {
1188        final CompletableFuture<Integer> f = new CompletableFuture<>();
1182          final AtomicInteger a = new AtomicInteger(0);
1183 <        final CFException ex = new CFException();
1184 <        if (!createIncomplete) assertTrue(f.complete(v1));
1183 >        final CFException ex1 = new CFException();
1184 >        final CFException ex2 = new CFException();
1185 >        final CompletableFuture<Integer> f = new CompletableFuture<>();
1186 >
1187 >        if (!createIncomplete) f.completeExceptionally(ex1);
1188          final CompletableFuture<Integer> g = m.handle
1189              (f,
1190 <             (Integer x, Throwable t) -> {
1190 >             (Integer result, Throwable t) -> {
1191                  m.checkExecutionMode();
1192 <                threadAssertSame(x, v1);
1193 <                threadAssertNull(t);
1192 >                threadAssertNull(result);
1193 >                threadAssertSame(ex1, t);
1194                  a.getAndIncrement();
1195 <                throw ex;
1195 >                throw ex2;
1196              });
1197 <        if (createIncomplete) assertTrue(f.complete(v1));
1197 >        if (createIncomplete) f.completeExceptionally(ex1);
1198  
1199 <        checkCompletedWithWrappedException(g, ex);
1200 <        checkCompletedNormally(f, v1);
1199 >        checkCompletedWithWrappedException(g, ex2);
1200 >        checkCompletedExceptionally(f, ex1);
1201          assertEquals(1, a.get());
1202      }}
1203  
# Line 1234 | Line 1230 | public class CompletableFutureTest exten
1230      {
1231          final FailingRunnable r = new FailingRunnable(m);
1232          final CompletableFuture<Void> f = m.runAsync(r);
1233 <        checkCompletedWithWrappedCFException(f);
1233 >        checkCompletedWithWrappedException(f, r.ex);
1234          r.assertInvoked();
1235      }}
1236  
# Line 1268 | Line 1264 | public class CompletableFutureTest exten
1264      {
1265          FailingSupplier r = new FailingSupplier(m);
1266          CompletableFuture<Integer> f = m.supplyAsync(r);
1267 <        checkCompletedWithWrappedCFException(f);
1267 >        checkCompletedWithWrappedException(f, r.ex);
1268          r.assertInvoked();
1269      }}
1270  
# Line 1390 | Line 1386 | public class CompletableFutureTest exten
1386          final CompletableFuture<Void> h4 = m.runAfterBoth(f, f, rs[4]);
1387          final CompletableFuture<Void> h5 = m.runAfterEither(f, f, rs[5]);
1388  
1389 <        checkCompletedWithWrappedCFException(h0);
1390 <        checkCompletedWithWrappedCFException(h1);
1391 <        checkCompletedWithWrappedCFException(h2);
1392 <        checkCompletedWithWrappedCFException(h3);
1393 <        checkCompletedWithWrappedCFException(h4);
1394 <        checkCompletedWithWrappedCFException(h5);
1389 >        checkCompletedWithWrappedException(h0, rs[0].ex);
1390 >        checkCompletedWithWrappedException(h1, rs[1].ex);
1391 >        checkCompletedWithWrappedException(h2, rs[2].ex);
1392 >        checkCompletedWithWrappedException(h3, rs[3].ex);
1393 >        checkCompletedWithWrappedException(h4, rs[4].ex);
1394 >        checkCompletedWithWrappedException(h5, rs[5].ex);
1395          checkCompletedNormally(f, v1);
1396      }}
1397  
# Line 1494 | Line 1490 | public class CompletableFutureTest exten
1490          final CompletableFuture<Integer> h2 = m.thenApply(f, rs[2]);
1491          final CompletableFuture<Integer> h3 = m.applyToEither(f, f, rs[3]);
1492  
1493 <        checkCompletedWithWrappedCFException(h0);
1494 <        checkCompletedWithWrappedCFException(h1);
1495 <        checkCompletedWithWrappedCFException(h2);
1496 <        checkCompletedWithWrappedCFException(h3);
1493 >        checkCompletedWithWrappedException(h0, rs[0].ex);
1494 >        checkCompletedWithWrappedException(h1, rs[1].ex);
1495 >        checkCompletedWithWrappedException(h2, rs[2].ex);
1496 >        checkCompletedWithWrappedException(h3, rs[3].ex);
1497          checkCompletedNormally(f, v1);
1498      }}
1499  
# Line 1596 | Line 1592 | public class CompletableFutureTest exten
1592          final CompletableFuture<Void> h2 = m.thenAccept(f, rs[2]);
1593          final CompletableFuture<Void> h3 = m.acceptEither(f, f, rs[3]);
1594  
1595 <        checkCompletedWithWrappedCFException(h0);
1596 <        checkCompletedWithWrappedCFException(h1);
1597 <        checkCompletedWithWrappedCFException(h2);
1598 <        checkCompletedWithWrappedCFException(h3);
1595 >        checkCompletedWithWrappedException(h0, rs[0].ex);
1596 >        checkCompletedWithWrappedException(h1, rs[1].ex);
1597 >        checkCompletedWithWrappedException(h2, rs[2].ex);
1598 >        checkCompletedWithWrappedException(h3, rs[3].ex);
1599          checkCompletedNormally(f, v1);
1600      }}
1601  
# Line 1761 | Line 1757 | public class CompletableFutureTest exten
1757          assertTrue(snd.complete(w2));
1758          final CompletableFuture<Integer> h3 = m.thenCombine(f, g, r3);
1759  
1760 <        checkCompletedWithWrappedCFException(h1);
1761 <        checkCompletedWithWrappedCFException(h2);
1762 <        checkCompletedWithWrappedCFException(h3);
1760 >        checkCompletedWithWrappedException(h1, r1.ex);
1761 >        checkCompletedWithWrappedException(h2, r2.ex);
1762 >        checkCompletedWithWrappedException(h3, r3.ex);
1763          r1.assertInvoked();
1764          r2.assertInvoked();
1765          r3.assertInvoked();
# Line 1925 | Line 1921 | public class CompletableFutureTest exten
1921          assertTrue(snd.complete(w2));
1922          final CompletableFuture<Void> h3 = m.thenAcceptBoth(f, g, r3);
1923  
1924 <        checkCompletedWithWrappedCFException(h1);
1925 <        checkCompletedWithWrappedCFException(h2);
1926 <        checkCompletedWithWrappedCFException(h3);
1924 >        checkCompletedWithWrappedException(h1, r1.ex);
1925 >        checkCompletedWithWrappedException(h2, r2.ex);
1926 >        checkCompletedWithWrappedException(h3, r3.ex);
1927          r1.assertInvoked();
1928          r2.assertInvoked();
1929          r3.assertInvoked();
# Line 2089 | Line 2085 | public class CompletableFutureTest exten
2085          assertTrue(snd.complete(w2));
2086          final CompletableFuture<Void> h3 = m.runAfterBoth(f, g, r3);
2087  
2088 <        checkCompletedWithWrappedCFException(h1);
2089 <        checkCompletedWithWrappedCFException(h2);
2090 <        checkCompletedWithWrappedCFException(h3);
2088 >        checkCompletedWithWrappedException(h1, r1.ex);
2089 >        checkCompletedWithWrappedException(h2, r2.ex);
2090 >        checkCompletedWithWrappedException(h3, r3.ex);
2091          r1.assertInvoked();
2092          r2.assertInvoked();
2093          r3.assertInvoked();
# Line 2381 | Line 2377 | public class CompletableFutureTest exten
2377          f.complete(v1);
2378          final CompletableFuture<Integer> h2 = m.applyToEither(f, g, rs[2]);
2379          final CompletableFuture<Integer> h3 = m.applyToEither(g, f, rs[3]);
2380 <        checkCompletedWithWrappedCFException(h0);
2381 <        checkCompletedWithWrappedCFException(h1);
2382 <        checkCompletedWithWrappedCFException(h2);
2383 <        checkCompletedWithWrappedCFException(h3);
2380 >        checkCompletedWithWrappedException(h0, rs[0].ex);
2381 >        checkCompletedWithWrappedException(h1, rs[1].ex);
2382 >        checkCompletedWithWrappedException(h2, rs[2].ex);
2383 >        checkCompletedWithWrappedException(h3, rs[3].ex);
2384          for (int i = 0; i < 4; i++) rs[i].assertValue(v1);
2385  
2386          g.complete(v2);
# Line 2393 | Line 2389 | public class CompletableFutureTest exten
2389          final CompletableFuture<Integer> h4 = m.applyToEither(f, g, rs[4]);
2390          final CompletableFuture<Integer> h5 = m.applyToEither(g, f, rs[5]);
2391  
2392 <        checkCompletedWithWrappedCFException(h4);
2392 >        checkCompletedWithWrappedException(h4, rs[4].ex);
2393          assertTrue(Objects.equals(v1, rs[4].value) ||
2394                     Objects.equals(v2, rs[4].value));
2395 <        checkCompletedWithWrappedCFException(h5);
2395 >        checkCompletedWithWrappedException(h5, rs[5].ex);
2396          assertTrue(Objects.equals(v1, rs[5].value) ||
2397                     Objects.equals(v2, rs[5].value));
2398  
# Line 2640 | Line 2636 | public class CompletableFutureTest exten
2636          f.complete(v1);
2637          final CompletableFuture<Void> h2 = m.acceptEither(f, g, rs[2]);
2638          final CompletableFuture<Void> h3 = m.acceptEither(g, f, rs[3]);
2639 <        checkCompletedWithWrappedCFException(h0);
2640 <        checkCompletedWithWrappedCFException(h1);
2641 <        checkCompletedWithWrappedCFException(h2);
2642 <        checkCompletedWithWrappedCFException(h3);
2639 >        checkCompletedWithWrappedException(h0, rs[0].ex);
2640 >        checkCompletedWithWrappedException(h1, rs[1].ex);
2641 >        checkCompletedWithWrappedException(h2, rs[2].ex);
2642 >        checkCompletedWithWrappedException(h3, rs[3].ex);
2643          for (int i = 0; i < 4; i++) rs[i].assertValue(v1);
2644  
2645          g.complete(v2);
# Line 2652 | Line 2648 | public class CompletableFutureTest exten
2648          final CompletableFuture<Void> h4 = m.acceptEither(f, g, rs[4]);
2649          final CompletableFuture<Void> h5 = m.acceptEither(g, f, rs[5]);
2650  
2651 <        checkCompletedWithWrappedCFException(h4);
2651 >        checkCompletedWithWrappedException(h4, rs[4].ex);
2652          assertTrue(Objects.equals(v1, rs[4].value) ||
2653                     Objects.equals(v2, rs[4].value));
2654 <        checkCompletedWithWrappedCFException(h5);
2654 >        checkCompletedWithWrappedException(h5, rs[5].ex);
2655          assertTrue(Objects.equals(v1, rs[5].value) ||
2656                     Objects.equals(v2, rs[5].value));
2657  
# Line 2895 | Line 2891 | public class CompletableFutureTest exten
2891          assertTrue(f.complete(v1));
2892          final CompletableFuture<Void> h2 = m.runAfterEither(f, g, rs[2]);
2893          final CompletableFuture<Void> h3 = m.runAfterEither(g, f, rs[3]);
2894 <        checkCompletedWithWrappedCFException(h0);
2895 <        checkCompletedWithWrappedCFException(h1);
2896 <        checkCompletedWithWrappedCFException(h2);
2897 <        checkCompletedWithWrappedCFException(h3);
2894 >        checkCompletedWithWrappedException(h0, rs[0].ex);
2895 >        checkCompletedWithWrappedException(h1, rs[1].ex);
2896 >        checkCompletedWithWrappedException(h2, rs[2].ex);
2897 >        checkCompletedWithWrappedException(h3, rs[3].ex);
2898          for (int i = 0; i < 4; i++) rs[i].assertInvoked();
2899          assertTrue(g.complete(v2));
2900          final CompletableFuture<Void> h4 = m.runAfterEither(f, g, rs[4]);
2901          final CompletableFuture<Void> h5 = m.runAfterEither(g, f, rs[5]);
2902 <        checkCompletedWithWrappedCFException(h4);
2903 <        checkCompletedWithWrappedCFException(h5);
2902 >        checkCompletedWithWrappedException(h4, rs[4].ex);
2903 >        checkCompletedWithWrappedException(h5, rs[5].ex);
2904  
2905          checkCompletedNormally(f, v1);
2906          checkCompletedNormally(g, v2);
# Line 2965 | Line 2961 | public class CompletableFutureTest exten
2961          final CompletableFuture<Integer> g = m.thenCompose(f, r);
2962          if (createIncomplete) assertTrue(f.complete(v1));
2963  
2964 <        checkCompletedWithWrappedCFException(g);
2964 >        checkCompletedWithWrappedException(g, r.ex);
2965          checkCompletedNormally(f, v1);
2966      }}
2967  
# Line 2990 | Line 2986 | public class CompletableFutureTest exten
2986          checkCancelled(f);
2987      }}
2988  
2989 +    /**
2990 +     * thenCompose result completes exceptionally if the result of the action does
2991 +     */
2992 +    public void testThenCompose_actionReturnsFailingFuture() {
2993 +        for (ExecutionMode m : ExecutionMode.values())
2994 +        for (int order = 0; order < 6; order++)
2995 +        for (Integer v1 : new Integer[] { 1, null })
2996 +    {
2997 +        final CFException ex = new CFException();
2998 +        final CompletableFuture<Integer> f = new CompletableFuture<>();
2999 +        final CompletableFuture<Integer> g = new CompletableFuture<>();
3000 +        final CompletableFuture<Integer> h;
3001 +        // Test all permutations of orders
3002 +        switch (order) {
3003 +        case 0:
3004 +            assertTrue(f.complete(v1));
3005 +            assertTrue(g.completeExceptionally(ex));
3006 +            h = m.thenCompose(f, (x -> g));
3007 +            break;
3008 +        case 1:
3009 +            assertTrue(f.complete(v1));
3010 +            h = m.thenCompose(f, (x -> g));
3011 +            assertTrue(g.completeExceptionally(ex));
3012 +            break;
3013 +        case 2:
3014 +            assertTrue(g.completeExceptionally(ex));
3015 +            assertTrue(f.complete(v1));
3016 +            h = m.thenCompose(f, (x -> g));
3017 +            break;
3018 +        case 3:
3019 +            assertTrue(g.completeExceptionally(ex));
3020 +            h = m.thenCompose(f, (x -> g));
3021 +            assertTrue(f.complete(v1));
3022 +            break;
3023 +        case 4:
3024 +            h = m.thenCompose(f, (x -> g));
3025 +            assertTrue(f.complete(v1));
3026 +            assertTrue(g.completeExceptionally(ex));
3027 +            break;
3028 +        case 5:
3029 +            h = m.thenCompose(f, (x -> g));
3030 +            assertTrue(f.complete(v1));
3031 +            assertTrue(g.completeExceptionally(ex));
3032 +            break;
3033 +        default: throw new AssertionError();
3034 +        }
3035 +
3036 +        checkCompletedExceptionally(g, ex);
3037 +        checkCompletedWithWrappedException(h, ex);
3038 +        checkCompletedNormally(f, v1);
3039 +    }}
3040 +
3041      // other static methods
3042  
3043      /**
# Line 3022 | Line 3070 | public class CompletableFutureTest exten
3070          }
3071      }
3072  
3073 <    public void testAllOf_backwards() throws Exception {
3073 >    public void testAllOf_normal_backwards() throws Exception {
3074          for (int k = 1; k < 10; k++) {
3075              CompletableFuture<Integer>[] fs
3076                  = (CompletableFuture<Integer>[]) new CompletableFuture[k];
# Line 3050 | Line 3098 | public class CompletableFutureTest exten
3098              for (int i = 0; i < k; i++) {
3099                  checkIncomplete(f);
3100                  checkIncomplete(CompletableFuture.allOf(fs));
3101 <                if (i != k/2) {
3101 >                if (i != k / 2) {
3102                      fs[i].complete(i);
3103                      checkCompletedNormally(fs[i], i);
3104                  } else {
# Line 3157 | Line 3205 | public class CompletableFutureTest exten
3205          CompletableFuture<Integer> f = new CompletableFuture<>();
3206          CompletableFuture<Integer> g = new CompletableFuture<>();
3207          CompletableFuture<Integer> nullFuture = (CompletableFuture<Integer>)null;
3160        CompletableFuture<?> h;
3208          ThreadExecutor exec = new ThreadExecutor();
3209  
3210          Runnable[] throwingActions = {
# Line 3254 | Line 3301 | public class CompletableFutureTest exten
3301              () -> CompletableFuture.anyOf(null, f),
3302  
3303              () -> f.obtrudeException(null),
3304 +
3305 +            () -> CompletableFuture.delayedExecutor(1L, SECONDS, null),
3306 +            () -> CompletableFuture.delayedExecutor(1L, null, exec),
3307 +            () -> CompletableFuture.delayedExecutor(1L, null),
3308 +
3309 +            () -> f.orTimeout(1L, null),
3310 +            () -> f.completeOnTimeout(42, 1L, null),
3311 +
3312 +            () -> CompletableFuture.failedFuture(null),
3313 +            () -> CompletableFuture.failedStage(null),
3314          };
3315  
3316          assertThrows(NullPointerException.class, throwingActions);
# Line 3261 | Line 3318 | public class CompletableFutureTest exten
3318      }
3319  
3320      /**
3321 +     * Test submissions to an executor that rejects all tasks.
3322 +     */
3323 +    public void testRejectingExecutor() {
3324 +        final RejectedExecutionException ex = new RejectedExecutionException();
3325 +        final Executor e = (Runnable r) -> { throw ex; };
3326 +
3327 +        for (Integer v : new Integer[] { 1, null }) {
3328 +
3329 +        final CompletableFuture<Integer> complete = CompletableFuture.completedFuture(v);
3330 +        final CompletableFuture<Integer> incomplete = new CompletableFuture<>();
3331 +
3332 +        List<CompletableFuture<?>> futures = new ArrayList<>();
3333 +
3334 +        List<CompletableFuture<Integer>> srcs = new ArrayList<>();
3335 +        srcs.add(complete);
3336 +        srcs.add(incomplete);
3337 +
3338 +        for (CompletableFuture<Integer> src : srcs) {
3339 +            List<CompletableFuture<?>> fs = new ArrayList<>();
3340 +            fs.add(src.thenRunAsync(() -> {}, e));
3341 +            fs.add(src.thenAcceptAsync((z) -> {}, e));
3342 +            fs.add(src.thenApplyAsync((z) -> z, e));
3343 +
3344 +            fs.add(src.thenCombineAsync(src, (x, y) -> x, e));
3345 +            fs.add(src.thenAcceptBothAsync(src, (x, y) -> {}, e));
3346 +            fs.add(src.runAfterBothAsync(src, () -> {}, e));
3347 +
3348 +            fs.add(src.applyToEitherAsync(src, (z) -> z, e));
3349 +            fs.add(src.acceptEitherAsync(src, (z) -> {}, e));
3350 +            fs.add(src.runAfterEitherAsync(src, () -> {}, e));
3351 +
3352 +            fs.add(src.thenComposeAsync((z) -> null, e));
3353 +            fs.add(src.whenCompleteAsync((z, t) -> {}, e));
3354 +            fs.add(src.handleAsync((z, t) -> null, e));
3355 +
3356 +            for (CompletableFuture<?> future : fs) {
3357 +                if (src.isDone())
3358 +                    checkCompletedWithWrappedException(future, ex);
3359 +                else
3360 +                    checkIncomplete(future);
3361 +            }
3362 +            futures.addAll(fs);
3363 +        }
3364 +
3365 +        {
3366 +            List<CompletableFuture<?>> fs = new ArrayList<>();
3367 +
3368 +            fs.add(complete.thenCombineAsync(incomplete, (x, y) -> x, e));
3369 +            fs.add(incomplete.thenCombineAsync(complete, (x, y) -> x, e));
3370 +
3371 +            fs.add(complete.thenAcceptBothAsync(incomplete, (x, y) -> {}, e));
3372 +            fs.add(incomplete.thenAcceptBothAsync(complete, (x, y) -> {}, e));
3373 +
3374 +            fs.add(complete.runAfterBothAsync(incomplete, () -> {}, e));
3375 +            fs.add(incomplete.runAfterBothAsync(complete, () -> {}, e));
3376 +
3377 +            for (CompletableFuture<?> future : fs)
3378 +                checkIncomplete(future);
3379 +            futures.addAll(fs);
3380 +        }
3381 +
3382 +        {
3383 +            List<CompletableFuture<?>> fs = new ArrayList<>();
3384 +
3385 +            fs.add(complete.applyToEitherAsync(incomplete, (z) -> z, e));
3386 +            fs.add(incomplete.applyToEitherAsync(complete, (z) -> z, e));
3387 +
3388 +            fs.add(complete.acceptEitherAsync(incomplete, (z) -> {}, e));
3389 +            fs.add(incomplete.acceptEitherAsync(complete, (z) -> {}, e));
3390 +
3391 +            fs.add(complete.runAfterEitherAsync(incomplete, () -> {}, e));
3392 +            fs.add(incomplete.runAfterEitherAsync(complete, () -> {}, e));
3393 +
3394 +            for (CompletableFuture<?> future : fs)
3395 +                checkCompletedWithWrappedException(future, ex);
3396 +            futures.addAll(fs);
3397 +        }
3398 +
3399 +        incomplete.complete(v);
3400 +
3401 +        for (CompletableFuture<?> future : futures)
3402 +            checkCompletedWithWrappedException(future, ex);
3403 +        }
3404 +    }
3405 +
3406 +    /**
3407 +     * Test submissions to an executor that rejects all tasks, but
3408 +     * should never be invoked because the dependent future is
3409 +     * explicitly completed.
3410 +     */
3411 +    public void testRejectingExecutorNeverInvoked() {
3412 +        final RejectedExecutionException ex = new RejectedExecutionException();
3413 +        class CountingRejectingExecutor implements Executor {
3414 +            final AtomicInteger count = new AtomicInteger(0);
3415 +            public void execute(Runnable r) {
3416 +                count.getAndIncrement();
3417 +                throw ex;
3418 +            }
3419 +        }
3420 +        final CountingRejectingExecutor e = new CountingRejectingExecutor();
3421 +
3422 +        for (Integer v : new Integer[] { 1, null }) {
3423 +
3424 +        final CompletableFuture<Integer> complete = CompletableFuture.completedFuture(v);
3425 +        final CompletableFuture<Integer> incomplete = new CompletableFuture<>();
3426 +
3427 +        List<CompletableFuture<?>> futures = new ArrayList<>();
3428 +
3429 +        List<CompletableFuture<Integer>> srcs = new ArrayList<>();
3430 +        srcs.add(complete);
3431 +        srcs.add(incomplete);
3432 +
3433 +        List<CompletableFuture<?>> fs = new ArrayList<>();
3434 +        fs.add(incomplete.thenRunAsync(() -> {}, e));
3435 +        fs.add(incomplete.thenAcceptAsync((z) -> {}, e));
3436 +        fs.add(incomplete.thenApplyAsync((z) -> z, e));
3437 +
3438 +        fs.add(incomplete.thenCombineAsync(incomplete, (x, y) -> x, e));
3439 +        fs.add(incomplete.thenAcceptBothAsync(incomplete, (x, y) -> {}, e));
3440 +        fs.add(incomplete.runAfterBothAsync(incomplete, () -> {}, e));
3441 +
3442 +        fs.add(incomplete.applyToEitherAsync(incomplete, (z) -> z, e));
3443 +        fs.add(incomplete.acceptEitherAsync(incomplete, (z) -> {}, e));
3444 +        fs.add(incomplete.runAfterEitherAsync(incomplete, () -> {}, e));
3445 +
3446 +        fs.add(incomplete.thenComposeAsync((z) -> null, e));
3447 +        fs.add(incomplete.whenCompleteAsync((z, t) -> {}, e));
3448 +        fs.add(incomplete.handleAsync((z, t) -> null, e));
3449 +
3450 +        fs.add(complete.thenCombineAsync(incomplete, (x, y) -> x, e));
3451 +        fs.add(incomplete.thenCombineAsync(complete, (x, y) -> x, e));
3452 +
3453 +        fs.add(complete.thenAcceptBothAsync(incomplete, (x, y) -> {}, e));
3454 +        fs.add(incomplete.thenAcceptBothAsync(complete, (x, y) -> {}, e));
3455 +
3456 +        fs.add(complete.runAfterBothAsync(incomplete, () -> {}, e));
3457 +        fs.add(incomplete.runAfterBothAsync(complete, () -> {}, e));
3458 +
3459 +        for (CompletableFuture<?> future : fs)
3460 +            checkIncomplete(future);
3461 +
3462 +        for (CompletableFuture<?> future : fs)
3463 +            future.complete(null);
3464 +
3465 +        incomplete.complete(v);
3466 +
3467 +        for (CompletableFuture<?> future : fs)
3468 +            checkCompletedNormally(future, null);
3469 +
3470 +        assertEquals(0, e.count.get());
3471 +
3472 +        }
3473 +    }
3474 +
3475 +    /**
3476       * toCompletableFuture returns this CompletableFuture.
3477       */
3478      public void testToCompletableFuture() {
# Line 3268 | Line 3480 | public class CompletableFutureTest exten
3480          assertSame(f, f.toCompletableFuture());
3481      }
3482  
3483 +    // jdk9
3484 +
3485 +    /**
3486 +     * newIncompleteFuture returns an incomplete CompletableFuture
3487 +     */
3488 +    public void testNewIncompleteFuture() {
3489 +        for (Integer v1 : new Integer[] { 1, null })
3490 +    {
3491 +        CompletableFuture<Integer> f = new CompletableFuture<>();
3492 +        CompletableFuture<Integer> g = f.newIncompleteFuture();
3493 +        checkIncomplete(f);
3494 +        checkIncomplete(g);
3495 +        f.complete(v1);
3496 +        checkCompletedNormally(f, v1);
3497 +        checkIncomplete(g);
3498 +        g.complete(v1);
3499 +        checkCompletedNormally(g, v1);
3500 +        assertSame(g.getClass(), CompletableFuture.class);
3501 +    }}
3502 +
3503 +    /**
3504 +     * completedStage returns a completed CompletionStage
3505 +     */
3506 +    public void testCompletedStage() {
3507 +        AtomicInteger x = new AtomicInteger(0);
3508 +        AtomicReference<Throwable> r = new AtomicReference<Throwable>();
3509 +        CompletionStage<Integer> f = CompletableFuture.completedStage(1);
3510 +        f.whenComplete((v, e) -> {if (e != null) r.set(e); else x.set(v);});
3511 +        assertEquals(x.get(), 1);
3512 +        assertNull(r.get());
3513 +    }
3514 +
3515 +    /**
3516 +     * defaultExecutor by default returns the commonPool if
3517 +     * it supports more than one thread.
3518 +     */
3519 +    public void testDefaultExecutor() {
3520 +        CompletableFuture<Integer> f = new CompletableFuture<>();
3521 +        Executor e = f.defaultExecutor();
3522 +        Executor c = ForkJoinPool.commonPool();
3523 +        if (ForkJoinPool.getCommonPoolParallelism() > 1)
3524 +            assertSame(e, c);
3525 +        else
3526 +            assertNotSame(e, c);
3527 +    }
3528 +
3529 +    /**
3530 +     * failedFuture returns a CompletableFuture completed
3531 +     * exceptionally with the given Exception
3532 +     */
3533 +    public void testFailedFuture() {
3534 +        CFException ex = new CFException();
3535 +        CompletableFuture<Integer> f = CompletableFuture.failedFuture(ex);
3536 +        checkCompletedExceptionally(f, ex);
3537 +    }
3538 +
3539 +    /**
3540 +     * failedFuture(null) throws NPE
3541 +     */
3542 +    public void testFailedFuture_null() {
3543 +        try {
3544 +            CompletableFuture<Integer> f = CompletableFuture.failedFuture(null);
3545 +            shouldThrow();
3546 +        } catch (NullPointerException success) {}
3547 +    }
3548 +
3549 +    /**
3550 +     * copy returns a CompletableFuture that is completed normally,
3551 +     * with the same value, when source is.
3552 +     */
3553 +    public void testCopy() {
3554 +        CompletableFuture<Integer> f = new CompletableFuture<>();
3555 +        CompletableFuture<Integer> g = f.copy();
3556 +        checkIncomplete(f);
3557 +        checkIncomplete(g);
3558 +        f.complete(1);
3559 +        checkCompletedNormally(f, 1);
3560 +        checkCompletedNormally(g, 1);
3561 +    }
3562 +
3563 +    /**
3564 +     * copy returns a CompletableFuture that is completed exceptionally
3565 +     * when source is.
3566 +     */
3567 +    public void testCopy2() {
3568 +        CompletableFuture<Integer> f = new CompletableFuture<>();
3569 +        CompletableFuture<Integer> g = f.copy();
3570 +        checkIncomplete(f);
3571 +        checkIncomplete(g);
3572 +        CFException ex = new CFException();
3573 +        f.completeExceptionally(ex);
3574 +        checkCompletedExceptionally(f, ex);
3575 +        checkCompletedWithWrappedException(g, ex);
3576 +    }
3577 +
3578 +    /**
3579 +     * minimalCompletionStage returns a CompletableFuture that is
3580 +     * completed normally, with the same value, when source is.
3581 +     */
3582 +    public void testMinimalCompletionStage() {
3583 +        CompletableFuture<Integer> f = new CompletableFuture<>();
3584 +        CompletionStage<Integer> g = f.minimalCompletionStage();
3585 +        AtomicInteger x = new AtomicInteger(0);
3586 +        AtomicReference<Throwable> r = new AtomicReference<Throwable>();
3587 +        checkIncomplete(f);
3588 +        g.whenComplete((v, e) -> {if (e != null) r.set(e); else x.set(v);});
3589 +        f.complete(1);
3590 +        checkCompletedNormally(f, 1);
3591 +        assertEquals(x.get(), 1);
3592 +        assertNull(r.get());
3593 +    }
3594 +
3595 +    /**
3596 +     * minimalCompletionStage returns a CompletableFuture that is
3597 +     * completed exceptionally when source is.
3598 +     */
3599 +    public void testMinimalCompletionStage2() {
3600 +        CompletableFuture<Integer> f = new CompletableFuture<>();
3601 +        CompletionStage<Integer> g = f.minimalCompletionStage();
3602 +        AtomicInteger x = new AtomicInteger(0);
3603 +        AtomicReference<Throwable> r = new AtomicReference<Throwable>();
3604 +        g.whenComplete((v, e) -> {if (e != null) r.set(e); else x.set(v);});
3605 +        checkIncomplete(f);
3606 +        CFException ex = new CFException();
3607 +        f.completeExceptionally(ex);
3608 +        checkCompletedExceptionally(f, ex);
3609 +        assertEquals(x.get(), 0);
3610 +        assertEquals(r.get().getCause(), ex);
3611 +    }
3612 +
3613 +    /**
3614 +     * failedStage returns a CompletionStage completed
3615 +     * exceptionally with the given Exception
3616 +     */
3617 +    public void testFailedStage() {
3618 +        CFException ex = new CFException();
3619 +        CompletionStage<Integer> f = CompletableFuture.failedStage(ex);
3620 +        AtomicInteger x = new AtomicInteger(0);
3621 +        AtomicReference<Throwable> r = new AtomicReference<Throwable>();
3622 +        f.whenComplete((v, e) -> {if (e != null) r.set(e); else x.set(v);});
3623 +        assertEquals(x.get(), 0);
3624 +        assertEquals(r.get(), ex);
3625 +    }
3626 +
3627 +    /**
3628 +     * completeAsync completes with value of given supplier
3629 +     */
3630 +    public void testCompleteAsync() {
3631 +        for (Integer v1 : new Integer[] { 1, null })
3632 +    {
3633 +        CompletableFuture<Integer> f = new CompletableFuture<>();
3634 +        f.completeAsync(() -> v1);
3635 +        f.join();
3636 +        checkCompletedNormally(f, v1);
3637 +    }}
3638 +
3639 +    /**
3640 +     * completeAsync completes exceptionally if given supplier throws
3641 +     */
3642 +    public void testCompleteAsync2() {
3643 +        CompletableFuture<Integer> f = new CompletableFuture<>();
3644 +        CFException ex = new CFException();
3645 +        f.completeAsync(() -> {if (true) throw ex; return 1;});
3646 +        try {
3647 +            f.join();
3648 +            shouldThrow();
3649 +        } catch (CompletionException success) {}
3650 +        checkCompletedWithWrappedException(f, ex);
3651 +    }
3652 +
3653 +    /**
3654 +     * completeAsync with given executor completes with value of given supplier
3655 +     */
3656 +    public void testCompleteAsync3() {
3657 +        for (Integer v1 : new Integer[] { 1, null })
3658 +    {
3659 +        CompletableFuture<Integer> f = new CompletableFuture<>();
3660 +        ThreadExecutor executor = new ThreadExecutor();
3661 +        f.completeAsync(() -> v1, executor);
3662 +        assertSame(v1, f.join());
3663 +        checkCompletedNormally(f, v1);
3664 +        assertEquals(1, executor.count.get());
3665 +    }}
3666 +
3667 +    /**
3668 +     * completeAsync with given executor completes exceptionally if
3669 +     * given supplier throws
3670 +     */
3671 +    public void testCompleteAsync4() {
3672 +        CompletableFuture<Integer> f = new CompletableFuture<>();
3673 +        CFException ex = new CFException();
3674 +        ThreadExecutor executor = new ThreadExecutor();
3675 +        f.completeAsync(() -> {if (true) throw ex; return 1;}, executor);
3676 +        try {
3677 +            f.join();
3678 +            shouldThrow();
3679 +        } catch (CompletionException success) {}
3680 +        checkCompletedWithWrappedException(f, ex);
3681 +        assertEquals(1, executor.count.get());
3682 +    }
3683 +
3684 +    /**
3685 +     * orTimeout completes with TimeoutException if not complete
3686 +     */
3687 +    public void testOrTimeout_timesOut() {
3688 +        long timeoutMillis = timeoutMillis();
3689 +        CompletableFuture<Integer> f = new CompletableFuture<>();
3690 +        long startTime = System.nanoTime();
3691 +        assertSame(f, f.orTimeout(timeoutMillis, MILLISECONDS));
3692 +        checkCompletedWithTimeoutException(f);
3693 +        assertTrue(millisElapsedSince(startTime) >= timeoutMillis);
3694 +    }
3695 +
3696 +    /**
3697 +     * orTimeout completes normally if completed before timeout
3698 +     */
3699 +    public void testOrTimeout_completed() {
3700 +        for (Integer v1 : new Integer[] { 1, null })
3701 +    {
3702 +        CompletableFuture<Integer> f = new CompletableFuture<>();
3703 +        CompletableFuture<Integer> g = new CompletableFuture<>();
3704 +        long startTime = System.nanoTime();
3705 +        f.complete(v1);
3706 +        assertSame(f, f.orTimeout(LONG_DELAY_MS, MILLISECONDS));
3707 +        assertSame(g, g.orTimeout(LONG_DELAY_MS, MILLISECONDS));
3708 +        g.complete(v1);
3709 +        checkCompletedNormally(f, v1);
3710 +        checkCompletedNormally(g, v1);
3711 +        assertTrue(millisElapsedSince(startTime) < LONG_DELAY_MS / 2);
3712 +    }}
3713 +
3714 +    /**
3715 +     * completeOnTimeout completes with given value if not complete
3716 +     */
3717 +    public void testCompleteOnTimeout_timesOut() {
3718 +        testInParallel(() -> testCompleteOnTimeout_timesOut(42),
3719 +                       () -> testCompleteOnTimeout_timesOut(null));
3720 +    }
3721 +
3722 +    /**
3723 +     * completeOnTimeout completes with given value if not complete
3724 +     */
3725 +    public void testCompleteOnTimeout_timesOut(Integer v) {
3726 +        long timeoutMillis = timeoutMillis();
3727 +        CompletableFuture<Integer> f = new CompletableFuture<>();
3728 +        long startTime = System.nanoTime();
3729 +        assertSame(f, f.completeOnTimeout(v, timeoutMillis, MILLISECONDS));
3730 +        assertSame(v, f.join());
3731 +        assertTrue(millisElapsedSince(startTime) >= timeoutMillis);
3732 +        f.complete(99);         // should have no effect
3733 +        checkCompletedNormally(f, v);
3734 +    }
3735 +
3736 +    /**
3737 +     * completeOnTimeout has no effect if completed within timeout
3738 +     */
3739 +    public void testCompleteOnTimeout_completed() {
3740 +        for (Integer v1 : new Integer[] { 1, null })
3741 +    {
3742 +        CompletableFuture<Integer> f = new CompletableFuture<>();
3743 +        CompletableFuture<Integer> g = new CompletableFuture<>();
3744 +        long startTime = System.nanoTime();
3745 +        f.complete(v1);
3746 +        assertSame(f, f.completeOnTimeout(-1, LONG_DELAY_MS, MILLISECONDS));
3747 +        assertSame(g, g.completeOnTimeout(-1, LONG_DELAY_MS, MILLISECONDS));
3748 +        g.complete(v1);
3749 +        checkCompletedNormally(f, v1);
3750 +        checkCompletedNormally(g, v1);
3751 +        assertTrue(millisElapsedSince(startTime) < LONG_DELAY_MS / 2);
3752 +    }}
3753 +
3754 +    /**
3755 +     * delayedExecutor returns an executor that delays submission
3756 +     */
3757 +    public void testDelayedExecutor() {
3758 +        testInParallel(() -> testDelayedExecutor(null, null),
3759 +                       () -> testDelayedExecutor(null, 1),
3760 +                       () -> testDelayedExecutor(new ThreadExecutor(), 1),
3761 +                       () -> testDelayedExecutor(new ThreadExecutor(), 1));
3762 +    }
3763 +
3764 +    public void testDelayedExecutor(Executor executor, Integer v) throws Exception {
3765 +        long timeoutMillis = timeoutMillis();
3766 +        // Use an "unreasonably long" long timeout to catch lingering threads
3767 +        long longTimeoutMillis = 1000 * 60 * 60 * 24;
3768 +        final Executor delayer, longDelayer;
3769 +        if (executor == null) {
3770 +            delayer = CompletableFuture.delayedExecutor(timeoutMillis, MILLISECONDS);
3771 +            longDelayer = CompletableFuture.delayedExecutor(longTimeoutMillis, MILLISECONDS);
3772 +        } else {
3773 +            delayer = CompletableFuture.delayedExecutor(timeoutMillis, MILLISECONDS, executor);
3774 +            longDelayer = CompletableFuture.delayedExecutor(longTimeoutMillis, MILLISECONDS, executor);
3775 +        }
3776 +        long startTime = System.nanoTime();
3777 +        CompletableFuture<Integer> f =
3778 +            CompletableFuture.supplyAsync(() -> v, delayer);
3779 +        CompletableFuture<Integer> g =
3780 +            CompletableFuture.supplyAsync(() -> v, longDelayer);
3781 +
3782 +        assertNull(g.getNow(null));
3783 +
3784 +        assertSame(v, f.get(LONG_DELAY_MS, MILLISECONDS));
3785 +        long millisElapsed = millisElapsedSince(startTime);
3786 +        assertTrue(millisElapsed >= timeoutMillis);
3787 +        assertTrue(millisElapsed < LONG_DELAY_MS / 2);
3788 +
3789 +        checkCompletedNormally(f, v);
3790 +
3791 +        checkIncomplete(g);
3792 +        assertTrue(g.cancel(true));
3793 +    }
3794 +
3795      //--- tests of implementation details; not part of official tck ---
3796  
3797      Object resultOf(CompletableFuture<?> f) {
3798 +        SecurityManager sm = System.getSecurityManager();
3799 +        if (sm != null) {
3800 +            try {
3801 +                System.setSecurityManager(null);
3802 +            } catch (SecurityException giveUp) {
3803 +                return "Reflection not available";
3804 +            }
3805 +        }
3806 +
3807          try {
3808              java.lang.reflect.Field resultField
3809                  = CompletableFuture.class.getDeclaredField("result");
3810              resultField.setAccessible(true);
3811              return resultField.get(f);
3812 <        } catch (Throwable t) { throw new AssertionError(t); }
3812 >        } catch (Throwable t) {
3813 >            throw new AssertionError(t);
3814 >        } finally {
3815 >            if (sm != null) System.setSecurityManager(sm);
3816 >        }
3817      }
3818  
3819      public void testExceptionPropagationReusesResultObject() {
# Line 3299 | Line 3836 | public class CompletableFutureTest exten
3836          funs.add((y) -> m.applyToEither(y, incomplete, new IncFunction(m)));
3837  
3838          funs.add((y) -> m.runAfterBoth(y, v42, new Noop(m)));
3839 +        funs.add((y) -> m.runAfterBoth(v42, y, new Noop(m)));
3840          funs.add((y) -> m.thenAcceptBoth(y, v42, new SubtractAction(m)));
3841 +        funs.add((y) -> m.thenAcceptBoth(v42, y, new SubtractAction(m)));
3842          funs.add((y) -> m.thenCombine(y, v42, new SubtractFunction(m)));
3843 +        funs.add((y) -> m.thenCombine(v42, y, new SubtractFunction(m)));
3844  
3845 <        funs.add((y) -> m.whenComplete(y, (Integer x, Throwable t) -> {}));
3845 >        funs.add((y) -> m.whenComplete(y, (Integer r, Throwable t) -> {}));
3846  
3847          funs.add((y) -> m.thenCompose(y, new CompletableFutureInc(m)));
3848  
3849 +        funs.add((y) -> CompletableFuture.allOf(new CompletableFuture<?>[] {y}));
3850          funs.add((y) -> CompletableFuture.allOf(new CompletableFuture<?>[] {y, v42}));
3851 +        funs.add((y) -> CompletableFuture.allOf(new CompletableFuture<?>[] {v42, y}));
3852 +        funs.add((y) -> CompletableFuture.anyOf(new CompletableFuture<?>[] {y}));
3853          funs.add((y) -> CompletableFuture.anyOf(new CompletableFuture<?>[] {y, incomplete}));
3854 +        funs.add((y) -> CompletableFuture.anyOf(new CompletableFuture<?>[] {incomplete, y}));
3855  
3856          for (Function<CompletableFuture<Integer>, CompletableFuture<?>>
3857                   fun : funs) {
# Line 3358 | Line 3902 | public class CompletableFutureTest exten
3902          }
3903      }}
3904  
3905 +    /**
3906 +     * Minimal completion stages throw UOE for all non-CompletionStage methods
3907 +     */
3908 +    public void testMinimalCompletionStage_minimality() {
3909 +        if (!testImplementationDetails) return;
3910 +        Function<Method, String> toSignature =
3911 +            (method) -> method.getName() + Arrays.toString(method.getParameterTypes());
3912 +        Predicate<Method> isNotStatic =
3913 +            (method) -> (method.getModifiers() & Modifier.STATIC) == 0;
3914 +        List<Method> minimalMethods =
3915 +            Stream.of(Object.class, CompletionStage.class)
3916 +            .flatMap((klazz) -> Stream.of(klazz.getMethods()))
3917 +            .filter(isNotStatic)
3918 +            .collect(Collectors.toList());
3919 +        // Methods from CompletableFuture permitted NOT to throw UOE
3920 +        String[] signatureWhitelist = {
3921 +            "newIncompleteFuture[]",
3922 +            "defaultExecutor[]",
3923 +            "minimalCompletionStage[]",
3924 +            "copy[]",
3925 +        };
3926 +        Set<String> permittedMethodSignatures =
3927 +            Stream.concat(minimalMethods.stream().map(toSignature),
3928 +                          Stream.of(signatureWhitelist))
3929 +            .collect(Collectors.toSet());
3930 +        List<Method> allMethods = Stream.of(CompletableFuture.class.getMethods())
3931 +            .filter(isNotStatic)
3932 +            .filter((method) -> !permittedMethodSignatures.contains(toSignature.apply(method)))
3933 +            .collect(Collectors.toList());
3934 +
3935 +        CompletionStage<Integer> minimalStage =
3936 +            new CompletableFuture<Integer>().minimalCompletionStage();
3937 +
3938 +        List<Method> bugs = new ArrayList<>();
3939 +        for (Method method : allMethods) {
3940 +            Class<?>[] parameterTypes = method.getParameterTypes();
3941 +            Object[] args = new Object[parameterTypes.length];
3942 +            // Manufacture boxed primitives for primitive params
3943 +            for (int i = 0; i < args.length; i++) {
3944 +                Class<?> type = parameterTypes[i];
3945 +                if (parameterTypes[i] == boolean.class)
3946 +                    args[i] = false;
3947 +                else if (parameterTypes[i] == int.class)
3948 +                    args[i] = 0;
3949 +                else if (parameterTypes[i] == long.class)
3950 +                    args[i] = 0L;
3951 +            }
3952 +            try {
3953 +                method.invoke(minimalStage, args);
3954 +                bugs.add(method);
3955 +            }
3956 +            catch (java.lang.reflect.InvocationTargetException expected) {
3957 +                if (! (expected.getCause() instanceof UnsupportedOperationException)) {
3958 +                    bugs.add(method);
3959 +                    // expected.getCause().printStackTrace();
3960 +                }
3961 +            }
3962 +            catch (ReflectiveOperationException bad) { throw new Error(bad); }
3963 +        }
3964 +        if (!bugs.isEmpty())
3965 +            throw new Error("Methods did not throw UOE: " + bugs.toString());
3966 +    }
3967 +
3968 +    static class Monad {
3969 +        static class ZeroException extends RuntimeException {
3970 +            public ZeroException() { super("monadic zero"); }
3971 +        }
3972 +        // "return", "unit"
3973 +        static <T> CompletableFuture<T> unit(T value) {
3974 +            return completedFuture(value);
3975 +        }
3976 +        // monadic zero ?
3977 +        static <T> CompletableFuture<T> zero() {
3978 +            return failedFuture(new ZeroException());
3979 +        }
3980 +        // >=>
3981 +        static <T,U,V> Function<T, CompletableFuture<V>> compose
3982 +            (Function<T, CompletableFuture<U>> f,
3983 +             Function<U, CompletableFuture<V>> g) {
3984 +            return (x) -> f.apply(x).thenCompose(g);
3985 +        }
3986 +
3987 +        static void assertZero(CompletableFuture<?> f) {
3988 +            try {
3989 +                f.getNow(null);
3990 +                throw new AssertionFailedError("should throw");
3991 +            } catch (CompletionException success) {
3992 +                assertTrue(success.getCause() instanceof ZeroException);
3993 +            }
3994 +        }
3995 +
3996 +        static <T> void assertFutureEquals(CompletableFuture<T> f,
3997 +                                           CompletableFuture<T> g) {
3998 +            T fval = null, gval = null;
3999 +            Throwable fex = null, gex = null;
4000 +
4001 +            try { fval = f.get(); }
4002 +            catch (ExecutionException ex) { fex = ex.getCause(); }
4003 +            catch (Throwable ex) { fex = ex; }
4004 +
4005 +            try { gval = g.get(); }
4006 +            catch (ExecutionException ex) { gex = ex.getCause(); }
4007 +            catch (Throwable ex) { gex = ex; }
4008 +
4009 +            if (fex != null || gex != null)
4010 +                assertSame(fex.getClass(), gex.getClass());
4011 +            else
4012 +                assertEquals(fval, gval);
4013 +        }
4014 +
4015 +        static class PlusFuture<T> extends CompletableFuture<T> {
4016 +            AtomicReference<Throwable> firstFailure = new AtomicReference<>(null);
4017 +        }
4018 +
4019 +        /** Implements "monadic plus". */
4020 +        static <T> CompletableFuture<T> plus(CompletableFuture<? extends T> f,
4021 +                                             CompletableFuture<? extends T> g) {
4022 +            PlusFuture<T> plus = new PlusFuture<T>();
4023 +            BiConsumer<T, Throwable> action = (T result, Throwable ex) -> {
4024 +                try {
4025 +                    if (ex == null) {
4026 +                        if (plus.complete(result))
4027 +                            if (plus.firstFailure.get() != null)
4028 +                                plus.firstFailure.set(null);
4029 +                    }
4030 +                    else if (plus.firstFailure.compareAndSet(null, ex)) {
4031 +                        if (plus.isDone())
4032 +                            plus.firstFailure.set(null);
4033 +                    }
4034 +                    else {
4035 +                        // first failure has precedence
4036 +                        Throwable first = plus.firstFailure.getAndSet(null);
4037 +
4038 +                        // may fail with "Self-suppression not permitted"
4039 +                        try { first.addSuppressed(ex); }
4040 +                        catch (Exception ignored) {}
4041 +
4042 +                        plus.completeExceptionally(first);
4043 +                    }
4044 +                } catch (Throwable unexpected) {
4045 +                    plus.completeExceptionally(unexpected);
4046 +                }
4047 +            };
4048 +            f.whenComplete(action);
4049 +            g.whenComplete(action);
4050 +            return plus;
4051 +        }
4052 +    }
4053 +
4054 +    /**
4055 +     * CompletableFuture is an additive monad - sort of.
4056 +     * https://en.wikipedia.org/wiki/Monad_(functional_programming)#Additive_monads
4057 +     */
4058 +    public void testAdditiveMonad() throws Throwable {
4059 +        Function<Long, CompletableFuture<Long>> unit = Monad::unit;
4060 +        CompletableFuture<Long> zero = Monad.zero();
4061 +
4062 +        // Some mutually non-commutative functions
4063 +        Function<Long, CompletableFuture<Long>> triple
4064 +            = (x) -> Monad.unit(3 * x);
4065 +        Function<Long, CompletableFuture<Long>> inc
4066 +            = (x) -> Monad.unit(x + 1);
4067 +
4068 +        // unit is a right identity: m >>= unit === m
4069 +        Monad.assertFutureEquals(inc.apply(5L).thenCompose(unit),
4070 +                                 inc.apply(5L));
4071 +        // unit is a left identity: (unit x) >>= f === f x
4072 +        Monad.assertFutureEquals(unit.apply(5L).thenCompose(inc),
4073 +                                 inc.apply(5L));
4074 +
4075 +        // associativity: (m >>= f) >>= g === m >>= ( \x -> (f x >>= g) )
4076 +        Monad.assertFutureEquals(
4077 +            unit.apply(5L).thenCompose(inc).thenCompose(triple),
4078 +            unit.apply(5L).thenCompose((x) -> inc.apply(x).thenCompose(triple)));
4079 +
4080 +        // The case for CompletableFuture as an additive monad is weaker...
4081 +
4082 +        // zero is a monadic zero
4083 +        Monad.assertZero(zero);
4084 +
4085 +        // left zero: zero >>= f === zero
4086 +        Monad.assertZero(zero.thenCompose(inc));
4087 +        // right zero: f >>= (\x -> zero) === zero
4088 +        Monad.assertZero(inc.apply(5L).thenCompose((x) -> zero));
4089 +
4090 +        // f plus zero === f
4091 +        Monad.assertFutureEquals(Monad.unit(5L),
4092 +                                 Monad.plus(Monad.unit(5L), zero));
4093 +        // zero plus f === f
4094 +        Monad.assertFutureEquals(Monad.unit(5L),
4095 +                                 Monad.plus(zero, Monad.unit(5L)));
4096 +        // zero plus zero === zero
4097 +        Monad.assertZero(Monad.plus(zero, zero));
4098 +        {
4099 +            CompletableFuture<Long> f = Monad.plus(Monad.unit(5L),
4100 +                                                   Monad.unit(8L));
4101 +            // non-determinism
4102 +            assertTrue(f.get() == 5L || f.get() == 8L);
4103 +        }
4104 +
4105 +        CompletableFuture<Long> godot = new CompletableFuture<>();
4106 +        // f plus godot === f (doesn't wait for godot)
4107 +        Monad.assertFutureEquals(Monad.unit(5L),
4108 +                                 Monad.plus(Monad.unit(5L), godot));
4109 +        // godot plus f === f (doesn't wait for godot)
4110 +        Monad.assertFutureEquals(Monad.unit(5L),
4111 +                                 Monad.plus(godot, Monad.unit(5L)));
4112 +    }
4113 +
4114 +    /**
4115 +     * A single CompletableFuture with many dependents.
4116 +     * A demo of scalability - runtime is O(n).
4117 +     */
4118 +    public void testManyDependents() throws Throwable {
4119 +        final int n = 1_000;
4120 +        final CompletableFuture<Void> head = new CompletableFuture<>();
4121 +        final CompletableFuture<Void> complete = CompletableFuture.completedFuture((Void)null);
4122 +        final AtomicInteger count = new AtomicInteger(0);
4123 +        for (int i = 0; i < n; i++) {
4124 +            head.thenRun(() -> count.getAndIncrement());
4125 +            head.thenAccept((x) -> count.getAndIncrement());
4126 +            head.thenApply((x) -> count.getAndIncrement());
4127 +
4128 +            head.runAfterBoth(complete, () -> count.getAndIncrement());
4129 +            head.thenAcceptBoth(complete, (x, y) -> count.getAndIncrement());
4130 +            head.thenCombine(complete, (x, y) -> count.getAndIncrement());
4131 +            complete.runAfterBoth(head, () -> count.getAndIncrement());
4132 +            complete.thenAcceptBoth(head, (x, y) -> count.getAndIncrement());
4133 +            complete.thenCombine(head, (x, y) -> count.getAndIncrement());
4134 +
4135 +            head.runAfterEither(new CompletableFuture<Void>(), () -> count.getAndIncrement());
4136 +            head.acceptEither(new CompletableFuture<Void>(), (x) -> count.getAndIncrement());
4137 +            head.applyToEither(new CompletableFuture<Void>(), (x) -> count.getAndIncrement());
4138 +            new CompletableFuture<Void>().runAfterEither(head, () -> count.getAndIncrement());
4139 +            new CompletableFuture<Void>().acceptEither(head, (x) -> count.getAndIncrement());
4140 +            new CompletableFuture<Void>().applyToEither(head, (x) -> count.getAndIncrement());
4141 +        }
4142 +        head.complete(null);
4143 +        assertEquals(5 * 3 * n, count.get());
4144 +    }
4145 +
4146 + //     static <U> U join(CompletionStage<U> stage) {
4147 + //         CompletableFuture<U> f = new CompletableFuture<>();
4148 + //         stage.whenComplete((v, ex) -> {
4149 + //             if (ex != null) f.completeExceptionally(ex); else f.complete(v);
4150 + //         });
4151 + //         return f.join();
4152 + //     }
4153 +
4154 + //     static <U> boolean isDone(CompletionStage<U> stage) {
4155 + //         CompletableFuture<U> f = new CompletableFuture<>();
4156 + //         stage.whenComplete((v, ex) -> {
4157 + //             if (ex != null) f.completeExceptionally(ex); else f.complete(v);
4158 + //         });
4159 + //         return f.isDone();
4160 + //     }
4161 +
4162 + //     static <U> U join2(CompletionStage<U> stage) {
4163 + //         return stage.toCompletableFuture().copy().join();
4164 + //     }
4165 +
4166 + //     static <U> boolean isDone2(CompletionStage<U> stage) {
4167 + //         return stage.toCompletableFuture().copy().isDone();
4168 + //     }
4169 +
4170   }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines