Make after() return type non-null

mockito · Jul 5, 2019 · feaf63e · feaf63e
1 parent 4779b29
commit feaf63e
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 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
@@ -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
@@ -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)
+        }
+    }
 }