JavaCodeHelp
diff --git a/‎core-java-concurrency/src/test/java/com/baeldung/completablefuture/CompletableFutureLongRunningUnitTest.java‎
Lines changed: 20 additions & 0 deletions b/‎core-java-concurrency/src/test/java/com/baeldung/completablefuture/CompletableFutureLongRunningUnitTest.java‎
Lines changed: 20 additions & 0 deletions
diff --git a/‎logging-modules/log4j2/${sys:logging.folder.path}‎ b/‎logging-modules/log4j2/${sys:logging.folder.path}‎
@@ -184,5 +184,25 @@ public void whenAddingThenApplyAsyncToFuture_thenFunctionExecutesAfterComputatio
 
         assertEquals("Hello World", future.get());
     }
+    
+    @Test
+    public void whenPassingTransformation_thenFunctionExecutionWithThenApply() throws InterruptedException, ExecutionException {
+        CompletableFuture<Integer> finalResult = compute().thenApply(s -> s + 1);
+        assertTrue(finalResult.get() == 11);
+    }
+    
+    @Test
+    public void whenPassingPreviousStage_thenFunctionExecutionWithThenCompose() throws InterruptedException, ExecutionException {
+        CompletableFuture<Integer> finalResult = compute().thenCompose(this::computeAnother);
+        assertTrue(finalResult.get() == 20);
+    }
+    
+    public CompletableFuture<Integer> compute(){
+        return CompletableFuture.supplyAsync(() -> 10);
+    }
+    
+    public CompletableFuture<Integer> computeAnother(Integer i){
+        return CompletableFuture.supplyAsync(() -> 10 + i);
+    }
 
 }