Add assertThatAndStateTurbineConsumed and assertThatAndEffectTurbineConsumed for simpler test setup (c29bc905) · Commits · e / os / Mail

core/ui/compose/testing/src/main/kotlin/app/k9mail/core/ui/compose/testing/mvi/MviTurbineExtension.kt

+30 −0

Original line number	Diff line number	Diff line
		@@ -72,6 +72,36 @@ fun <T, STATE, EFFECT> assertThatAndMviTurbinesConsumed(
		turbines.effectTurbine.ensureAllEventsConsumed()
		}

		/**
		* The `assertThatAndStateTurbineConsumed` function ensures that the assertion passed and
		* all events in the state turbine have been consumed.
		*/
		suspend fun <STATE, EFFECT> MviTurbines<STATE, EFFECT>.assertThatAndStateTurbineConsumed(
		assertion: Assert<STATE>.() -> Unit,
		) {
		assertThat(stateTurbine.awaitItem()).all {
		assertion()
		}

		stateTurbine.ensureAllEventsConsumed()
		effectTurbine.ensureAllEventsConsumed()
		}

		/**
		* The `assertThatAndEffectTurbineConsumed` function ensures that the assertion passed and
		* all events in the effect turbine have been consumed.
		*/
		suspend fun <STATE, EFFECT> MviTurbines<STATE, EFFECT>.assertThatAndEffectTurbineConsumed(
		assertion: Assert<EFFECT>.() -> Unit,
		) {
		assertThat(effectTurbine.awaitItem()).all {
		assertion()
		}

		stateTurbine.ensureAllEventsConsumed()
		effectTurbine.ensureAllEventsConsumed()
		}

		/**
		* A container class for the state and effect turbines of an MVI ViewModel.
		*/