From 5a5fa35f76c12668c0432eef432d0e320bfcf12a Mon Sep 17 00:00:00 2001
From: pawel bochenski <pawel.bochenski@siili.com>
Date: Thu, 30 Mar 2017 15:15:53 +0200
Subject: [PATCH] Added stub extension function

with this method one can define mock for class

val someMock = mock<Object>()

and then define stubs in test

someMock.stub {
	on { something } doReturn somethingElse
}

It is useful when some tests reuse mock object but with different stubbing
---
 .../src/main/kotlin/com/nhaarman/mockito_kotlin/Mockito.kt      | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/mockito-kotlin/src/main/kotlin/com/nhaarman/mockito_kotlin/Mockito.kt b/mockito-kotlin/src/main/kotlin/com/nhaarman/mockito_kotlin/Mockito.kt
index 07e20179..6d62267c 100644
--- a/mockito-kotlin/src/main/kotlin/com/nhaarman/mockito_kotlin/Mockito.kt
+++ b/mockito-kotlin/src/main/kotlin/com/nhaarman/mockito_kotlin/Mockito.kt
@@ -175,6 +175,8 @@ inline fun <reified T : Any> mock(
     KStubbing(this).stubbing(this)
 }!!
 
+inline fun <T : Any> T.stub(stubbing: KStubbing<T>.(T) -> Unit) = this.apply { KStubbing(this).stubbing(this) }
+
 @Deprecated("Use mock() with optional arguments instead.", ReplaceWith("mock<T>(defaultAnswer = a)"), level = WARNING)
 inline fun <reified T : Any> mock(a: Answer<Any>): T = mock(defaultAnswer = a)