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

   1 package net.pterodactylus.sone.test
   2
   3 import com.google.inject.Module
   4 import org.mockito.ArgumentCaptor
   5 import org.mockito.Mockito
   6 import org.mockito.stubbing.OngoingStubbing
   7
   8 inline fun <reified T : Any> mock(): T = Mockito.mock<T>(T::class.java)!!
   9 inline fun <reified T : Any> mockBuilder(): T = Mockito.mock<T>(T::class.java, Mockito.RETURNS_SELF)!!
  10 inline fun <reified T : Any> deepMock(): T = Mockito.mock<T>(T::class.java, Mockito.RETURNS_DEEP_STUBS)!!
  11 inline fun <reified T : Any> capture(): ArgumentCaptor<T> = ArgumentCaptor.forClass(T::class.java)
  12
  13 inline fun <reified T : Any> bind(implementation: T): Module =
  14                 Module { it!!.bind(T::class.java).toInstance(implementation) }
  15
  16 inline fun <reified T : Any> bindMock(): Module =
  17                 Module { it!!.bind(T::class.java).toInstance(mock<T>()) }
  18
  19 inline fun <reified T: Any> whenever(methodCall: T) = Mockito.`when`(methodCall)
  20
  21 inline fun <reified T : Any> OngoingStubbing<T>.thenReturnMock(): OngoingStubbing<T> = this.thenReturn(mock<T>())