如何使用协程测试ViewModel中的LiveData？遇值为空问题

看起来你遇到的核心问题是协程调度器的硬编码导致测试线程与ViewModel的协程线程不同步，断言在LiveData更新前就执行了，哪怕用了advanceUntilIdle()也没效果。我来帮你拆解问题并给出解决方案：

问题根源 #

你的execute函数硬编码使用了Dispatchers.IO来启动协程：

fun ViewModel.execute(block: () -> Unit) = viewModelScope.launch(Dispatchers.IO) { block() }

而测试中只替换了Main调度器为Unconfined，但IO调度器还是系统默认的后台线程。runTest和advanceUntilIdle()只能管理绑定到测试调度器的协程，后台线程的协程不受测试框架控制，所以断言会在协程完成更新LiveData前执行，导致eventState.value为空。

另外还有个小细节：你的block参数应该定义为suspend () -> Unit，因为useCase是挂起函数（你用了coEvery模拟，说明它是挂起的），否则非挂起的block无法调用挂起函数。

解决方案：依赖注入调度器 + 统一测试调度器 #

我们需要把调度器从ViewModel中解耦，通过构造函数注入，测试时替换为测试专用的调度器，让所有协程都在测试框架的管控下运行。

步骤1：修改ViewModel，注入调度器 #

把硬编码的Dispatchers.IO换成构造参数，默认值保持为Dispatchers.IO不影响正常运行：

class MyViewModel(
    private val useCase: UseCase,
    // 注入IO调度器，默认使用系统IO调度器
    private val ioDispatcher: CoroutineDispatcher = Dispatchers.IO
) : ViewModel() {
    var eventState: MutableLiveData<MyEvents> = MutableLiveData()

    // 修改block为suspend类型，支持调用挂起函数
    fun execute(block: suspend () -> Unit) = viewModelScope.launch(ioDispatcher) { block() }

    fun function(id: String, flag: Boolean) {
        execute {
            useCase(id).fold(
                onSuccess = { 
                    // 测试时用setValue更直接（因为InstantTaskExecutorRule已经同步LiveData更新）
                    eventState.setValue(MyEvents.InitData)
                    // 或者保留postValue也可以，两者在测试环境下差异不大
                    // eventState.postValue(MyEvents.InitData)
                },
                onFailure = { 
                    eventState.setValue(MyEvents.ErrorState)
                }
            )
        }
    }
}

步骤2：修改测试代码，使用统一测试调度器 #

用StandardTestDispatcher替换所有调度器，确保ViewModel的协程和测试代码都在同一个调度器下运行：

@OptIn(ExperimentalCoroutinesApi::class)
class RealEstateDetailViewModelTest {

    @get:Rule
    var rule: InstantTaskExecutorRule = InstantTaskExecutorRule()

    @RelaxedMockK
    lateinit var useCase: UseCase

    // 创建测试专用调度器
    private val testDispatcher = StandardTestDispatcher()
    private lateinit var viewModel: MyViewModel


    @Before
    fun setup() {
        MockKAnnotations.init(this)
        // 替换Main调度器为测试调度器
        Dispatchers.setMain(testDispatcher)
        // 初始化ViewModel时传入测试调度器作为IO调度器
        viewModel = MyViewModel(useCase, testDispatcher)
    }

    @After
    fun onAfter() {
        Dispatchers.resetMain()
    }

    @Test
    fun `viewmodel updates livedata when use case succeeds`() = runTest(testDispatcher) {
        // Given
        val mockDomainModel = mockk<RealEstateDomainModel>(relaxed = true)
        coEvery { useCase("123") } returns Result.success(mockDomainModel)

        // When
        viewModel.function("123", false)
        // 等待测试调度器中所有待执行的协程完成
        advanceUntilIdle()

        // Then
        assertTrue(viewModel.eventState.value is MyEvents.InitData)
    }
}

关键说明 #

1. 依赖注入调度器：让ViewModel不依赖具体的调度器实现，提高可测试性，同时也方便在不同环境（比如后台任务）切换调度器。
2. 统一测试调度器：StandardTestDispatcher会把所有协程任务放入队列，advanceUntilIdle()会等待队列中所有任务执行完毕，确保LiveData已经被更新后再执行断言。
3. setValue vs postValue：因为InstantTaskExecutorRule已经让LiveData的更新在主线程同步执行，测试环境下用setValue更直接，避免异步更新的时序问题。

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