diff --git a/vavr/src/test/java/com/baeldung/vavr/future/FutureTest.java b/vavr/src/test/java/com/baeldung/vavr/future/FutureTest.java index 002919a937..d5345cad55 100644 --- a/vavr/src/test/java/com/baeldung/vavr/future/FutureTest.java +++ b/vavr/src/test/java/com/baeldung/vavr/future/FutureTest.java @@ -58,6 +58,14 @@ public class FutureTest { assertThat(result) .isEqualTo(HELLO); } + + @Test + public void whenTransform_thenCorrect() { + Future future = Future.of(() -> 5) + .transformValue(result -> Try.of(() -> HELLO + result.get())); + + assertThat(future.get()).isEqualTo(HELLO + 5); + } @Test public void whenChainingCallbacks_thenCorrect() { @@ -139,6 +147,14 @@ public class FutureTest { assertThat(futureResult.get()) .isEqualTo("Hello from Baeldung"); } + + @Test + public void whenCallFlatMap_thenCorrect() { + Future futureMap = Future.of(() -> 1) + .flatMap((i) -> Future.of(() -> "Hello: " + i)); + + assertThat(futureMap.get()).isEqualTo("Hello: 1"); + } @Test public void whenFutureFails_thenGetErrorMessage() {