25
25
26
26
package com.nhaarman.mockito_kotlin
27
27
28
+ import org.mockito.InOrder
28
29
import org.mockito.MockSettings
30
+ import org.mockito.MockingDetails
29
31
import org.mockito.Mockito
30
32
import org.mockito.invocation.InvocationOnMock
31
33
import org.mockito.stubbing.Answer
34
+ import org.mockito.stubbing.DeprecatedOngoingStubbing
35
+ import org.mockito.stubbing.OngoingStubbing
36
+ import org.mockito.stubbing.Stubber
32
37
import org.mockito.verification.VerificationMode
38
+ import org.mockito.verification.VerificationWithTimeout
33
39
import kotlin.reflect.KClass
34
40
35
41
fun after (millis : Long ) = Mockito .after(millis)
@@ -44,59 +50,58 @@ inline fun <reified T : Any> anyVararg() = Mockito.anyVararg<T>() ?: createInsta
44
50
45
51
inline fun <reified T : Any > argThat (noinline predicate : T .() -> Boolean ) = Mockito .argThat<T > { it -> (it as T ).predicate() } ? : createInstance(T ::class )
46
52
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)!!
51
57
52
58
fun <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)
54
60
55
- fun <T > doAnswer (answer : (InvocationOnMock ) -> T ? ) = Mockito .doAnswer { answer(it) }
61
+ fun <T > doAnswer (answer : (InvocationOnMock ) -> T ? ): Stubber = Mockito .doAnswer { answer(it) }!!
56
62
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)!!
63
69
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)!!
67
73
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)
70
76
inline fun <reified T : Any > isNull (): T ? = Mockito .isNull(T ::class .java)
71
77
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)!!
76
82
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)
82
88
83
- fun reset () = Mockito .reset<Any >()
84
89
fun <T > reset (vararg mocks : T ) = Mockito .reset(* mocks)
85
90
86
- fun <T > same (value : T ) = Mockito .same(value)
91
+ fun <T > same (value : T ): T ? = Mockito .same(value)
87
92
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)!!
90
95
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)!!
94
99
fun validateMockitoUsage () = Mockito .validateMockitoUsage()
95
100
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)!!
98
103
fun <T > verifyNoMoreInteractions (vararg mocks : T ) = Mockito .verifyNoMoreInteractions(* mocks)
99
104
fun verifyZeroInteractions (vararg mocks : Any ) = Mockito .verifyZeroInteractions(* mocks)
100
105
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