Add more stubbing tests for consecutive calls and builder mocking

Author: m-koops

tests/src/test/kotlin/test/OngoingStubbingTest.kt

@@ -28,6 +28,64 @@ class OngoingStubbingTest : TestBase() {
         expect(result).toBe("A")
     }
 
+    @Test
+    fun `should stub consecutive function calls`() {
+        /* Given */
+        val mock = mock<Open> {
+            on { stringResult() }.doReturn ("A", "B", "C")
+        }
+
+        /* When */
+        val result =(1..3).map { _ ->
+            mock.stringResult()
+        }
+
+        /* Then */
+        expect(result).toBe(listOf("A", "B", "C"))
+    }
+
+    @Test
+    fun `should stub consecutive function calls by a list of answers`() {
+        /* Given */
+        val mock = mock<Open> {
+            on { stringResult() } doReturnConsecutively listOf("A", "B", "C")
+        }
+
+        /* When */
+        val result =(1..3).map { _ ->
+            mock.stringResult()
+        }
+
+        /* Then */
+        expect(result).toBe(listOf("A", "B", "C"))
+    }
+
+    @Test
+    fun `should stub builder method returning mock itself via answer`() {
+        /* Given */
+        val mock = mock<SynchronousFunctions> {
+            on { builderMethod() } doAnswer Mockito.RETURNS_SELF
+        }
+
+        /* When */
+        val result = mock.builderMethod()
+
+        /* Then */
+        expect(result).toBe(mock)
+    }
+
+    @Test
+    fun `should stub builder method returning mock itself via defaultAnswer`() {
+        /* Given */
+        val mock = mock<SynchronousFunctions>(defaultAnswer = Mockito.RETURNS_SELF)
+
+        /* When */
+        val result = mock.builderMethod()
+
+        /* Then */
+        expect(result).toBeTheSameAs(mock)
+    }
+
     @Test
     fun `should stub builder method returning mock itself`() {
         /* Given */
@@ -149,20 +207,6 @@ class OngoingStubbingTest : TestBase() {
         expect(result).toBe("result")
     }
 
-    @Test
-    fun `should stub builder method returning mock itself via answer`() {
-        /* Given */
-        val mock = mock<SynchronousFunctions> {
-            on { builderMethod() } doAnswer Mockito.RETURNS_SELF
-        }
-
-        /* When */
-        val result = mock.builderMethod()
-
-        /* Then */
-        expect(result).toBe(mock)
-    }
-
     @Test
     fun `should stub function call with result from lambda with argument`() {
         /* Given */
@@ -207,18 +251,6 @@ class OngoingStubbingTest : TestBase() {
         expect(result).toBe(true)
     }
 
-    @Test
-    fun `should stub consecutive function calls by a list of answers`() {
-        /* Given */
-        val mock = mock<Open> {
-            on { stringResult() } doReturnConsecutively listOf("a", "b")
-        }
-
-        /* Then */
-        expect(mock.stringResult()).toBe("a")
-        expect(mock.stringResult()).toBe("b")
-    }
-
     @Test
     fun `should stub function call with integer result`() {
         /* Given */