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().apply { + int(3) + verify(this, after(10)).int(3) + } + } } \ No newline at end of file