Как протестировать LiveData в ViewModel с помощью сопрограмм ⇐ Android

[Anonymous]

Я пытаюсь добавить модульный тест в свой класс ViewModel, используя JUnit 4 и Mockk, но безуспешно. Чтобы выполнить вызов API, я создал функцию расширения только для создания сопрограммы

Код: Выделить всё

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

В своей ViewModel я просто использую ее, а затем внутри этого блока кода вызываю вариант использования. Функция сгиба является функцией расширения класса Result Kotlin, и тогда я получаю опции onSuccess и OnFailure.

Код: Выделить всё

   fun function() {
execute {
useCase().fold(
onSuccess = { },
onFailure = { }
)
}
}

В любом из этих случаев я обновляю объект LiveData, хранящийся в моей ViewModel.

Код: Выделить всё

    var eventState: MutableLiveData = MutableLiveData()

Код: Выделить всё

   fun function() {
execute {
useCase().fold(
onSuccess = {
eventState.postValue(AnyNewState)
},
onFailure = {
eventState.postValue(AnyNewState)
}
)
}
}

Поскольку это LiveData, я прочитал, что мне следует использовать это правило для тестирования Livedata

Код: Выделить всё

    @get:Rule
val instantTaskExecutorRule = InstantTaskExecutorRule()

и я также добавляю Dispatchers.setMain(Dispatchers.Unconfined) и Dispatchers.resetMain() для использования тестовой сопрограммы

Код: Выделить всё

   @Before
fun setup() {
MockKAnnotations.init(this)
Dispatchers.setMain(Dispatchers.Unconfined)
....
}

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

Мой модульный тест выглядит так

Код: Выделить всё

@OptIn(ExperimentalCoroutinesApi::class)
class RealEstateDetailViewModelTest {

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

@RelaxedMockK
lateinit var useCase: UseCase

private lateinit var viewModel: MyViewModel

@Before
fun setup() {
MockKAnnotations.init(this)
Dispatchers.setMain(Dispatchers.Unconfined)

viewModel = MyViewModel(useCase)
}

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

@Test
fun `viewmodel updates livedata`() =
runTest {
//Given
val result = mockk(relaxed = true)
val resultUiModel = mockk(relaxed = true)

coEvery { useCase(any()) } returns Result.success(result)

//When
viewModel.function("123", false)

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

Но когда я запускаю тест, я получаю сообщение об ошибке, потому что эта строка

Код: Выделить всё

assertTrue(viewModel.eventState.value is MyEvents.InitData)

получает ошибку утверждения, поскольку viewModel.eventState.value имеет значение null. При отладке ViewModel похоже, что утверждение вызывается до того, как viewModel обновит eventState LiveData. Я попробовал использовать заранееUntilIdle(), но все равно получаю сообщение об ошибке.
Я что-то пропустил?
Спасибо!

Подробнее здесь: https://stackoverflow.com/questions/751 ... coroutines