X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Ftest%2FGuice.kt;h=f9119c32e3111905cd73214d9d693b8434b7053c;hp=5eca8fa7db7cc144ff41eee3389f5655261a7de9;hb=03c29a3838e23ed0b9731ca4d84cf58038c30dfe;hpb=32ca1db3e092387c59ef351cf5bd75dee0494ec9

diff --git a/src/test/kotlin/net/pterodactylus/sone/test/Guice.kt b/src/test/kotlin/net/pterodactylus/sone/test/Guice.kt
index 5eca8fa..f9119c3 100644
--- a/src/test/kotlin/net/pterodactylus/sone/test/Guice.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/test/Guice.kt
@@ -1,10 +1,27 @@
 package net.pterodactylus.sone.test
 
+import com.google.inject.Injector
 import com.google.inject.Module
 import javax.inject.Provider
 import kotlin.reflect.KClass
 
-fun <T : Any> KClass<T>.isInstance(instance: T) = Module { it.bind(this.java).toInstance(instance) }
 fun <T : Any> KClass<T>.isProvidedBy(instance: T) = Module { it.bind(this.java).toProvider { instance } }
 fun <T : Any> KClass<T>.isProvidedBy(provider: com.google.inject.Provider<T>) = Module { it.bind(this.java).toProvider(provider) }
 fun <T : Any> KClass<T>.isProvidedBy(provider: KClass<out Provider<T>>) = Module { it.bind(this.java).toProvider(provider.java) }
+inline fun <reified T : Any> KClass<T>.isProvidedByMock() = Module { it.bind(this.java).toProvider { mock() } }
+
+inline fun <reified T : Any> Injector.getInstance() = getInstance(T::class.java)!!
+
+fun <T : Any> supply(javaClass: Class<T>): Source<T> = object : Source<T> {
+	override fun fromInstance(instance: T) = Module { it.bind(javaClass).toInstance(instance) }
+	override fun byInstance(instance: T) = Module { it.bind(javaClass).toProvider { 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) }
+}
+
+interface Source<T : Any> {
+	fun fromInstance(instance: T): Module
+	fun byInstance(instance: T): Module
+	fun byProvider(provider: com.google.inject.Provider<T>): Module
+	fun byProvider(provider: Class<Provider<T>>): Module
+}