2525
2626package com.nhaarman.mockito_kotlin
2727
28+ import org.mockito.InOrder
2829import org.mockito.MockSettings
30+ import org.mockito.MockingDetails
2931import org.mockito.Mockito
3032import org.mockito.invocation.InvocationOnMock
3133import org.mockito.stubbing.Answer
34+ import org.mockito.stubbing.DeprecatedOngoingStubbing
35+ import org.mockito.stubbing.OngoingStubbing
36+ import org.mockito.stubbing.Stubber
3237import org.mockito.verification.VerificationMode
38+ import org.mockito.verification.VerificationWithTimeout
3339import kotlin.reflect.KClass
3440
3541fun after (millis : Long ) = Mockito .after(millis)
@@ -44,59 +50,58 @@ inline fun <reified T : Any> anyVararg() = Mockito.anyVararg<T>() ?: createInsta
4450
4551inline fun <reified T : Any > argThat (noinline predicate : T .() -> Boolean ) = Mockito .argThat<T > { it -> (it as T ).predicate() } ? : createInstance(T ::class )
4652
47- fun atLeast (numInvocations : Int ) = Mockito .atLeast(numInvocations)
48- fun atLeastOnce () = Mockito .atLeastOnce()
49- fun atMost (maxNumberOfInvocations : Int ) = Mockito .atMost(maxNumberOfInvocations)
50- fun calls (wantedNumberOfInvocations : Int ) = Mockito .calls(wantedNumberOfInvocations)
53+ fun atLeast (numInvocations : Int ): VerificationMode = Mockito .atLeast(numInvocations)!!
54+ fun atLeastOnce (): VerificationMode = Mockito .atLeastOnce()!!
55+ fun atMost (maxNumberOfInvocations : Int ): VerificationMode = Mockito .atMost(maxNumberOfInvocations)!!
56+ fun calls (wantedNumberOfInvocations : Int ): VerificationMode = Mockito .calls(wantedNumberOfInvocations)!!
5157
5258fun <T > clearInvocations (vararg mocks : T ) = Mockito .clearInvocations(* mocks)
53- fun description (description : String ) = Mockito .description(description)
59+ fun description (description : String ): VerificationMode = Mockito .description(description)
5460
55- fun <T > doAnswer (answer : (InvocationOnMock ) -> T ? ) = Mockito .doAnswer { answer(it) }
61+ fun <T > doAnswer (answer : (InvocationOnMock ) -> T ? ): Stubber = Mockito .doAnswer { answer(it) }!!
5662
57- fun doCallRealMethod () = Mockito .doCallRealMethod()
58- fun doNothing () = Mockito .doNothing()
59- fun doReturn (value : Any ) = Mockito .doReturn(value)
60- fun doReturn (toBeReturned : Any , vararg toBeReturnedNext : Any ) = Mockito .doReturn(toBeReturned, * toBeReturnedNext)
61- fun doThrow (toBeThrown : KClass <out Throwable >) = Mockito .doThrow(toBeThrown.java)
62- fun doThrow (vararg toBeThrown : Throwable ) = Mockito .doThrow(* toBeThrown)
63+ fun doCallRealMethod (): Stubber = Mockito .doCallRealMethod()!!
64+ fun doNothing (): Stubber = Mockito .doNothing()!!
65+ fun doReturn (value : Any? ): Stubber = Mockito .doReturn(value)!!
66+ fun doReturn (toBeReturned : Any? , vararg toBeReturnedNext : Any? ): Stubber = Mockito .doReturn(toBeReturned, * toBeReturnedNext)!!
67+ fun doThrow (toBeThrown : KClass <out Throwable >): Stubber = Mockito .doThrow(toBeThrown.java)!!
68+ fun doThrow (vararg toBeThrown : Throwable ): Stubber = Mockito .doThrow(* toBeThrown)!!
6369
64- inline fun <reified T : Any > eq (value : T ) = Mockito .eq(value) ? : createInstance<T >()
65- fun ignoreStubs (vararg mocks : Any ) = Mockito .ignoreStubs(* mocks)
66- fun inOrder (vararg mocks : Any ) = Mockito .inOrder(* mocks)
70+ inline fun <reified T : Any > eq (value : T ): T = Mockito .eq(value) ? : createInstance<T >()
71+ fun ignoreStubs (vararg mocks : Any ): Array < out Any > = Mockito .ignoreStubs(* mocks)!!
72+ fun inOrder (vararg mocks : Any ): InOrder = Mockito .inOrder(* mocks)!!
6773
68- inline fun <reified T : Any > isA () = Mockito .isA(T ::class .java)
69- inline fun <reified T : Any > isNotNull () = Mockito .isNotNull(T ::class .java)
74+ inline fun <reified T : Any > isA (): T ? = Mockito .isA(T ::class .java)
75+ inline fun <reified T : Any > isNotNull (): T ? = Mockito .isNotNull(T ::class .java)
7076inline fun <reified T : Any > isNull (): T ? = Mockito .isNull(T ::class .java)
7177
72- inline fun <reified T : Any > mock () = Mockito .mock(T ::class .java)
73- inline fun <reified T : Any > mock (defaultAnswer : Answer <Any >) = Mockito .mock(T ::class .java, defaultAnswer)
74- inline fun <reified T : Any > mock (s : MockSettings ) = Mockito .mock(T ::class .java, s)
75- inline fun <reified T : Any > mock (s : String ) = Mockito .mock(T ::class .java, s)
78+ inline fun <reified T : Any > mock (): T = Mockito .mock(T ::class .java)!!
79+ inline fun <reified T : Any > mock (defaultAnswer : Answer <Any >): T = Mockito .mock(T ::class .java, defaultAnswer)!!
80+ inline fun <reified T : Any > mock (s : MockSettings ): T = Mockito .mock(T ::class .java, s)!!
81+ inline fun <reified T : Any > mock (s : String ): T = Mockito .mock(T ::class .java, s)!!
7682
77- fun mockingDetails (toInspect : Any ) = Mockito .mockingDetails(toInspect)
78- fun never () = Mockito .never()
79- inline fun <reified T : Any > notNull () = Mockito .notNull(T ::class .java)
80- fun only () = Mockito .only()
81- fun <T > refEq (value : T , vararg excludeFields : String ) = Mockito .refEq(value, * excludeFields)
83+ fun mockingDetails (toInspect : Any ): MockingDetails = Mockito .mockingDetails(toInspect)!!
84+ fun never (): VerificationMode = Mockito .never()!!
85+ inline fun <reified T : Any > notNull (): T ? = Mockito .notNull(T ::class .java)
86+ fun only (): VerificationMode = Mockito .only()!!
87+ fun <T > refEq (value : T , vararg excludeFields : String ): T ? = Mockito .refEq(value, * excludeFields)
8288
83- fun reset () = Mockito .reset<Any >()
8489fun <T > reset (vararg mocks : T ) = Mockito .reset(* mocks)
8590
86- fun <T > same (value : T ) = Mockito .same(value)
91+ fun <T > same (value : T ): T ? = Mockito .same(value)
8792
88- inline fun <reified T : Any > spy () = Mockito .spy(T ::class .java)
89- fun <T > spy (value : T ) = Mockito .spy(value)
93+ inline fun <reified T : Any > spy (): T = Mockito .spy(T ::class .java)!!
94+ fun <T > spy (value : T ): T = Mockito .spy(value)!!
9095
91- fun <T > stub (methodCall : T ) = Mockito .stub(methodCall)
92- fun timeout (millis : Long ) = Mockito .timeout(millis)
93- fun times (numInvocations : Int ) = Mockito .times(numInvocations)
96+ fun <T > stub (methodCall : T ): DeprecatedOngoingStubbing < T > = Mockito .stub(methodCall)!!
97+ fun timeout (millis : Long ): VerificationWithTimeout = Mockito .timeout(millis)!!
98+ fun times (numInvocations : Int ): VerificationMode = Mockito .times(numInvocations)!!
9499fun validateMockitoUsage () = Mockito .validateMockitoUsage()
95100
96- fun <T > verify (mock : T ) = Mockito .verify(mock)
97- fun <T > verify (mock : T , mode : VerificationMode ) = Mockito .verify(mock, mode)
101+ fun <T > verify (mock : T ): T = Mockito .verify(mock)!!
102+ fun <T > verify (mock : T , mode : VerificationMode ): T = Mockito .verify(mock, mode)!!
98103fun <T > verifyNoMoreInteractions (vararg mocks : T ) = Mockito .verifyNoMoreInteractions(* mocks)
99104fun verifyZeroInteractions (vararg mocks : Any ) = Mockito .verifyZeroInteractions(* mocks)
100105
101- fun <T > whenever (methodCall : T ) = Mockito .`when `(methodCall)
102- fun withSettings () = Mockito .withSettings()
106+ fun <T > whenever (methodCall : T ): OngoingStubbing < T > = Mockito .`when `(methodCall)!!
107+ fun withSettings (): MockSettings = Mockito .withSettings()!!
0 commit comments