X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Ftest%2FMockotlin.kt;h=b30e98c2ef431a0c46cae314fb3d33260921918b;hb=8dd998006410b230392c0df06a29f95be1571b6f;hp=ed2f3fc47656d6c3c0e9f1bcb96318cee8dcf963;hpb=831779f8fcdf131063016644b7fb39baade8527b;p=Sone.git

diff --git a/src/test/kotlin/net/pterodactylus/sone/test/Mockotlin.kt b/src/test/kotlin/net/pterodactylus/sone/test/Mockotlin.kt
index ed2f3fc..b30e98c 100644
--- a/src/test/kotlin/net/pterodactylus/sone/test/Mockotlin.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/test/Mockotlin.kt
@@ -1,5 +1,18 @@
 package net.pterodactylus.sone.test
 
+import com.google.inject.Module
+import org.mockito.ArgumentCaptor
 import org.mockito.Mockito
 
 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)!!
+inline fun <reified T : Any> deepMock(): T = Mockito.mock<T>(T::class.java, Mockito.RETURNS_DEEP_STUBS)!!
+inline fun <reified T : Any> capture(): ArgumentCaptor<T> = ArgumentCaptor.forClass(T::class.java)
+
+inline fun <reified T : Any> bind(implementation: T): Module =
+		Module { it!!.bind(T::class.java).toInstance(implementation) }
+
+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)