From 210c73300d12fc50cd158e8b0d1b9550ec8887b5 Mon Sep 17 00:00:00 2001
From: Niek Haarman <haarman.niek@gmail.com>
Date: Mon, 19 Dec 2016 15:12:31 +0100
Subject: [PATCH] Add BDDMockito functions

---
 .../com/nhaarman/mockito_kotlin/BDDMockito.kt |  13 +++
 .../src/test/kotlin/test/BDDMockitoTest.kt    | 103 ++++++++++++++++++
 2 files changed, 116 insertions(+)
 create mode 100644 mockito-kotlin/src/main/kotlin/com/nhaarman/mockito_kotlin/BDDMockito.kt
 create mode 100644 mockito-kotlin/src/test/kotlin/test/BDDMockitoTest.kt

diff --git a/mockito-kotlin/src/main/kotlin/com/nhaarman/mockito_kotlin/BDDMockito.kt b/mockito-kotlin/src/main/kotlin/com/nhaarman/mockito_kotlin/BDDMockito.kt
new file mode 100644
index 00000000..203232d8
--- /dev/null
+++ b/mockito-kotlin/src/main/kotlin/com/nhaarman/mockito_kotlin/BDDMockito.kt
@@ -0,0 +1,13 @@
+package com.nhaarman.mockito_kotlin
+
+import org.mockito.BDDMockito
+
+fun <T> given(methodCall: T): BDDMockito.BDDMyOngoingStubbing<T> = BDDMockito.given(methodCall)
+fun <T> given(methodCall: () -> T) = given(methodCall())
+
+fun <T> then(mock: T): BDDMockito.Then<T> = BDDMockito.then(mock)
+
+infix fun <T> BDDMockito.BDDMyOngoingStubbing<T>.willAnswer(value: () -> T): BDDMockito.BDDMyOngoingStubbing<T> = willAnswer { value() }
+infix fun <T> BDDMockito.BDDMyOngoingStubbing<T>.willReturn(value: () -> T): BDDMockito.BDDMyOngoingStubbing<T> = willReturn(value())
+infix fun <T> BDDMockito.BDDMyOngoingStubbing<T>.willThrow(value: () -> Throwable): BDDMockito.BDDMyOngoingStubbing<T> = willThrow(value())
+
diff --git a/mockito-kotlin/src/test/kotlin/test/BDDMockitoTest.kt b/mockito-kotlin/src/test/kotlin/test/BDDMockitoTest.kt
new file mode 100644
index 00000000..1df22392
--- /dev/null
+++ b/mockito-kotlin/src/test/kotlin/test/BDDMockitoTest.kt
@@ -0,0 +1,103 @@
+package test
+
+import com.nhaarman.expect.expect
+import com.nhaarman.mockito_kotlin.*
+import org.junit.Test
+
+class BDDMockitoTest {
+
+    @Test
+    fun given_willReturn_properlyStubs() {
+        /* Given */
+        val mock = mock<Methods>()
+
+        /* When */
+        given(mock.stringResult()).willReturn("Test")
+
+        /* Then */
+        expect(mock.stringResult()).toBe("Test")
+    }
+
+    @Test
+    fun givenLambda_willReturn_properlyStubs() {
+        /* Given */
+        val mock = mock<Methods>()
+
+        /* When */
+        given { mock.stringResult() }.willReturn("Test")
+
+        /* Then */
+        expect(mock.stringResult()).toBe("Test")
+    }
+
+    @Test
+    fun given_willReturnLambda_properlyStubs() {
+        /* Given */
+        val mock = mock<Methods>()
+
+        /* When */
+        given(mock.stringResult()).willReturn { "Test" }
+
+        /* Then */
+        expect(mock.stringResult()).toBe("Test")
+    }
+
+    @Test
+    fun givenLambda_willReturnLambda_properlyStubs() {
+        /* Given */
+        val mock = mock<Methods>()
+
+        /* When */
+        given { mock.stringResult() } willReturn { "Test" }
+
+        /* Then */
+        expect(mock.stringResult()).toBe("Test")
+    }
+
+    @Test
+    fun given_willAnswer_properlyStubs() {
+        /* Given */
+        val mock = mock<Methods>()
+
+        /* When */
+        given(mock.stringResult()).willAnswer { "Test" }
+
+        /* Then */
+        expect(mock.stringResult()).toBe("Test")
+    }
+
+    @Test
+    fun given_willAnswerInfix_properlyStubs() {
+        /* Given */
+        val mock = mock<Methods>()
+
+        /* When */
+        given(mock.stringResult()) willAnswer { "Test" }
+
+        /* Then */
+        expect(mock.stringResult()).toBe("Test")
+    }
+
+    @Test(expected = IllegalStateException::class)
+    fun given_willThrowInfix_properlyStubs() {
+        /* Given */
+        val mock = mock<Methods>()
+
+        /* When */
+        given(mock.stringResult()) willThrow { IllegalStateException() }
+        mock.stringResult()
+    }
+
+    @Test
+    fun then() {
+        /* Given */
+        val mock = mock<Methods>()
+        whenever(mock.stringResult()).thenReturn("Test")
+
+        /* When */
+        mock.stringResult()
+
+        /* Then */
+        then(mock).should().stringResult()
+    }
+}