A helper like this seems to do the trick, though perhaps it's overly specific to my use case. Feedba...

# squarelibraries

[Aidan Low]

A helper like this seems to do the trick, though perhaps it's overly specific to my use case. Feedback is welcome.

fun TestScope.observeStateFlows(
    vararg stateFlows: StateFlow<Any?>,
    block: () -> Unit
) {
    val observationJobs = stateFlows.map { flow ->
        launch {
            flow.collect {}
        }
    }

    try {
        runCurrent()
        block.invoke()
    } finally {
        observationJobs.forEach { it.cancel() }
    }
}