X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Ftest%2FGuice.kt;h=e27ad079bb096aa1e4f7cc053c12c71563cdee27;hb=cd72add62ab407336b471d4b7cda8e33dd2df5c6;hp=360b0cd20f6d711e06091cd3f550310aec970a6d;hpb=da8b893e6fc024b7d9f5ed1fc7632138dcd129e7;p=Sone.git

diff --git a/src/test/kotlin/net/pterodactylus/sone/test/Guice.kt b/src/test/kotlin/net/pterodactylus/sone/test/Guice.kt
index 360b0cd..e27ad07 100644
--- a/src/test/kotlin/net/pterodactylus/sone/test/Guice.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/test/Guice.kt
@@ -2,6 +2,7 @@ package net.pterodactylus.sone.test
 
 import com.google.inject.Injector
 import com.google.inject.Module
+import org.mockito.*
 import javax.inject.Provider
 import kotlin.reflect.KClass
 
@@ -17,6 +18,7 @@ fun <T : Any> supply(javaClass: Class<T>): Source<T> = object : Source<T> {
 	override fun byInstance(instance: T) = Module { it.bind(javaClass).toProvider(Provider<T> { instance }) }
 	override fun byProvider(provider: com.google.inject.Provider<T>) = Module { it.bind(javaClass).toProvider(provider) }
 	override fun byProvider(provider: Class<Provider<T>>) = Module { it.bind(javaClass).toProvider(provider) }
+	override fun byMock() = Module { it.bind(javaClass).toInstance(Mockito.mock(javaClass)) }
 }
 
 interface Source<T : Any> {
@@ -24,4 +26,5 @@ interface Source<T : Any> {
 	fun byInstance(instance: T): Module
 	fun byProvider(provider: com.google.inject.Provider<T>): Module
 	fun byProvider(provider: Class<Provider<T>>): Module
+	fun byMock(): Module
 }