Add doReturn().on { method() } helper to KStubbing (#453) (#481)

Co-authored-by: Kate Corcoran <[email protected]>

Author: thecodewarrior

mockito-kotlin/src/main/kotlin/org/mockito/kotlin/KStubbing.kt

@@ -30,6 +30,7 @@ import kotlinx.coroutines.runBlocking
 import org.mockito.Mockito
 import org.mockito.exceptions.misusing.NotAMockException
 import org.mockito.stubbing.OngoingStubbing
+import org.mockito.stubbing.Stubber
 import kotlin.reflect.KClass
 
 
@@ -85,4 +86,8 @@ class KStubbing<out T : Any>(val mock: T) {
     ): OngoingStubbing<R> {
         return runBlocking { Mockito.`when`(mock.m()) }
     }
+
+    fun Stubber.on(methodCall: T.() -> Unit) {
+        this.`when`(mock).methodCall()
+    }
 }

tests/src/test/kotlin/test/StubberTest.kt

@@ -100,4 +100,13 @@ class StubberTest : TestBase() {
             mock.go()
         }
     }
+
+    @Test
+    fun testStubberOnBlockExtension() {
+        val mock = mock<Methods> {
+            doReturn("Test").on { stringResult() }
+        }
+
+        expect(mock.stringResult()).toBe("Test")
+    }
 }