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.134 by jsr166, Sun Nov 15 19:55:38 2015 UTC vs.
Revision 1.149 by jsr166, Sun Jun 26 16:01:17 2016 UTC

# Line 876 | Line 876 | public class CompletableFutureTest exten
876          assertEquals(1, a.get());
877      }}
878  
879 +    /**
880 +     * If an "exceptionally action" throws an exception, it completes
881 +     * exceptionally with that exception
882 +     */
883      public void testExceptionally_exceptionalCompletionActionFailed() {
884          for (boolean createIncomplete : new boolean[] { true, false })
885      {
# Line 894 | Line 898 | public class CompletableFutureTest exten
898          if (createIncomplete) f.completeExceptionally(ex1);
899  
900          checkCompletedWithWrappedException(g, ex2);
901 +        checkCompletedExceptionally(f, ex1);
902          assertEquals(1, a.get());
903      }}
904  
# Line 911 | Line 916 | public class CompletableFutureTest exten
916          if (!createIncomplete) assertTrue(f.complete(v1));
917          final CompletableFuture<Integer> g = m.whenComplete
918              (f,
919 <             (Integer x, Throwable t) -> {
919 >             (Integer result, Throwable t) -> {
920                  m.checkExecutionMode();
921 <                threadAssertSame(x, v1);
921 >                threadAssertSame(result, v1);
922                  threadAssertNull(t);
923                  a.getAndIncrement();
924              });
# Line 938 | Line 943 | public class CompletableFutureTest exten
943          if (!createIncomplete) f.completeExceptionally(ex);
944          final CompletableFuture<Integer> g = m.whenComplete
945              (f,
946 <             (Integer x, Throwable t) -> {
946 >             (Integer result, Throwable t) -> {
947                  m.checkExecutionMode();
948 <                threadAssertNull(x);
948 >                threadAssertNull(result);
949                  threadAssertSame(t, ex);
950                  a.getAndIncrement();
951              });
# Line 965 | Line 970 | public class CompletableFutureTest exten
970          if (!createIncomplete) assertTrue(f.cancel(mayInterruptIfRunning));
971          final CompletableFuture<Integer> g = m.whenComplete
972              (f,
973 <             (Integer x, Throwable t) -> {
973 >             (Integer result, Throwable t) -> {
974                  m.checkExecutionMode();
975 <                threadAssertNull(x);
975 >                threadAssertNull(result);
976                  threadAssertTrue(t instanceof CancellationException);
977                  a.getAndIncrement();
978              });
# Line 993 | Line 998 | public class CompletableFutureTest exten
998          if (!createIncomplete) assertTrue(f.complete(v1));
999          final CompletableFuture<Integer> g = m.whenComplete
1000              (f,
1001 <             (Integer x, Throwable t) -> {
1001 >             (Integer result, Throwable t) -> {
1002                  m.checkExecutionMode();
1003 <                threadAssertSame(x, v1);
1003 >                threadAssertSame(result, v1);
1004                  threadAssertNull(t);
1005                  a.getAndIncrement();
1006                  throw ex;
# Line 1024 | Line 1029 | public class CompletableFutureTest exten
1029          if (!createIncomplete) f.completeExceptionally(ex1);
1030          final CompletableFuture<Integer> g = m.whenComplete
1031              (f,
1032 <             (Integer x, Throwable t) -> {
1032 >             (Integer result, Throwable t) -> {
1033                  m.checkExecutionMode();
1034                  threadAssertSame(t, ex1);
1035 <                threadAssertNull(x);
1035 >                threadAssertNull(result);
1036                  a.getAndIncrement();
1037                  throw ex2;
1038              });
# Line 1035 | Line 1040 | public class CompletableFutureTest exten
1040  
1041          checkCompletedWithWrappedException(g, ex1);
1042          checkCompletedExceptionally(f, ex1);
1043 +        if (testImplementationDetails) {
1044 +            assertEquals(1, ex1.getSuppressed().length);
1045 +            assertSame(ex2, ex1.getSuppressed()[0]);
1046 +        }
1047          assertEquals(1, a.get());
1048      }}
1049  
# Line 1052 | Line 1061 | public class CompletableFutureTest exten
1061          if (!createIncomplete) assertTrue(f.complete(v1));
1062          final CompletableFuture<Integer> g = m.handle
1063              (f,
1064 <             (Integer x, Throwable t) -> {
1064 >             (Integer result, Throwable t) -> {
1065                  m.checkExecutionMode();
1066 <                threadAssertSame(x, v1);
1066 >                threadAssertSame(result, v1);
1067                  threadAssertNull(t);
1068                  a.getAndIncrement();
1069                  return inc(v1);
# Line 1081 | Line 1090 | public class CompletableFutureTest exten
1090          if (!createIncomplete) f.completeExceptionally(ex);
1091          final CompletableFuture<Integer> g = m.handle
1092              (f,
1093 <             (Integer x, Throwable t) -> {
1093 >             (Integer result, Throwable t) -> {
1094                  m.checkExecutionMode();
1095 <                threadAssertNull(x);
1095 >                threadAssertNull(result);
1096                  threadAssertSame(t, ex);
1097                  a.getAndIncrement();
1098                  return v1;
# Line 1110 | Line 1119 | public class CompletableFutureTest exten
1119          if (!createIncomplete) assertTrue(f.cancel(mayInterruptIfRunning));
1120          final CompletableFuture<Integer> g = m.handle
1121              (f,
1122 <             (Integer x, Throwable t) -> {
1122 >             (Integer result, Throwable t) -> {
1123                  m.checkExecutionMode();
1124 <                threadAssertNull(x);
1124 >                threadAssertNull(result);
1125                  threadAssertTrue(t instanceof CancellationException);
1126                  a.getAndIncrement();
1127                  return v1;
# Line 1139 | Line 1148 | public class CompletableFutureTest exten
1148          if (!createIncomplete) assertTrue(f.complete(v1));
1149          final CompletableFuture<Integer> g = m.handle
1150              (f,
1151 <             (Integer x, Throwable t) -> {
1151 >             (Integer result, Throwable t) -> {
1152                  m.checkExecutionMode();
1153 <                threadAssertSame(x, v1);
1153 >                threadAssertSame(result, v1);
1154                  threadAssertNull(t);
1155                  a.getAndIncrement();
1156                  throw ex;
# Line 1170 | Line 1179 | public class CompletableFutureTest exten
1179          if (!createIncomplete) f.completeExceptionally(ex1);
1180          final CompletableFuture<Integer> g = m.handle
1181              (f,
1182 <             (Integer x, Throwable t) -> {
1182 >             (Integer result, Throwable t) -> {
1183                  m.checkExecutionMode();
1184 <                threadAssertNull(x);
1184 >                threadAssertNull(result);
1185                  threadAssertSame(ex1, t);
1186                  a.getAndIncrement();
1187                  throw ex2;
# Line 3286 | Line 3295 | public class CompletableFutureTest exten
3295              () -> f.obtrudeException(null),
3296  
3297              () -> CompletableFuture.delayedExecutor(1L, SECONDS, null),
3298 <            () -> CompletableFuture.delayedExecutor(1L, null, new ThreadExecutor()),
3298 >            () -> CompletableFuture.delayedExecutor(1L, null, exec),
3299              () -> CompletableFuture.delayedExecutor(1L, null),
3300  
3301              () -> f.orTimeout(1L, null),
# Line 3516 | Line 3525 | public class CompletableFutureTest exten
3525          long timeoutMillis = timeoutMillis();
3526          CompletableFuture<Integer> f = new CompletableFuture<>();
3527          long startTime = System.nanoTime();
3528 <        f.orTimeout(timeoutMillis, MILLISECONDS);
3528 >        assertSame(f, f.orTimeout(timeoutMillis, MILLISECONDS));
3529          checkCompletedWithTimeoutException(f);
3530          assertTrue(millisElapsedSince(startTime) >= timeoutMillis);
3531      }
# Line 3531 | Line 3540 | public class CompletableFutureTest exten
3540          CompletableFuture<Integer> g = new CompletableFuture<>();
3541          long startTime = System.nanoTime();
3542          f.complete(v1);
3543 <        f.orTimeout(LONG_DELAY_MS, MILLISECONDS);
3544 <        g.orTimeout(LONG_DELAY_MS, MILLISECONDS);
3543 >        assertSame(f, f.orTimeout(LONG_DELAY_MS, MILLISECONDS));
3544 >        assertSame(g, g.orTimeout(LONG_DELAY_MS, MILLISECONDS));
3545          g.complete(v1);
3546          checkCompletedNormally(f, v1);
3547          checkCompletedNormally(g, v1);
# Line 3547 | Line 3556 | public class CompletableFutureTest exten
3556                         () -> testCompleteOnTimeout_timesOut(null));
3557      }
3558  
3559 +    /**
3560 +     * completeOnTimeout completes with given value if not complete
3561 +     */
3562      public void testCompleteOnTimeout_timesOut(Integer v) {
3563          long timeoutMillis = timeoutMillis();
3564          CompletableFuture<Integer> f = new CompletableFuture<>();
3565          long startTime = System.nanoTime();
3566 <        f.completeOnTimeout(v, timeoutMillis, MILLISECONDS);
3566 >        assertSame(f, f.completeOnTimeout(v, timeoutMillis, MILLISECONDS));
3567          assertSame(v, f.join());
3568          assertTrue(millisElapsedSince(startTime) >= timeoutMillis);
3569          f.complete(99);         // should have no effect
# Line 3568 | Line 3580 | public class CompletableFutureTest exten
3580          CompletableFuture<Integer> g = new CompletableFuture<>();
3581          long startTime = System.nanoTime();
3582          f.complete(v1);
3583 <        f.completeOnTimeout(-1, LONG_DELAY_MS, MILLISECONDS);
3584 <        g.completeOnTimeout(-1, LONG_DELAY_MS, MILLISECONDS);
3583 >        assertSame(f, f.completeOnTimeout(-1, LONG_DELAY_MS, MILLISECONDS));
3584 >        assertSame(g, g.completeOnTimeout(-1, LONG_DELAY_MS, MILLISECONDS));
3585          g.complete(v1);
3586          checkCompletedNormally(f, v1);
3587          checkCompletedNormally(g, v1);
# Line 3620 | Line 3632 | public class CompletableFutureTest exten
3632      //--- tests of implementation details; not part of official tck ---
3633  
3634      Object resultOf(CompletableFuture<?> f) {
3635 +        SecurityManager sm = System.getSecurityManager();
3636 +        if (sm != null) {
3637 +            try {
3638 +                System.setSecurityManager(null);
3639 +            } catch (SecurityException giveUp) {
3640 +                return "Reflection not available";
3641 +            }
3642 +        }
3643 +
3644          try {
3645              java.lang.reflect.Field resultField
3646                  = CompletableFuture.class.getDeclaredField("result");
3647              resultField.setAccessible(true);
3648              return resultField.get(f);
3649 <        } catch (Throwable t) { throw new AssertionError(t); }
3649 >        } catch (Throwable t) {
3650 >            throw new AssertionError(t);
3651 >        } finally {
3652 >            if (sm != null) System.setSecurityManager(sm);
3653 >        }
3654      }
3655  
3656      public void testExceptionPropagationReusesResultObject() {
# Line 3648 | Line 3673 | public class CompletableFutureTest exten
3673          funs.add((y) -> m.applyToEither(y, incomplete, new IncFunction(m)));
3674  
3675          funs.add((y) -> m.runAfterBoth(y, v42, new Noop(m)));
3676 +        funs.add((y) -> m.runAfterBoth(v42, y, new Noop(m)));
3677          funs.add((y) -> m.thenAcceptBoth(y, v42, new SubtractAction(m)));
3678 +        funs.add((y) -> m.thenAcceptBoth(v42, y, new SubtractAction(m)));
3679          funs.add((y) -> m.thenCombine(y, v42, new SubtractFunction(m)));
3680 +        funs.add((y) -> m.thenCombine(v42, y, new SubtractFunction(m)));
3681  
3682 <        funs.add((y) -> m.whenComplete(y, (Integer x, Throwable t) -> {}));
3682 >        funs.add((y) -> m.whenComplete(y, (Integer r, Throwable t) -> {}));
3683  
3684          funs.add((y) -> m.thenCompose(y, new CompletableFutureInc(m)));
3685  
3686 +        funs.add((y) -> CompletableFuture.allOf(new CompletableFuture<?>[] {y}));
3687          funs.add((y) -> CompletableFuture.allOf(new CompletableFuture<?>[] {y, v42}));
3688 +        funs.add((y) -> CompletableFuture.allOf(new CompletableFuture<?>[] {v42, y}));
3689 +        funs.add((y) -> CompletableFuture.anyOf(new CompletableFuture<?>[] {y}));
3690          funs.add((y) -> CompletableFuture.anyOf(new CompletableFuture<?>[] {y, incomplete}));
3691 +        funs.add((y) -> CompletableFuture.anyOf(new CompletableFuture<?>[] {incomplete, y}));
3692  
3693          for (Function<CompletableFuture<Integer>, CompletableFuture<?>>
3694                   fun : funs) {
# Line 3821 | Line 3853 | public class CompletableFutureTest exten
3853              AtomicReference<Throwable> firstFailure = new AtomicReference<>(null);
3854          }
3855  
3856 <        // Monadic "plus"
3856 >        /** Implements "monadic plus". */
3857          static <T> CompletableFuture<T> plus(CompletableFuture<? extends T> f,
3858                                               CompletableFuture<? extends T> g) {
3859              PlusFuture<T> plus = new PlusFuture<T>();
3860              BiConsumer<T, Throwable> action = (T result, Throwable ex) -> {
3861 <                if (ex == null) {
3862 <                    if (plus.complete(result))
3863 <                        if (plus.firstFailure.get() != null)
3861 >                try {
3862 >                    if (ex == null) {
3863 >                        if (plus.complete(result))
3864 >                            if (plus.firstFailure.get() != null)
3865 >                                plus.firstFailure.set(null);
3866 >                    }
3867 >                    else if (plus.firstFailure.compareAndSet(null, ex)) {
3868 >                        if (plus.isDone())
3869                              plus.firstFailure.set(null);
3870 <                }
3871 <                else if (plus.firstFailure.compareAndSet(null, ex)) {
3872 <                    if (plus.isDone())
3873 <                        plus.firstFailure.set(null);
3874 <                }
3875 <                else {
3876 <                    // first failure has precedence
3877 <                    Throwable first = plus.firstFailure.getAndSet(null);
3878 <
3879 <                    // may fail with "Self-suppression not permitted"
3880 <                    try { first.addSuppressed(ex); }
3881 <                    catch (Exception ignored) {}
3882 <
3846 <                    plus.completeExceptionally(first);
3870 >                    }
3871 >                    else {
3872 >                        // first failure has precedence
3873 >                        Throwable first = plus.firstFailure.getAndSet(null);
3874 >
3875 >                        // may fail with "Self-suppression not permitted"
3876 >                        try { first.addSuppressed(ex); }
3877 >                        catch (Exception ignored) {}
3878 >
3879 >                        plus.completeExceptionally(first);
3880 >                    }
3881 >                } catch (Throwable unexpected) {
3882 >                    plus.completeExceptionally(unexpected);
3883                  }
3884              };
3885              f.whenComplete(action);
# Line 3912 | Line 3948 | public class CompletableFutureTest exten
3948                                   Monad.plus(godot, Monad.unit(5L)));
3949      }
3950  
3951 +    /**
3952 +     * A single CompletableFuture with many dependents.
3953 +     * A demo of scalability - runtime is O(n).
3954 +     */
3955 +    public void testManyDependents() throws Throwable {
3956 +        final int n = 1_000;
3957 +        final CompletableFuture<Void> head = new CompletableFuture<>();
3958 +        final CompletableFuture<Void> complete = CompletableFuture.completedFuture((Void)null);
3959 +        final AtomicInteger count = new AtomicInteger(0);
3960 +        for (int i = 0; i < n; i++) {
3961 +            head.thenRun(() -> count.getAndIncrement());
3962 +            head.thenAccept((x) -> count.getAndIncrement());
3963 +            head.thenApply((x) -> count.getAndIncrement());
3964 +
3965 +            head.runAfterBoth(complete, () -> count.getAndIncrement());
3966 +            head.thenAcceptBoth(complete, (x, y) -> count.getAndIncrement());
3967 +            head.thenCombine(complete, (x, y) -> count.getAndIncrement());
3968 +            complete.runAfterBoth(head, () -> count.getAndIncrement());
3969 +            complete.thenAcceptBoth(head, (x, y) -> count.getAndIncrement());
3970 +            complete.thenCombine(head, (x, y) -> count.getAndIncrement());
3971 +
3972 +            head.runAfterEither(new CompletableFuture<Void>(), () -> count.getAndIncrement());
3973 +            head.acceptEither(new CompletableFuture<Void>(), (x) -> count.getAndIncrement());
3974 +            head.applyToEither(new CompletableFuture<Void>(), (x) -> count.getAndIncrement());
3975 +            new CompletableFuture<Void>().runAfterEither(head, () -> count.getAndIncrement());
3976 +            new CompletableFuture<Void>().acceptEither(head, (x) -> count.getAndIncrement());
3977 +            new CompletableFuture<Void>().applyToEither(head, (x) -> count.getAndIncrement());
3978 +        }
3979 +        head.complete(null);
3980 +        assertEquals(5 * 3 * n, count.get());
3981 +    }
3982 +
3983   //     static <U> U join(CompletionStage<U> stage) {
3984   //         CompletableFuture<U> f = new CompletableFuture<>();
3985   //         stage.whenComplete((v, ex) -> {

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines