From feaf63e2ea38828b25dd23bf34a8128aefe0d50b Mon Sep 17 00:00:00 2001
From: Niek Haarman <haarman.niek@gmail.com>
Date: Fri, 5 Jul 2019 14:39:15 +0200
Subject: [PATCH] Make after() return type non-null

---
 .../kotlin/com/nhaarman/mockitokotlin2/Verification.kt    | 4 ++--
 tests/src/test/kotlin/test/VerificationTest.kt            | 8 ++++++++
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/mockito-kotlin/src/main/kotlin/com/nhaarman/mockitokotlin2/Verification.kt b/mockito-kotlin/src/main/kotlin/com/nhaarman/mockitokotlin2/Verification.kt
index b52e61e0..b8e124cb 100644
--- a/mockito-kotlin/src/main/kotlin/com/nhaarman/mockitokotlin2/Verification.kt
+++ b/mockito-kotlin/src/main/kotlin/com/nhaarman/mockitokotlin2/Verification.kt
@@ -165,8 +165,8 @@ fun description(description: String): VerificationMode {
  * interaction rather than failing immediately if has not already happened. May be useful for testing in concurrent
  * conditions.
  */
-fun after(millis: Long): VerificationAfterDelay? {
-    return Mockito.after(millis)
+fun after(millis: Long): VerificationAfterDelay {
+    return Mockito.after(millis)!!
 }
 
 /**
diff --git a/tests/src/test/kotlin/test/VerificationTest.kt b/tests/src/test/kotlin/test/VerificationTest.kt
index c2253bc0..1858dec5 100644
--- a/tests/src/test/kotlin/test/VerificationTest.kt
+++ b/tests/src/test/kotlin/test/VerificationTest.kt
@@ -103,4 +103,12 @@ class VerificationTest : TestBase() {
             expect(e.message).toContain("Test")
         }
     }
+
+    @Test
+    fun testAfter() {
+        mock<Methods>().apply {
+            int(3)
+            verify(this, after(10)).int(3)
+        }
+    }
 }
\ No newline at end of file