test: add CompletableFutures For method exception

lokic · lokic · commit 7423c2fe8e2a · 2021-09-08T22:48:05.000+08:00
diff --git a/src/test/java/com/github/lokic/javaplus/CompletableFuturesTest.java b/src/test/java/com/github/lokic/javaplus/CompletableFuturesTest.java
@@ -11,7 +11,7 @@
 public class CompletableFuturesTest {
 
     @Test
-    public void test_For() {
+    public void test_For_success() {
         CompletableFuture<String> r = CompletableFuture.completedFuture("1")
                 .thenCompose(For(t1 -> CompletableFuture.completedFuture("2")))
                 .thenCompose(For((t1, t2) -> CompletableFuture.completedFuture("3")))
@@ -20,4 +20,19 @@ public void test_For() {
         Assert.assertEquals("1234", r.join());
     }
 
+    @Test
+    public void test_For_exception() {
+        CompletableFuture<String> r = CompletableFuture.completedFuture("1")
+                .thenCompose(For(t1 -> CompletableFuture.completedFuture("2")))
+                .thenCompose(For((t1, t2) -> {
+                    CompletableFuture<String> f = new CompletableFuture<>();
+                    f.completeExceptionally(new RuntimeException("test"));
+                    return f;
+                }))
+                .thenCompose(For((t1, t2, t3) -> CompletableFuture.completedFuture("4")))
+                .thenApply(Yield((t1, t2, t3, t4) -> t1 + t2 + t3 + t4));
+
+        Assert.assertTrue(r.isCompletedExceptionally());
+    }
+
 }
