X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Ftest%2FMockotlin.kt;fp=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Ftest%2FMockotlin.kt;h=d3e8dad4910bf712aad8d9cb616f252d55642b01;hp=dcd139bf3cca6a9591818e18853d5fc7d8f325a0;hb=069bcf304238fb08fe7a5d5ed1d48d7d8e73ebc5;hpb=5096eaa5c9259972a07ba5fec125dc9103d727bd

diff --git a/src/test/kotlin/net/pterodactylus/sone/test/Mockotlin.kt b/src/test/kotlin/net/pterodactylus/sone/test/Mockotlin.kt
index dcd139b..d3e8dad 100644
--- a/src/test/kotlin/net/pterodactylus/sone/test/Mockotlin.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/test/Mockotlin.kt
@@ -3,6 +3,7 @@ package net.pterodactylus.sone.test
 import com.google.inject.Module
 import org.mockito.ArgumentCaptor
 import org.mockito.Mockito
+import org.mockito.invocation.InvocationOnMock
 import org.mockito.stubbing.OngoingStubbing
 
 inline fun <reified T : Any> mock(): T = Mockito.mock<T>(T::class.java)!!
@@ -22,3 +23,5 @@ inline fun <reified T : Any> bindMock(): Module =
 inline fun <reified T: Any?> whenever(methodCall: T) = Mockito.`when`(methodCall)!!
 
 inline fun <reified T : Any> OngoingStubbing<T>.thenReturnMock(): OngoingStubbing<T> = this.thenReturn(mock<T>())
+
+operator fun <T> InvocationOnMock.get(index: Int): T = getArgument(index)