From 5a303f67f56c1931fc851bcf2f73c9545a04887e Mon Sep 17 00:00:00 2001
From: Clint Checketts <clint.checketts@domo.com>
Date: Fri, 12 Oct 2018 22:03:26 -0600
Subject: [PATCH 1/3] Add argumentCaptor that create many at once for easy
 destructuring

---
 .../nhaarman/mockitokotlin2/ArgumentCaptor.kt | 48 ++++++++++++
 .../test/kotlin/test/ArgumentCaptorTest.kt    | 76 +++++++++++++++++++
 2 files changed, 124 insertions(+)

diff --git a/mockito-kotlin/src/main/kotlin/com/nhaarman/mockitokotlin2/ArgumentCaptor.kt b/mockito-kotlin/src/main/kotlin/com/nhaarman/mockitokotlin2/ArgumentCaptor.kt
index 9542446f..426eb321 100644
--- a/mockito-kotlin/src/main/kotlin/com/nhaarman/mockitokotlin2/ArgumentCaptor.kt
+++ b/mockito-kotlin/src/main/kotlin/com/nhaarman/mockitokotlin2/ArgumentCaptor.kt
@@ -36,6 +36,54 @@ inline fun <reified T : Any> argumentCaptor(): KArgumentCaptor<T> {
     return KArgumentCaptor(ArgumentCaptor.forClass(T::class.java), T::class)
 }
 
+/**
+ * Creates 2 [KArgumentCaptor]s for given types.
+ */
+inline fun <reified A: Any, reified B: Any> argumentCaptor2(): Pair<KArgumentCaptor<A>, KArgumentCaptor<B>> {
+    return Pair(KArgumentCaptor(ArgumentCaptor.forClass(A::class.java), A::class),
+            KArgumentCaptor(ArgumentCaptor.forClass(B::class.java), B::class))
+}
+
+/**
+ * Creates 3 [KArgumentCaptor]s for given types.
+ */
+inline fun <reified A: Any, reified B: Any, reified C: Any> argumentCaptor3(): Triple<KArgumentCaptor<A>, KArgumentCaptor<B>, KArgumentCaptor<C>> {
+    return Triple(
+            KArgumentCaptor(ArgumentCaptor.forClass(A::class.java), A::class),
+            KArgumentCaptor(ArgumentCaptor.forClass(B::class.java), B::class),
+            KArgumentCaptor(ArgumentCaptor.forClass(C::class.java), C::class))
+}
+
+data class ArgumentCaptorHolder4<out A, out B, out C, out D>(
+        val first: A, val second: B, val third: C, val fourth: D)
+data class ArgumentCaptorHolder5<out A, out B, out C, out D, out E>(
+        val first: A, val second: B, val third: C, val fourth: D, val fifth: E)
+
+
+/**
+ * Creates 4 [KArgumentCaptor]s for given types.
+ */
+inline fun <reified A: Any, reified B: Any, reified C: Any, reified D: Any> argumentCaptor4(): ArgumentCaptorHolder4<KArgumentCaptor<A>, KArgumentCaptor<B>, KArgumentCaptor<C>, KArgumentCaptor<D>> {
+    return ArgumentCaptorHolder4(
+            KArgumentCaptor(ArgumentCaptor.forClass(A::class.java), A::class),
+            KArgumentCaptor(ArgumentCaptor.forClass(B::class.java), B::class),
+            KArgumentCaptor(ArgumentCaptor.forClass(C::class.java), C::class),
+            KArgumentCaptor(ArgumentCaptor.forClass(D::class.java), D::class))
+}
+/**
+ * Creates 4 [KArgumentCaptor]s for given types.
+ */
+inline fun <reified A: Any, reified B: Any, reified C: Any, reified D: Any, reified E: Any> argumentCaptor5(): ArgumentCaptorHolder5<KArgumentCaptor<A>, KArgumentCaptor<B>, KArgumentCaptor<C>, KArgumentCaptor<D>, KArgumentCaptor<E>> {
+    return ArgumentCaptorHolder5(
+            KArgumentCaptor(ArgumentCaptor.forClass(A::class.java), A::class),
+            KArgumentCaptor(ArgumentCaptor.forClass(B::class.java), B::class),
+            KArgumentCaptor(ArgumentCaptor.forClass(C::class.java), C::class),
+            KArgumentCaptor(ArgumentCaptor.forClass(D::class.java), D::class),
+            KArgumentCaptor(ArgumentCaptor.forClass(E::class.java), E::class))
+}
+
+
+
 /**
  * Creates a [KArgumentCaptor] for given type, taking in a lambda to allow fast verification.
  */
