Merge branch 'unlimitedsola-2.x' into 2.x

2 files changed, 18 insertions, 2 deletions

diff --git a/mockito-kotlin/src/main/kotlin/com/nhaarman/mockitokotlin2/BDDMockito.kt b/mockito-kotlin/src/main/kotlin/com/nhaarman/mockitokotlin2/BDDMockito.kt
index 02f8c3c..ead224d 100644
--- a/mockito-kotlin/src/main/kotlin/com/nhaarman/mockitokotlin2/BDDMockito.kt
+++ b/mockito-kotlin/src/main/kotlin/com/nhaarman/mockitokotlin2/BDDMockito.kt
@@ -27,6 +27,7 @@ package com.nhaarman.mockitokotlin2
 
 import org.mockito.BDDMockito
 import org.mockito.BDDMockito.BDDMyOngoingStubbing
+import org.mockito.invocation.InvocationOnMock
 import org.mockito.stubbing.Answer
 
 /**
@@ -60,8 +61,8 @@ infix fun <T> BDDMyOngoingStubbing<T>.will(value: Answer<T>): BDDMockito.BDDMyOn
 /**
  * Alias for [BBDMyOngoingStubbing.willAnswer], accepting a lambda.
  */
-infix fun <T> BDDMyOngoingStubbing<T>.willAnswer(value: () -> T): BDDMockito.BDDMyOngoingStubbing<T> {
-    return willAnswer { value() }
+infix fun <T> BDDMyOngoingStubbing<T>.willAnswer(value: (InvocationOnMock) -> T?): BDDMockito.BDDMyOngoingStubbing<T> {
+    return willAnswer { value(it) }
 }
 
 /**
diff --git a/tests/src/test/kotlin/test/BDDMockitoTest.kt b/tests/src/test/kotlin/test/BDDMockitoTest.kt
index 0d090b2..3d43ac4 100644
--- a/tests/src/test/kotlin/test/BDDMockitoTest.kt
+++ b/tests/src/test/kotlin/test/BDDMockitoTest.kt
@@ -91,6 +91,21 @@ class BDDMockitoTest {
         expect(mock.stringResult()).toBe("Test")
     }
 
+    @Test
+    fun given_willAnswerInfix_withInvocationInfo_properlyStubs() {
+        /* Given */
+        val mock = mock<Methods>()
+
+        /* When */
+        given(mock.stringResult(any())) willAnswer { invocation ->
+            (invocation.arguments[0] as String)
+                .reversed()
+        }
+
+        /* Then */
+        expect(mock.stringResult("Test")).toBe("tseT")
+    }
+
     @Test(expected = IllegalStateException::class)
     fun given_willThrowInfix_properlyStubs() {
         /* Given */