eugenp · eric-martin · Jun 24, 2024 · Jun 3, 2024 · Jun 11, 2024 · Jun 13, 2024
diff --git a/testing-modules/mockito-3/pom.xml b/testing-modules/mockito-3/pom.xml
@@ -34,6 +34,12 @@
             <version>5.11.0</version>
             <scope>test</scope>
         </dependency>
+        <dependency>
+            <groupId>org.awaitility</groupId>
+            <artifactId>awaitility</artifactId>
+            <version>${awaitility.version}</version>
+            <scope>test</scope>
+        </dependency>
     </dependencies>
 
     <build>
@@ -46,4 +52,8 @@
         </resources>
     </build>
 
+    <properties>
+        <awaitility.version>4.2.1</awaitility.version>
+    </properties>
+
 </project>
diff --git a/...ng-modules/mockito-3/src/main/java/com/baeldung/mockito/delayresponse/PaymentService.java b/...ng-modules/mockito-3/src/main/java/com/baeldung/mockito/delayresponse/PaymentService.java
@@ -0,0 +1,9 @@
+package com.baeldung.mockito.delayresponse;
+
+public class PaymentService {
+
+    public String processPayment(){
+        // simulate processing payment and return completion status
+        return "SUCCESS";
+    }
+}
diff --git a/...es/mockito-3/src/test/java/com/baeldung/mockito/delayresponse/PaymentServiceUnitTest.java b/...es/mockito-3/src/test/java/com/baeldung/mockito/delayresponse/PaymentServiceUnitTest.java
@@ -0,0 +1,67 @@
+package com.baeldung.mockito.delayresponse;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertTrue;
+import static org.mockito.Mockito.when;
+
+import java.util.concurrent.TimeUnit;
+
+import org.awaitility.Awaitility;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.AdditionalAnswers;
+import org.mockito.Mock;
+import org.mockito.junit.MockitoJUnitRunner;
+
+@RunWith(MockitoJUnitRunner.class)
+public class PaymentServiceUnitTest {
+
+    @Mock
+    PaymentService paymentService;
+    private static final long DELAY = 1;
+
+    @Test
+    public void whenProcessingPayment_thenDelayResponseUsingThreadSleep(){
+        when(paymentService.processPayment()).thenAnswer(invocation -> {
+            Thread.sleep(DELAY); // Delay of 1 second
+            return "SUCCESS";
+        });
+
+        long startTime = System.currentTimeMillis();
+        String result = paymentService.processPayment();
+        long endTime = System.currentTimeMillis();
+
+        assertEquals("SUCCESS", result);
+        assertTrue((endTime - startTime) >= DELAY); // Verify the delay
+    }
+
+    @Test
+    public void whenProcessingPayment_thenDelayResponseUsingAnswersWithDelay() {
+
+        when(paymentService.processPayment()).thenAnswer(AdditionalAnswers.answersWithDelay(DELAY, invocation -> "SUCCESS"));
+
+        long startTime = System.currentTimeMillis();
+        String result = paymentService.processPayment();
+        long endTime = System.currentTimeMillis();
+
+        assertEquals("SUCCESS", result);
+        assertTrue((endTime - startTime) >= DELAY); // Verify the delay
+    }
+
+
+    @Test
+    public void whenProcessingPayment_thenDelayResponseUsingAwaitility() {
+
+        when(paymentService.processPayment()).thenAnswer(invocation -> {
+            Awaitility.await().pollDelay(DELAY, TimeUnit.MILLISECONDS).until(()->true);
+            return "SUCCESS";
+        });
+
+        long startTime = System.currentTimeMillis();
+        String result = paymentService.processPayment();
+        long endTime = System.currentTimeMillis();
+
+        assertEquals("SUCCESS", result);
+        assertTrue((endTime - startTime) >= DELAY); // Verify the delay
+    }
+}