diff --git a/tests/src/test/kotlin/test/ArgumentCaptorTest.kt b/tests/src/test/kotlin/test/ArgumentCaptorTest.kt
index b79227b2..6f64f13c 100644
--- a/tests/src/test/kotlin/test/ArgumentCaptorTest.kt
+++ b/tests/src/test/kotlin/test/ArgumentCaptorTest.kt
@@ -22,6 +22,82 @@ class ArgumentCaptorTest : TestBase() {
         expect(captor.lastValue).toBe(5L)
     }
 
+    @Test
+    fun argumentCaptor_destructuring2() {
+        /* Given */
+        val date: Date = mock()
+
+        /* When */
+        date.time = 5L
+
+        /* Then */
+        val (captor1, captor2) = argumentCaptor2<Long, Long>()
+        verify(date).time = captor1.capture()
+        verify(date).time = captor2.capture()
+        expect(captor1.lastValue).toBe(5L)
+        expect(captor2.lastValue).toBe(5L)
+    }
+
+    @Test
+    fun argumentCaptor_destructuring3() {
+        /* Given */
+        val date: Date = mock()
+
+        /* When */
+        date.time = 5L
+
+        /* Then */
+        val (captor1, captor2, captor3) = argumentCaptor3<Long, Long, Long>()
+        val verifyCaptor:KArgumentCaptor<Long>.() ->Unit = {
+            verify(date).time = capture()
+            expect(lastValue).toBe(5L)
+        }
+        captor1.apply(verifyCaptor)
+        captor2.apply(verifyCaptor)
+        captor3.apply(verifyCaptor)
+    }
+
+    @Test
+    fun argumentCaptor_destructuring4() {
+        /* Given */
+        val date: Date = mock()
+
+        /* When */
+        date.time = 5L
+
+        /* Then */
+        val (captor1, captor2, captor3, captor4) = argumentCaptor4<Long, Long, Long, Long>()
+        val verifyCaptor:KArgumentCaptor<Long>.() ->Unit = {
+            verify(date).time = capture()
+            expect(lastValue).toBe(5L)
+        }
+        captor1.apply(verifyCaptor)
+        captor2.apply(verifyCaptor)
+        captor3.apply(verifyCaptor)
+        captor4.apply(verifyCaptor)
+    }
+
+    @Test
+    fun argumentCaptor_destructuring5() {
+        /* Given */
+        val date: Date = mock()
+
+        /* When */
+        date.time = 5L
+
+        /* Then */
+        val (captor1, captor2, captor3, captor4, captor5) = argumentCaptor5<Long, Long, Long, Long, Long>()
+        val verifyCaptor:KArgumentCaptor<Long>.() ->Unit = {
+            verify(date).time = capture()
+            expect(lastValue).toBe(5L)
+        }
+        captor1.apply(verifyCaptor)
+        captor2.apply(verifyCaptor)
+        captor3.apply(verifyCaptor)
+        captor4.apply(verifyCaptor)
+        captor5.apply(verifyCaptor)
+    }
+
     @Test
     fun argumentCaptor_withNullValue_usingNonNullable() {
         /* Given */

From f4405832a866e7f18cda90796b5c5e593cf68be1 Mon Sep 17 00:00:00 2001
From: Niek Haarman <haarman.niek@gmail.com>
Date: Wed, 19 Jun 2019 20:19:19 +0200
Subject: [PATCH 2/3] Include class parameters to avoid numbers in
 argumentCaptor signatures

---
 .../nhaarman/mockitokotlin2/ArgumentCaptor.kt | 69 +++++++++++++------
 .../test/kotlin/test/ArgumentCaptorTest.kt    | 14 ++--
 2 files changed, 54 insertions(+), 29 deletions(-)

diff --git a/mockito-kotlin/src/main/kotlin/com/nhaarman/mockitokotlin2/ArgumentCaptor.kt b/mockito-kotlin/src/main/kotlin/com/nhaarman/mockitokotlin2/ArgumentCaptor.kt
index 426eb321..ce069753 100644
--- a/mockito-kotlin/src/main/kotlin/com/nhaarman/mockitokotlin2/ArgumentCaptor.kt
+++ b/mockito-kotlin/src/main/kotlin/com/nhaarman/mockitokotlin2/ArgumentCaptor.kt
@@ -39,51 +39,76 @@ inline fun <reified T : Any> argumentCaptor(): KArgumentCaptor<T> {
 /**
  * Creates 2 [KArgumentCaptor]s for given types.
  */
-inline fun <reified A: Any, reified B: Any> argumentCaptor2(): Pair<KArgumentCaptor<A>, KArgumentCaptor<B>> {
-    return Pair(KArgumentCaptor(ArgumentCaptor.forClass(A::class.java), A::class),
-            KArgumentCaptor(ArgumentCaptor.forClass(B::class.java), B::class))
+inline fun <reified A : Any, reified B : Any> argumentCaptor(
+    a: KClass<A> = A::class,
+    b: KClass<B> = B::class
+): Pair<KArgumentCaptor<A>, KArgumentCaptor<B>> {
+    return Pair(
+          KArgumentCaptor(ArgumentCaptor.forClass(a.java), a),
+          KArgumentCaptor(ArgumentCaptor.forClass(b.java), b)
+    )
 }
 
 /**
  * Creates 3 [KArgumentCaptor]s for given types.
  */
-inline fun <reified A: Any, reified B: Any, reified C: Any> argumentCaptor3(): Triple<KArgumentCaptor<A>, KArgumentCaptor<B>, KArgumentCaptor<C>> {
+inline fun <reified A : Any, reified B : Any, reified C : Any> argumentCaptor(
+    a: KClass<A> = A::class,
+    b: KClass<B> = B::class,
+    c: KClass<C> = C::class
+): Triple<KArgumentCaptor<A>, KArgumentCaptor<B>, KArgumentCaptor<C>> {
     return Triple(
-            KArgumentCaptor(ArgumentCaptor.forClass(A::class.java), A::class),
-            KArgumentCaptor(ArgumentCaptor.forClass(B::class.java), B::class),
-            KArgumentCaptor(ArgumentCaptor.forClass(C::class.java), C::class))
+          KArgumentCaptor(ArgumentCaptor.forClass(a.java), a),
+          KArgumentCaptor(ArgumentCaptor.forClass(b.java), b),
+          KArgumentCaptor(ArgumentCaptor.forClass(c.java), c)
+    )
 }
 
 data class ArgumentCaptorHolder4<out A, out B, out C, out D>(
-        val first: A, val second: B, val third: C, val fourth: D)
+    val first: A, val second: B, val third: C, val fourth: D
+)
+
 data class ArgumentCaptorHolder5<out A, out B, out C, out D, out E>(
-        val first: A, val second: B, val third: C, val fourth: D, val fifth: E)
+    val first: A, val second: B, val third: C, val fourth: D, val fifth: E
+)
 
 
 /**
  * Creates 4 [KArgumentCaptor]s for given types.
  */
-inline fun <reified A: Any, reified B: Any, reified C: Any, reified D: Any> argumentCaptor4(): ArgumentCaptorHolder4<KArgumentCaptor<A>, KArgumentCaptor<B>, KArgumentCaptor<C>, KArgumentCaptor<D>> {
+inline fun <reified A : Any, reified B : Any, reified C : Any, reified D : Any> argumentCaptor(
+    a: KClass<A> = A::class,
+    b: KClass<B> = B::class,
+    c: KClass<C> = C::class,
+    d: KClass<D> = D::class
+): ArgumentCaptorHolder4<KArgumentCaptor<A>, KArgumentCaptor<B>, KArgumentCaptor<C>, KArgumentCaptor<D>> {
     return ArgumentCaptorHolder4(
-            KArgumentCaptor(ArgumentCaptor.forClass(A::class.java), A::class),
-            KArgumentCaptor(ArgumentCaptor.forClass(B::class.java), B::class),
-            KArgumentCaptor(ArgumentCaptor.forClass(C::class.java), C::class),
-            KArgumentCaptor(ArgumentCaptor.forClass(D::class.java), D::class))
+          KArgumentCaptor(ArgumentCaptor.forClass(a.java), a),
+          KArgumentCaptor(ArgumentCaptor.forClass(b.java), b),
+          KArgumentCaptor(ArgumentCaptor.forClass(c.java), c),
+          KArgumentCaptor(ArgumentCaptor.forClass(d.java), d)
+    )
 }
+
 /**
  * Creates 4 [KArgumentCaptor]s for given types.
  */
-inline fun <reified A: Any, reified B: Any, reified C: Any, reified D: Any, reified E: Any> argumentCaptor5(): ArgumentCaptorHolder5<KArgumentCaptor<A>, KArgumentCaptor<B>, KArgumentCaptor<C>, KArgumentCaptor<D>, KArgumentCaptor<E>> {
+inline fun <reified A : Any, reified B : Any, reified C : Any, reified D : Any, reified E : Any> argumentCaptor(
+    a: KClass<A> = A::class,
+    b: KClass<B> = B::class,
+    c: KClass<C> = C::class,
+    d: KClass<D> = D::class,
+    e: KClass<E> = E::class
+): ArgumentCaptorHolder5<KArgumentCaptor<A>, KArgumentCaptor<B>, KArgumentCaptor<C>, KArgumentCaptor<D>, KArgumentCaptor<E>> {
     return ArgumentCaptorHolder5(
-            KArgumentCaptor(ArgumentCaptor.forClass(A::class.java), A::class),
-            KArgumentCaptor(ArgumentCaptor.forClass(B::class.java), B::class),
-            KArgumentCaptor(ArgumentCaptor.forClass(C::class.java), C::class),
-            KArgumentCaptor(ArgumentCaptor.forClass(D::class.java), D::class),
-            KArgumentCaptor(ArgumentCaptor.forClass(E::class.java), E::class))
+          KArgumentCaptor(ArgumentCaptor.forClass(a.java), a),
+          KArgumentCaptor(ArgumentCaptor.forClass(b.java), b),
+          KArgumentCaptor(ArgumentCaptor.forClass(c.java), c),
+          KArgumentCaptor(ArgumentCaptor.forClass(d.java), d),
+          KArgumentCaptor(ArgumentCaptor.forClass(e.java), e)
+    )
 }
 
-
-
 /**
  * Creates a [KArgumentCaptor] for given type, taking in a lambda to allow fast verification.
  */
diff --git a/tests/src/test/kotlin/test/ArgumentCaptorTest.kt b/tests/src/test/kotlin/test/ArgumentCaptorTest.kt
index 6f64f13c..10ddb5e0 100644
--- a/tests/src/test/kotlin/test/ArgumentCaptorTest.kt
+++ b/tests/src/test/kotlin/test/ArgumentCaptorTest.kt
@@ -31,7 +31,7 @@ class ArgumentCaptorTest : TestBase() {
         date.time = 5L
 
         /* Then */
-        val (captor1, captor2) = argumentCaptor2<Long, Long>()
+        val (captor1, captor2) = argumentCaptor<Long, Long>()
         verify(date).time = captor1.capture()
         verify(date).time = captor2.capture()
         expect(captor1.lastValue).toBe(5L)
@@ -47,8 +47,8 @@ class ArgumentCaptorTest : TestBase() {
         date.time = 5L
 
         /* Then */
-        val (captor1, captor2, captor3) = argumentCaptor3<Long, Long, Long>()
-        val verifyCaptor:KArgumentCaptor<Long>.() ->Unit = {
+        val (captor1, captor2, captor3) = argumentCaptor<Long, Long, Long>()
+        val verifyCaptor: KArgumentCaptor<Long>.() -> Unit = {
             verify(date).time = capture()
             expect(lastValue).toBe(5L)
         }
@@ -66,8 +66,8 @@ class ArgumentCaptorTest : TestBase() {
         date.time = 5L
 
         /* Then */
-        val (captor1, captor2, captor3, captor4) = argumentCaptor4<Long, Long, Long, Long>()
-        val verifyCaptor:KArgumentCaptor<Long>.() ->Unit = {
+        val (captor1, captor2, captor3, captor4) = argumentCaptor<Long, Long, Long, Long>()
+        val verifyCaptor: KArgumentCaptor<Long>.() -> Unit = {
             verify(date).time = capture()
             expect(lastValue).toBe(5L)
         }
@@ -86,8 +86,8 @@ class ArgumentCaptorTest : TestBase() {
         date.time = 5L
 
         /* Then */
-        val (captor1, captor2, captor3, captor4, captor5) = argumentCaptor5<Long, Long, Long, Long, Long>()
-        val verifyCaptor:KArgumentCaptor<Long>.() ->Unit = {
+        val (captor1, captor2, captor3, captor4, captor5) = argumentCaptor<Long, Long, Long, Long, Long>()
+        val verifyCaptor: KArgumentCaptor<Long>.() -> Unit = {
             verify(date).time = capture()
             expect(lastValue).toBe(5L)
         }

From 9c90f0fde4fcda8e51c9d92874af49ea2b606837 Mon Sep 17 00:00:00 2001
From: Niek Haarman <haarman.niek@gmail.com>
Date: Wed, 19 Jun 2019 20:23:08 +0200
Subject: [PATCH 3/3] Manually provide component functions for captor holder
 classes

This tightens the public API for these classes.
---
 .../nhaarman/mockitokotlin2/ArgumentCaptor.kt | 32 +++++++++++++++----
 1 file changed, 26 insertions(+), 6 deletions(-)

diff --git a/mockito-kotlin/src/main/kotlin/com/nhaarman/mockitokotlin2/ArgumentCaptor.kt b/mockito-kotlin/src/main/kotlin/com/nhaarman/mockitokotlin2/ArgumentCaptor.kt
index ce069753..be69337a 100644
--- a/mockito-kotlin/src/main/kotlin/com/nhaarman/mockitokotlin2/ArgumentCaptor.kt
+++ b/mockito-kotlin/src/main/kotlin/com/nhaarman/mockitokotlin2/ArgumentCaptor.kt
@@ -64,13 +64,33 @@ inline fun <reified A : Any, reified B : Any, reified C : Any> argumentCaptor(
     )
 }
 
-data class ArgumentCaptorHolder4<out A, out B, out C, out D>(
-    val first: A, val second: B, val third: C, val fourth: D
-)
+class ArgumentCaptorHolder4<out A, out B, out C, out D>(
+    val first: A,
+    val second: B,
+    val third: C,
+    val fourth: D
+) {
+
+    operator fun component1() = first
+    operator fun component2() = second
+    operator fun component3() = third
+    operator fun component4() = fourth
+}
 
-data class ArgumentCaptorHolder5<out A, out B, out C, out D, out E>(
-    val first: A, val second: B, val third: C, val fourth: D, val fifth: E
-)
+class ArgumentCaptorHolder5<out A, out B, out C, out D, out E>(
+    val first: A,
+    val second: B,
+    val third: C,
+    val fourth: D,
+    val fifth: E
+) {
+
+    operator fun component1() = first
+    operator fun component2() = second
+    operator fun component3() = third
+    operator fun component4() = fourth
+    operator fun component5() = fifth
+}
 
 
 /**