commit | 64077603cd6825fe52b4212ccbf67d9e21d83a4e | [log] [tgz] |
---|---|---|
author | Steve Elliott <steell@google.com> | Thu Sep 08 20:46:40 2022 +0000 |
committer | Automerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com> | Thu Sep 08 20:46:40 2022 +0000 |
tree | 9ba29d4c50f592dd58a54bc020e886132e1a3077 | |
parent | a1121c79d26ca897e862ba1f1368c8f7f7fc835b [diff] | |
parent | 3a24aba20e0f48160d257f8c65265bb62a4c0781 [diff] |
Merge "Add captureMany { } mockito test utility" into tm-qpr-dev am: 3a24aba20e Original change: https://googleplex-android-review.googlesource.com/c/platform/frameworks/base/+/19859999 Change-Id: Ide09bcd4d0e4d272d51e2910688f582d860bf764 Signed-off-by: Automerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com>
diff --git a/packages/SystemUI/tests/utils/src/com/android/systemui/util/mockito/KotlinMockitoHelpers.kt b/packages/SystemUI/tests/utils/src/com/android/systemui/util/mockito/KotlinMockitoHelpers.kt index f539dbd..d058053 100644 --- a/packages/SystemUI/tests/utils/src/com/android/systemui/util/mockito/KotlinMockitoHelpers.kt +++ b/packages/SystemUI/tests/utils/src/com/android/systemui/util/mockito/KotlinMockitoHelpers.kt
@@ -118,3 +118,17 @@ */ inline fun <reified T : Any> withArgCaptor(block: KotlinArgumentCaptor<T>.() -> Unit): T = kotlinArgumentCaptor<T>().apply { block() }.value + +/** + * Variant of [withArgCaptor] for capturing multiple arguments. + * + * val captor = argumentCaptor<Foo>() + * verify(...).someMethod(captor.capture()) + * val captured: List<Foo> = captor.allValues + * + * becomes: + * + * val capturedList = captureMany<Foo> { verify(...).someMethod(capture()) } + */ +inline fun <reified T : Any> captureMany(block: KotlinArgumentCaptor<T>.() -> Unit): List<T> = + kotlinArgumentCaptor<T>().apply{ block() }.allValues