X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Ftest%2FMockotlin.kt;h=38625ead493dea971395618c436f167cc4c9c47b;hp=b30e98c2ef431a0c46cae314fb3d33260921918b;hb=9824d45ada252d673a259029052f5ac4eecc5420;hpb=8883cd2aa6d2294e10259f7a5338e6c16b26c8b2

diff --git a/src/test/kotlin/net/pterodactylus/sone/test/Mockotlin.kt b/src/test/kotlin/net/pterodactylus/sone/test/Mockotlin.kt
index b30e98c..38625ea 100644
--- a/src/test/kotlin/net/pterodactylus/sone/test/Mockotlin.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/test/Mockotlin.kt
@@ -3,6 +3,7 @@ package net.pterodactylus.sone.test
 import com.google.inject.Module
 import org.mockito.ArgumentCaptor
 import org.mockito.Mockito
+import org.mockito.stubbing.OngoingStubbing
 
 inline fun <reified T : Any> mock(): T = Mockito.mock<T>(T::class.java)!!
 inline fun <reified T : Any> mockBuilder(): T = Mockito.mock<T>(T::class.java, Mockito.RETURNS_SELF)!!
@@ -16,3 +17,5 @@ inline fun <reified T : Any> bindMock(): Module =
 		Module { it!!.bind(T::class.java).toInstance(mock<T>()) }
 
 inline fun <reified T: Any> whenever(methodCall: T) = Mockito.`when`(methodCall)
+
+inline fun <reified T : Any> OngoingStubbing<T>.thenReturnMock(): OngoingStubbing<T> = this.thenReturn(mock<T>())