1 package net.pterodactylus.sone.test
3 import com.google.inject.Injector
4 import com.google.inject.Key
5 import com.google.inject.Module
6 import com.google.inject.TypeLiteral
7 import com.google.inject.name.Names
8 import org.hamcrest.MatcherAssert.assertThat
9 import org.hamcrest.Matchers.sameInstance
10 import org.mockito.Mockito
11 import javax.inject.Provider
12 import kotlin.reflect.KClass
14 fun <T : Any> KClass<T>.isProvidedBy(instance: T) = Module { it.bind(this.java).toProvider(Provider<T> { instance }) }
15 fun <T : Any> KClass<T>.withNameIsProvidedBy(instance: T, name: String) = Module { it.bind(this.java).annotatedWith(Names.named(name)).toProvider(Provider<T> { instance }) }
16 fun <T : Any> KClass<T>.isProvidedBy(provider: com.google.inject.Provider<T>) = Module { it.bind(this.java).toProvider(provider) }
17 fun <T : Any> KClass<T>.isProvidedBy(provider: KClass<out Provider<T>>) = Module { it.bind(this.java).toProvider(provider.java) }
18 fun <T : Any> Key<T>.isProvidedBy(instance: T) = Module { it.bind(this).toProvider(Provider<T> { instance }) }
19 inline fun <reified T : Any> KClass<T>.isProvidedByMock() = Module { it.bind(this.java).toProvider(Provider<T> { mock() }) }
20 inline fun <reified T : Any> KClass<T>.isProvidedByDeepMock() = Module { it.bind(this.java).toProvider(Provider<T> { deepMock() }) }
22 inline fun <reified T> key(): Key<T> = Key.get(object : TypeLiteral<T>() {})
23 inline fun <reified T> key(annotation: Annotation): Key<T> = Key.get(object : TypeLiteral<T>() {}, annotation)
25 inline fun <reified T : Any> Injector.getInstance(annotation: Annotation? = null): T = annotation
26 ?.let { getInstance(Key.get(object : TypeLiteral<T>() {}, it)) }
27 ?: getInstance(Key.get(object : TypeLiteral<T>() {}))
30 inline fun <reified T : Any> Injector.verifySingletonInstance(annotation: Annotation? = null) {
31 val firstInstance = getInstance<T>(annotation)
32 val secondInstance = getInstance<T>(annotation)
33 assertThat(firstInstance, sameInstance(secondInstance))
36 fun <T : Any> supply(javaClass: Class<T>): Source<T> = object : Source<T> {
37 override fun fromInstance(instance: T) = Module { it.bind(javaClass).toInstance(instance) }
38 override fun byInstance(instance: T) = Module { it.bind(javaClass).toProvider(Provider<T> { instance }) }
39 override fun byProvider(provider: com.google.inject.Provider<T>) = Module { it.bind(javaClass).toProvider(provider) }
40 override fun byProvider(provider: Class<Provider<T>>) = Module { it.bind(javaClass).toProvider(provider) }
41 override fun byMock() = Module { it.bind(javaClass).toInstance(Mockito.mock(javaClass)) }
44 interface Source<T : Any> {
45 fun fromInstance(instance: T): Module
46 fun byInstance(instance: T): Module
47 fun byProvider(provider: com.google.inject.Provider<T>): Module
48 fun byProvider(provider: Class<Provider<T>>): Module