src/test/kotlin/net/pterodactylus/sone/test/Guice.kt

   1 package net.pterodactylus.sone.test
   2
   3 import com.google.inject.*
   4 import com.google.inject.name.*
   5 import org.mockito.*
   6 import javax.inject.Provider
   7 import kotlin.reflect.*
   8
   9 fun <T : Any> KClass<T>.isProvidedBy(instance: T) = Module { it.bind(this.java).toProvider(Provider<T> { instance }) }
  10 fun <T : Any> KClass<T>.withNameIsProvidedBy(instance: T, name: String) = Module { it.bind(this.java).annotatedWith(Names.named(name)).toProvider(Provider<T> { instance }) }
  11 fun <T : Any> KClass<T>.isProvidedBy(provider: com.google.inject.Provider<T>) = Module { it.bind(this.java).toProvider(provider) }
  12 fun <T : Any> KClass<T>.isProvidedBy(provider: KClass<out Provider<T>>) = Module { it.bind(this.java).toProvider(provider.java) }
  13 inline fun <reified T : Any> KClass<T>.isProvidedByMock() = Module { it.bind(this.java).toProvider(Provider<T> { mock() }) }
  14 inline fun <reified T : Any> KClass<T>.isProvidedByDeepMock() = Module { it.bind(this.java).toProvider(Provider<T> { deepMock() }) }
  15
  16 inline fun <reified T : Any> Injector.getInstance(annotation: Annotation? = null): T = annotation
  17                 ?.let { getInstance(Key.get(object : TypeLiteral<T>() {}, it)) }
  18                 ?: getInstance(Key.get(object : TypeLiteral<T>() {}))
  19
  20 fun <T : Any> supply(javaClass: Class<T>): Source<T> = object : Source<T> {
  21         override fun fromInstance(instance: T) = Module { it.bind(javaClass).toInstance(instance) }
  22         override fun byInstance(instance: T) = Module { it.bind(javaClass).toProvider(Provider<T> { instance }) }
  23         override fun byProvider(provider: com.google.inject.Provider<T>) = Module { it.bind(javaClass).toProvider(provider) }
  24         override fun byProvider(provider: Class<Provider<T>>) = Module { it.bind(javaClass).toProvider(provider) }
  25         override fun byMock() = Module { it.bind(javaClass).toInstance(Mockito.mock(javaClass)) }
  26 }
  27
  28 interface Source<T : Any> {
  29         fun fromInstance(instance: T): Module
  30         fun byInstance(instance: T): Module
  31         fun byProvider(provider: com.google.inject.Provider<T>): Module
  32         fun byProvider(provider: Class<Provider<T>>): Module
  33         fun byMock(): Module
  34 }