测试多Item发射的StateFlow：ViewModel单元测试断言失败排查

你的测试失败核心原因有三个：

1. StateFlow初始状态未处理：StateFlow会在订阅时立即发射初始的UiState(showLoading=false)，你的测试直接断言第一个awaitItem()是loading=true，自然不匹配。
2. 虚拟时间未推进：使用runTest时，delay(100)会被挂起，必须手动推进虚拟时间才能让协程执行到更新loading为false的逻辑。
3. 协程调度器选择问题：UnconfinedTestDispatcher的执行顺序容易混乱，改用StandardTestDispatcher更适合控制测试流程。

修正后的测试代码 #

class TestViewModelTest {

    @MockK
    private lateinit var signInUseCase: SignInUseCase

    private lateinit var viewModel: TestViewModel

    @get:Rule
    val coroutineTestRule = CoroutineTestRule(StandardTestDispatcher())

    @Before
    fun setup() {
        MockKAnnotations.init(this)
        viewModel = TestViewModel(signInUseCase)
    }

    @Test
    fun `test SignInFlow`() = runTest {
        viewModel.uiState.test {
            // 先获取并验证StateFlow的初始状态
            val initialState = awaitItem()
            assert(initialState.showLoading == false)

            // 触发API调用逻辑
            viewModel.simulateAPICall()

            // 执行当前可运行的协程任务，获取loading=true的状态
            runCurrent()
            val loadingState = awaitItem()
            assert(loadingState.showLoading == true)

            // 推进虚拟时间100ms，让delay执行完毕，获取loading=false的状态
            advanceTimeBy(100)
            val completedState = awaitItem()
            assert(completedState.showLoading == false)
        }
    }
}

关键调整说明 #

• 处理初始状态：先调用awaitItem()获取并验证StateFlow的初始值，避免后续断言被初始值干扰。
• 改用StandardTestDispatcher：这个调度器会严格按照虚拟时间执行协程，更容易控制测试流程。
• 手动推进虚拟时间：
  • runCurrent()：执行所有当前可运行的协程任务（这里就是更新loading为true的代码）。
  • advanceTimeBy(100)：推进虚拟时间100ms，让delay(100)执行完毕，触发后续更新loading为false的逻辑。
• 简化断言：用assert替代Assert.assertEquals，更简洁直观（如果需要更详细的错误信息，也可以保留assertEquals）。

额外优化建议 #

如果你的UiState类没有重写equals和hashCode，assertEquals会比较对象引用而非内容，这也可能导致断言失败。确保AuthContract.UiState是data class（data class会自动生成equals和hashCode）：

data class UiState(
    val showLoading: Boolean = false,
    // 其他字段...
) : AuthContract.UiState

内容的提问来源于stack exchange，提问作者Sam