From a0648e296ca2b396ee359c101ae7f9f61b7cd0da Mon Sep 17 00:00:00 2001
From: Gabriel Casado <gdgcasado@gmail.com>
Date: Sat, 15 Dec 2018 21:05:07 -0200
Subject: [PATCH] Add BDD will extension function

---
 .../com/nhaarman/mockitokotlin2/BDDMockito.kt       |  8 ++++++++
 tests/src/test/kotlin/test/BDDMockitoTest.kt        | 13 +++++++++++++
 2 files changed, 21 insertions(+)

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 e2ca7aed..02f8c3c6 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.stubbing.Answer
 
 /**
  * Alias for [BDDMockito.given].
@@ -49,6 +50,13 @@ fun <T> then(mock: T): BDDMockito.Then<T> {
     return BDDMockito.then(mock)
 }
 
+/**
+ * Alias for [BDDMyOngoingStubbing.will]
+ * */
+infix fun <T> BDDMyOngoingStubbing<T>.will(value: Answer<T>): BDDMockito.BDDMyOngoingStubbing<T> {
+    return will(value)
+}
+
 /**
  * Alias for [BBDMyOngoingStubbing.willAnswer], accepting a lambda.
  */
diff --git a/tests/src/test/kotlin/test/BDDMockitoTest.kt b/tests/src/test/kotlin/test/BDDMockitoTest.kt
index ded14628..0d090b29 100644
--- a/tests/src/test/kotlin/test/BDDMockitoTest.kt
+++ b/tests/src/test/kotlin/test/BDDMockitoTest.kt
@@ -3,9 +3,22 @@ package test
 import com.nhaarman.expect.expect
 import com.nhaarman.mockitokotlin2.*
 import org.junit.Test
+import org.mockito.stubbing.Answer
 
 class BDDMockitoTest {
 
+    @Test
+    fun given_will_properlyStubs() {
+        /* Given */
+        val mock = mock<Methods>()
+
+        /* When */
+        given(mock.stringResult()) will Answer<String> { "Test" }
+
+        /* Then */
+        expect(mock.stringResult()).toBe("Test")
+    }
+
     @Test
     fun given_willReturn_properlyStubs() {
         /* Given */