From 2b5a125440c16a45d464e4a80bd7111e6c495f52 Mon Sep 17 00:00:00 2001
From: Ahmed Tawila <atawila@yahoo.com>
Date: Tue, 9 Jan 2018 13:31:45 +0200
Subject: [PATCH] BAEL-1080 Introduction to Future in Vavr (#3372)

* Add missing features

* Add missing features
---
 .../com/baeldung/vavr/future/FutureTest.java     | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

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<Object> 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<Object> futureMap = Future.of(() -> 1)
+          .flatMap((i) -> Future.of(() -> "Hello: " + i));
+     
+        assertThat(futureMap.get()).isEqualTo("Hello: 1");
+    }
 
     @Test
     public void whenFutureFails_thenGetErrorMessage() {