如何测试含协程逻辑的ViewModel，确保断言前协程执行完毕？

核心问题根因 #

你遇到的返回null问题和InstantTaskExecutorRule无关，核心原因是LiveData赋值逻辑不符合线程要求：

• 只有在主线程调用LiveData.value赋值才会生效，在其他线程调用该API会抛出异常，而异常会被viewModelScope默认的异常处理器静默捕获，没有崩溃提示，最终导致LiveData没有被赋值，拿到null值。
• 你的测试代码注入了TestCoroutineDispatcher作为调度器，默认没有绑定到主线程，所以赋值时触发了上述异常。

修复步骤 #

1. 修正LiveData赋值逻辑 #

两种方案二选一即可：

fun doSomething() {
    viewModelScope.launch(dispatcher) {
        // 你的Retrofit挂起函数逻辑
        
        // 方案1：跨线程直接用postValue赋值，不需要关心当前线程
        _livedata.postValue(true)
        
        // 方案2：切到主线程再用value赋值
        // withContext(Dispatchers.Main) {
        //     _livedata.value = true
        // }
    }
}

2. 简化测试代码 #

你现在的MainCoroutineRule实现是正确的，也已经完成了调度器注入，不需要额外修改，测试代码可以直接简化为：

// 插桩测试可以删掉InstantTaskExecutorRule，该规则仅用于本地JVM单元测试
class MyViewModelTest {
    private lateinit var viewModel: MyViewModel

    @get:Rule
    var mainCoroutineRule = MainCoroutineRule()

    @Before
    fun setup() {
        viewModel = MyViewModel(mainCoroutineRule.dispatcher)
    }

    @Test
    fun testMyViewModel() = mainCoroutineRule.runBlockingTest {
        viewModel.doSomething()
        // runBlockingTest会自动执行所有已调度的协程，不需要手动调用advanceUntilIdle
        val result = viewModel.livedata.getOrAwaitValue()
        assertThat(result).isTrue()
    }
}

常见疑问说明 #

• 插桩测试确实不需要InstantTaskExecutorRule：该规则的作用是把Architecture Components的异步任务执行器替换为同步实现，仅适用于没有Android主线程环境的本地JVM测试，插桩测试运行在真机/模拟器上有原生主线程调度，删除该规则即可。
• 你当前的调度器注入方案已经是最佳实践：不需要额外替换为所谓的"可阻塞主线程的调度器"，TestCoroutineDispatcher本身就支持控制协程执行时序，只要赋值逻辑正确就可以正常拿到测试结果。
• 如果使用kotlinx-coroutines-test 1.6.0+版本，建议将废弃的TestCoroutineDispatcher和runBlockingTest替换为StandardTestDispatcher和runTest，兼容新的测试API规范。

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