ObserveStateTest.kt

@Test
fun `Turn on DND when Pomodoro starts`() = runBlockingTest {
    zenModeApi.mode = ZenMode.Off
    whenever(transitionApi.observeTransitionLog()).thenReturn(
        flowOf(
            Transition(State.IDLE, 0),
            Transition(State.POMODORO, 0),
            Transition(State.DONE, 0)
        )
    )

    observeState().test { cancelAndIgnoreRemainingEvents() }

    verify(zenModeApi).mode = ZenMode.AlarmsOnly
    assertThat(zenModeApi.mode).isEqualTo(ZenMode.Off)
}