promise util with propagate cancellation options

futur-api/src/test/java/dev/tommyjs/futur/PromiseTests.java

@@ -9,19 +9,22 @@ import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import reactor.core.publisher.Mono;
 
+import java.util.List;
+import java.util.Map;
 import java.util.Objects;
-import java.util.concurrent.ScheduledFuture;
+import java.util.concurrent.Future;
 import java.util.concurrent.TimeUnit;
+import java.util.concurrent.TimeoutException;
 import java.util.concurrent.atomic.AtomicBoolean;
 
 public final class PromiseTests {
 
     private final Logger logger = LoggerFactory.getLogger(PromiseTests.class);
-    private final PromiseExecutor<ScheduledFuture<?>> executor = SinglePoolExecutor.create(1);
+    private final PromiseExecutor<Future<?>> executor = SinglePoolExecutor.create(5);
     private final PromiseFactory pfac = new SimplePromiseFactory<>(executor, logger);
 
     @Test
-    void testMono() {
+    public void testMono() {
         Exception value = new Exception("Test Error");
 
         var error = pfac.wrap(Mono.error(value));
@@ -34,15 +37,118 @@ public final class PromiseTests {
     }
 
     @Test
-    void testErrorCancellation() throws InterruptedException {
-        var finish = new AtomicBoolean();
+    public void testErrorCancellation() throws InterruptedException {
+        var finished = new AtomicBoolean();
         pfac.start()
-            .thenRunDelayedAsync(() -> finish.set(true), 50, TimeUnit.MILLISECONDS)
+            .thenRunDelayedAsync(() -> finished.set(true), 50, TimeUnit.MILLISECONDS)
             .thenRunAsync(() -> {})
             .cancel();
 
         Thread.sleep(100L);
-        assert !finish.get();
+        assert !finished.get();
+    }
+
+    @Test
+    public void testCombineUtil() throws TimeoutException {
+        pfac.all(
+                pfac.start().thenRunDelayedAsync(() -> {}, 50, TimeUnit.MILLISECONDS),
+                pfac.start().thenRunDelayedAsync(() -> {}, 50, TimeUnit.MILLISECONDS)
+            )
+            .join(100L);
+
+        pfac.allSettled(
+                pfac.start().thenRunDelayedAsync(() -> {}, 50, TimeUnit.MILLISECONDS),
+                pfac.start().thenRunDelayedAsync(() -> {}, 50, TimeUnit.MILLISECONDS)
+            )
+            .join(100L);
+
+        pfac.combine(
+                pfac.start().thenRunDelayedAsync(() -> {}, 50, TimeUnit.MILLISECONDS),
+                pfac.start().thenRunDelayedAsync(() -> {}, 50, TimeUnit.MILLISECONDS)
+            )
+            .join(100L);
+
+        pfac.combine(
+                List.of(
+                    pfac.start().thenRunDelayedAsync(() -> {}, 49, TimeUnit.MILLISECONDS),
+                    pfac.start().thenRunDelayedAsync(() -> {}, 50, TimeUnit.MILLISECONDS),
+                    pfac.start().thenRunDelayedAsync(() -> {}, 51, TimeUnit.MILLISECONDS)
+                )
+            )
+            .join(100L);
+
+        pfac.combine(
+                Map.of(
+                    "a", pfac.start().thenRunDelayedAsync(() -> {}, 50, TimeUnit.MILLISECONDS),
+                    "b", pfac.start().thenRunDelayedAsync(() -> {}, 50, TimeUnit.MILLISECONDS)
+                )
+            )
+            .join(100L);
+    }
+
+    @Test
+    public void testCombineUtilPropagation() throws InterruptedException {
+        var finished1 = new AtomicBoolean();
+        pfac.all(
+                true,
+                pfac.start().thenRunDelayedAsync(() -> finished1.set(true), 50, TimeUnit.MILLISECONDS),
+                pfac.start().thenRunDelayedAsync(() -> finished1.set(true), 50, TimeUnit.MILLISECONDS)
+            )
+            .cancel();
+
+        var finished2 = new AtomicBoolean();
+        pfac.allSettled(
+                true,
+                pfac.start().thenRunDelayedAsync(() -> finished2.set(true), 50, TimeUnit.MILLISECONDS),
+                pfac.start().thenRunDelayedAsync(() -> finished2.set(true), 50, TimeUnit.MILLISECONDS)
+            )
+            .cancel();
+
+        var finished3 = new AtomicBoolean();
+        pfac.combine(
+                true,
+                pfac.start().thenRunDelayedAsync(() -> finished3.set(true), 50, TimeUnit.MILLISECONDS),
+                pfac.start().thenRunDelayedAsync(() -> finished3.set(true), 50, TimeUnit.MILLISECONDS)
+            )
+            .cancel();
+
+        var finished4 = new AtomicBoolean();
+        pfac.combine(
+                true,
+                List.of(
+                    pfac.start().thenRunDelayedAsync(() -> finished4.set(true), 50, TimeUnit.MILLISECONDS),
+                    pfac.start().thenRunDelayedAsync(() -> finished4.set(true), 50, TimeUnit.MILLISECONDS),
+                    pfac.start().thenRunDelayedAsync(() -> finished4.set(true), 50, TimeUnit.MILLISECONDS)
+                )
+            )
+            .cancel();
+
+        var finished5 = new AtomicBoolean();
+        pfac.combine(
+                true,
+                Map.of(
+                    "a", pfac.start().thenRunDelayedAsync(() -> finished5.set(true), 50, TimeUnit.MILLISECONDS),
+                    "b", pfac.start().thenRunDelayedAsync(() -> finished5.set(true), 50, TimeUnit.MILLISECONDS)
+                )
+            )
+            .cancel();
+
+        Thread.sleep(100L);
+        assert !finished1.get();
+        assert !finished2.get();
+        assert !finished3.get();
+        assert !finished4.get();
+        assert !finished5.get();
+    }
+
+    @Test
+    public void testRace() throws TimeoutException {
+        assert pfac.race(
+            List.of(
+                pfac.start().thenSupplyDelayedAsync(() -> true, 50, TimeUnit.MILLISECONDS),
+                pfac.start().thenSupplyDelayedAsync(() -> false, 150, TimeUnit.MILLISECONDS)
+            )
+        ).join(100L);
     }
 
 }