diff options
| -rw-r--r-- | packages/SystemUI/tests/utils/src/com/android/systemui/coroutines/Flow.kt | 14 |
1 files changed, 2 insertions, 12 deletions
diff --git a/packages/SystemUI/tests/utils/src/com/android/systemui/coroutines/Flow.kt b/packages/SystemUI/tests/utils/src/com/android/systemui/coroutines/Flow.kt index e1ba074ac860..ce8d93e2a0e7 100644 --- a/packages/SystemUI/tests/utils/src/com/android/systemui/coroutines/Flow.kt +++ b/packages/SystemUI/tests/utils/src/com/android/systemui/coroutines/Flow.kt @@ -69,10 +69,10 @@ fun <T> TestScope.collectValues( flow: Flow<T>, context: CoroutineContext = EmptyCoroutineContext, start: CoroutineStart = CoroutineStart.DEFAULT, -): FlowValues<T> { +): FlowValue<List<T>> { val values = mutableListOf<T>() backgroundScope.launch(context, start) { flow.collect(values::add) } - return FlowValuesImpl { + return FlowValueImpl { runCurrent() values.toList() } @@ -83,17 +83,7 @@ interface FlowValue<T> : ReadOnlyProperty<Any?, T> { operator fun invoke(): T } -/** @see collectValues */ -interface FlowValues<T> : ReadOnlyProperty<Any?, List<T>> { - operator fun invoke(): List<T> -} - private class FlowValueImpl<T>(private val block: () -> T) : FlowValue<T> { override operator fun invoke(): T = block() override fun getValue(thisRef: Any?, property: KProperty<*>): T = invoke() } - -private class FlowValuesImpl<T>(private val block: () -> List<T>) : FlowValues<T> { - override operator fun invoke(): List<T> = block() - override fun getValue(thisRef: Any?, property: KProperty<*>): List<T> = invoke() -} |