Add infix modifiers to doReturn methods

Author: nhaarman

mockito-kotlin/src/main/kotlin/com/nhaarman/mockito_kotlin/Mockito.kt

@@ -87,8 +87,9 @@ class KStubbing<out T>(private val mock: T) {
     fun <R> on(methodCall: T.() -> R) = Mockito.`when`(mock.methodCall())
 }
 
-fun <T> OngoingStubbing<T>.doReturn(t: T) = thenReturn(t)
-fun <T> OngoingStubbing<T>.doReturn(t: T, vararg ts: T) = thenReturn(t, *ts)
+infix fun <T> OngoingStubbing<T>.doReturn(t: T): OngoingStubbing<T> = thenReturn(t)
+fun <T> OngoingStubbing<T>.doReturn(t: T, vararg ts: T): OngoingStubbing<T> = thenReturn(t, *ts)
+inline infix fun <reified T> OngoingStubbing<T>.doReturn(ts: List<T>): OngoingStubbing<T> = thenReturn(ts[0], *ts.drop(1).toTypedArray())
 
 fun mockingDetails(toInspect: Any): MockingDetails = Mockito.mockingDetails(toInspect)!!
 fun never(): VerificationMode = Mockito.never()!!

mockito-kotlin/src/test/kotlin/MockitoTest.kt

@@ -287,7 +287,7 @@ class MockitoTest {
     fun testMockStubbing_lambda() {
         /* Given */
         val mock = mock<Open>() {
-            on { stringResult() }.doReturn("A")
+            on { stringResult() } doReturn "A"
         }
 
         /* When */
@@ -326,4 +326,16 @@ class MockitoTest {
         /* Then */
         expect(result).toBe("A")
     }
+
+    @Test
+    fun doReturn_withSingleItemList() {
+        /* Given */
+        val mock = mock<Open> {
+            on { stringResult() } doReturn listOf("a", "b")
+        }
+
+        /* Then */
+        expect(mock.stringResult()).toBe("a")
+        expect(mock.stringResult()).toBe("b")
+    }
 }