Make after() return type non-null

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 b52e61e..b8e124c 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 c2253bc..1858dec 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