better inOrder for single arguments

Author: Danny Preussler

mockito-kotlin/src/main/kotlin/com/nhaarman/mockitokotlin2/Verification.kt

@@ -199,6 +199,23 @@ fun inOrder(
     Mockito.inOrder(*mocks).evaluation()
 }
 
+/**
+ * Allows [InOrder] verification for a single mocked instance:
+ *
+ * mock.inOrder {
+ *    verify().foo()
+ * }
+ *
+ */
+inline fun <T> T.inOrder(block: InOrderOnType<T>.() -> Any) {
+    block.invoke(InOrderOnType(this))
+}
+
+class InOrderOnType<T>(private val t: T) : InOrder by inOrder(t as Any) {
+
+    fun verify(): T = verify(t)
+}
+
 /**
  * Allows checking if given method was the only one invoked.
  */

tests/src/test/kotlin/test/VerificationTest.kt

@@ -64,6 +64,23 @@ class VerificationTest : TestBase() {
         }
     }
 
+    @Test
+    fun testInOrderWithReceiver() {
+        /* Given */
+        val mock = mock<Methods>()
+
+        /* When */
+        mock.string("")
+        mock.int(0)
+
+        /* Then */
+        mock.inOrder {
+            verify().string(any())
+            verify().int(any())
+            verifyNoMoreInteractions()
+        }
+    }
+
     @Test
     fun testClearInvocations() {
         val mock = mock<Methods>().apply {