如何讓 viewModelScope 在單元測試中等待掛起函數

Question

正如我們所知，viewModelScope 的默認調度程序並行執行。 如何讓 viewModelScope 等待單元測試中的掛起功能。 在我的代碼中，我想從repository.getData()獲取結果，以便我的斷言通過。 現在，我沒有按時得到結果。

class MainViewModel(private val repository: Repository,
                     private val dispatcher: CoroutineDispatcherProvider) : ViewModel() {

    private val viewState = MutableLiveData<Results<Data>>()
    fun getViewState() : LiveData<Results<Data>> = viewState



    fun getData(query: search) {
        viewState.value = Loading
        viewModelScope.launch(dispatcher.main()) {
            val results = repository.getData(query) //need to wait for this
            when(results){
                    is Success -> {
                        viewState.value = Success(results.data)
                    }
                    is Error -> viewState.value = Error(“Error”)
                }
        }

主視圖模型測試：

 Class MainViewModelTest {

    @get:Rule
    val instantExecutorRule = InstantTaskExecutorRule()

    @get:Rule
    val coroutineTestRule = CoroutineTestRule()


    @Test
    fun `test get data`() = coroutineTestRule.testCoroutineDispatcher.runBlockingTest {
        coEvery{repository.getData(“query”)} returns Success(Data)

        var observer:Observer<Results<Data>> = mock()

        viewModel.getViewState().observeForever(observer)

        viewModel.getData(“query”)

          assertEquals(Loading, state)
          assertEquals(resultSuccess, state)

       }

    }

Answer 1

我認為您應該使用observer.onChanged()進行斷言和驗證。 否則我看不出有什么問題。

val successResult = Result.Success(Data)
coEvery{repository.getData(“query”)} returns successResult

verify { observer.onChanged(Result.Loading) }
verify { observer.onChanged(successResult) }