Android:使用协程进行片状的ViewModel单元测试

Question

我有一个虚拟机，例如

class CityListViewModel(private val repository: Repository) : ViewModel() {
@VisibleForTesting
val allCities: LiveData<Resource<List<City>>> =
liveData(context = viewModelScope.coroutineContext + Dispatchers.IO) {
emit(Resource.Loading())
emit(repository.getCities())
}
}

我的测试是：

@ExperimentalCoroutinesApi
class CityListViewModelTest {
@get:Rule
val rule = InstantTaskExecutorRule()
@get:Rule
val coroutineTestRule = CoroutinesTestRule()
@Test
fun `allCities should emit first loading and then a Resource#Success value`() =
runBlockingTest {
val fakeSuccessResource = Resource.Success(
listOf(
City(
1,
"UK",
"London",
Coordinates(34.5, 56.2)
)
)
)
val observer: Observer<Resource<List<City>>> = mock()
val repositoryMock: Repository = mock()
val sut =
CityListViewModel(repositoryMock)
doAnswer { fakeSuccessResource }.whenever(repositoryMock).getCities()
sut.allCities.observeForever(observer)
sut.allCities
val captor = argumentCaptor<Resource<List<City>>>()
captor.run {
verify(observer, times(2)).onChanged(capture())
assertEquals(fakeSuccessResource.data, lastValue.data)
}
}
@Test
fun `allCities should emit first loading and then a Resource#Error value`() =
runBlockingTest {
val fakeErrorResource = Resource.Error<List<City>>("Error")
val observer: Observer<Resource<List<City>>> = mock()
val repositoryMock: Repository = mock()
val sut =
CityListViewModel(repositoryMock)
doAnswer { fakeErrorResource }.whenever(repositoryMock).getCities()
sut.allCities.observeForever(observer)
sut.allCities
val captor = argumentCaptor<Resource<List<City>>>()
captor.run {
verify(observer, times(2)).onChanged(capture())
assertEquals(fakeErrorResource.data, lastValue.data)
}
}
}

我遇到的问题是测试非常不稳定：有时它们都通过，有时一个失败，但我似乎无法找出问题所在。

谢谢！

Answer 1

问题是在测试中，您无法控制 IO 调度程序。我假设你的CoroutinesTestRule是这样的要点？这只会覆盖Dispatchers.Main，但您的CityListViewModel使用Dispatchers.IO。

有几种不同的选择：

1. 在CityListViewModel中，您可以避免显式使用Dispatchers.IO，而只需依赖默认为Dispatchers.Main的viewModelScope。在实际Repository实现中，请确保您的挂起getCities()方法重定向到Dispatchers.IO，即

suspend fun getCities(): List<City> {
withContext(Dispatchers.IO) {
// do work
return cities
}
}

在CityListViewModel：

val allCities: LiveData<Resource<List<City>>> = 
liveData(context = viewModelScope.coroutineContext) {
emit(Resource.Loading())
emit(repository.getCities())
}

在这种情况下，事情将继续像现在一样工作，在你的测试中，模拟Repository会立即返回一个值。

2. 改为注入Dispatchers.IO。如果您使用的是 DI 框架(如 Dagger(，这会更容易，但您基本上可以执行以下操作：

class CityListViewModel(
private val ioDispatcher: CoroutineDispatcher = Dispatchers.IO,
private val repository: Repository
) : ViewModel() {
@VisibleForTesting
val allCities: LiveData<Resource<List<City>>> = 
liveData(context = viewModelScope.coroutineContext + ioDispatcher) {
emit(Resource.Loading())
emit(repository.getCities())
}
}

然后在您的测试中：

val viewModel = CityListViewModel(
ioDispatcher = TestCoroutineDispatcher(),
repository = repository
)

其中任何一个都应该使测试具有确定性。如果您使用的是 Dagger，那么我建议您同时执行这两项操作(创建一个生产模块来提供 Main、IO 和 Default 调度程序，但有一个提供TestCoroutineDispatcher实例的测试模块(，但也要执行选项 1，以确保您的挂起函数将工作定向到另一个调度程序，如果他们正在执行阻塞工作。

Answer 2

我相信最好的方法是将调度程序传递给您的视图模型，因此测试非常容易。 像这里一样： https://stackoverflow.com/a/62629410/2719243