测试使用Flow的ViewModel：如何断言loading StateFlow状态变化

首先，你的ViewModel代码存在一个语法错误：onStart的lambda表达式缺少闭合的右括号，先修正这个问题：

fun fetchItems(){
    viewModelScope.launch {
        newsInterface.getStories()
            .flowOn(Dispatchers.IO)
            .onStart { _loading.update { true } } // 补充缺失的闭合括号
            .collect { items ->
                _loading.update { false }   //Loading has finished
            }
    }
}

接下来解决测试中异步断言失败的问题，我们需要使用kotlinx-coroutines-test库来控制协程执行，确保Flow的onStart和collect逻辑完成后再进行断言。

步骤1：添加测试依赖 #

在模块的build.gradle（或build.gradle.kts）中添加协程测试库：

testImplementation "org.jetbrains.kotlinx:kotlinx-coroutines-test:1.7.3"

步骤2：编写loading状态变为true的测试 #

使用runTest包裹测试逻辑，通过测试调度器控制协程执行，并收集StateFlow的发射值进行断言：

@RunWith(JUnit4::class)
class MainViewModelTest {

    private lateinit var viewModel: MainViewModel
    private val testDispatcher = UnconfinedTestDispatcher()

    @Before
    fun setup() {
        // 设置测试调度器替换主线程和IO线程
        Dispatchers.setMain(testDispatcher)
        viewModel = MainViewModel(FakeNetwork())
    }

    @After
    fun tearDown() {
        // 重置主线程调度器
        Dispatchers.resetMain()
    }

    @Test
    fun fetchItems_setsLoadingToTrueOnStart() = runTest {
        // 初始化StateFlow测试收集器
        val loadingTestCollector = viewModel.loading.test()

        viewModel.fetchItems()

        // 跳过初始的false值，断言下一个发射值为true
        loadingTestCollector.awaitItem() // 接收初始值false
        assertEquals(true, loadingTestCollector.awaitItem())

        loadingTestCollector.cancel()
    }
}

步骤3：编写loading状态变为false的测试 #

当Fake仓库的Flow完成数据发射后，ViewModel会将loading设为false，我们可以通过advanceUntilIdle()让所有协程执行到空闲状态，再断言最终的loading值：

@Test
fun fetchItems_setsLoadingToFalseAfterCompletion() = runTest {
    var currentLoadingValue = false
    // 启动协程收集loading状态变化
    val collectJob = launch {
        viewModel.loading.collect { currentLoadingValue = it }
    }

    viewModel.fetchItems()
    // 等待所有协程执行完毕（包括Flow的collect逻辑）
    advanceUntilIdle()

    assertEquals(false, currentLoadingValue)
    collectJob.cancel()
}

关键说明 #

• runTest：提供测试协程作用域，自动管理测试中的协程生命周期。
• UnconfinedTestDispatcher：让协程在当前线程立即执行，便于控制异步逻辑的执行时机。
• StateFlow.test()：扩展函数，方便收集StateFlow的发射值并进行断言；也可以通过手动launch协程收集状态。
• advanceUntilIdle()：让测试调度器执行所有 pending 的协程任务，确保Flow的onStart、emit、collect逻辑全部完成。

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