如何在androidTest上正确模拟ViewModel

我目前正在为一个片段编写一些UI单元测试，其中一个@Test是查看是否正确显示对象列表，这不是集成测试，因此我想模拟ViewModel。

片段的变量：

class FavoritesFragment : Fragment() {
    private lateinit var adapter: FavoritesAdapter
    private lateinit var viewModel: FavoritesViewModel
    @Inject lateinit var viewModelFactory: FavoritesViewModelFactory
    (...)

代码如下：

@MediumTest
@RunWith(AndroidJUnit4::class)
class FavoritesFragmentTest {
    @Rule @JvmField val activityRule = ActivityTestRule(TestFragmentActivity::class.java, true, true)
    @Rule @JvmField val instantTaskExecutorRule = InstantTaskExecutorRule()
    private val results = MutableLiveData<Resource<List<FavoriteView>>>()
    private val viewModel = mock(FavoritesViewModel::class.java)
    private lateinit var favoritesFragment: FavoritesFragment
    @Before
    fun setup() {
        favoritesFragment = FavoritesFragment.newInstance()
        activityRule.activity.addFragment(favoritesFragment)
        `when`(viewModel.getFavourites()).thenReturn(results)
    }
    (...)
    // This is the initial part of the test where I intend to push to the view
    @Test
    fun whenDataComesInItIsCorrectlyDisplayedOnTheList() {
        val resultsList = TestFactoryFavoriteView.generateFavoriteViewList()
        results.postValue(Resource.success(resultsList))
        (...)
    }

我能够嘲笑ViewModel但当然，这与Fragment内部创建的ViewModel不同。

所以我的问题真的，有人成功地做到了这一点，或者有一些可能帮助我的指示/参考？

另外，我尝试查看谷歌样本，但没有运气。
作为参考，该项目可以在这里找到：https://github.com/JoaquimLey/transport-eta/

在测试设置中，您需要提供正在注入片段的 FavoritesViewModelFactory 的测试版本。

您可以执行以下操作，其中需要将模块添加到 TestAppComponent：

@Module
object TestFavoritesViewModelModule {
    val viewModelFactory: FavoritesViewModelFactory = mock()
    @JvmStatic
    @Provides
    fun provideFavoritesViewModelFactory(): FavoritesViewModelFactory {
        return viewModelFactory
    }
}

然后，您可以在测试中提供模拟视图模型。

fun setupViewModelFactory() {
    whenever(TestFavoritesViewModelModule.viewModelFactory.create(FavoritesViewModel::class.java)).thenReturn(viewModel)
}

我已经使用Dagger注入的额外对象解决了这个问题，你可以在这里找到完整的例子：https://github.com/fabioCollini/ArchitectureComponentsDemo

在我没有直接使用 ViewModelFactory 的片段中，我定义了一个定义为 Dagger 单例的自定义工厂：https://github.com/fabioCollini/ArchitectureComponentsDemo/blob/master/uisearch/src/main/java/it/codingjam/github/ui/search/SearchFragment.kt

然后在测试中，我使用 DaggerMock 替换这个自定义工厂，使用始终返回模拟而不是真实视图模型的工厂：https://github.com/fabioCollini/ArchitectureComponentsDemo/blob/master/uisearchTest/src/androidTest/java/it/codingjam/github/ui/repo/SearchFragmentTest.kt

看起来，你使用 kotlin 和 koin(1.0-beta(。这是我嘲笑的决定

@RunWith(AndroidJUnit4::class)
class DashboardFragmentTest : KoinTest {
@Rule
@JvmField
val activityRule = ActivityTestRule(SingleFragmentActivity::class.java, true, true)
@Rule
@JvmField
val executorRule = TaskExecutorWithIdlingResourceRule()
@Rule
@JvmField
val countingAppExecutors = CountingAppExecutorsRule()
private val testFragment = DashboardFragment()
private lateinit var dashboardViewModel: DashboardViewModel
private lateinit var router: Router
private val devicesSuccess = MutableLiveData<List<Device>>()
private val devicesFailure = MutableLiveData<String>()
@Before
fun setUp() {
    dashboardViewModel = Mockito.mock(DashboardViewModel::class.java)
    Mockito.`when`(dashboardViewModel.devicesSuccess).thenReturn(devicesSuccess)
    Mockito.`when`(dashboardViewModel.devicesFailure).thenReturn(devicesFailure)
    Mockito.`when`(dashboardViewModel.getDevices()).thenAnswer { _ -> Any() }
    router = Mockito.mock(Router::class.java)
    Mockito.`when`(router.loginActivity(activityRule.activity)).thenAnswer { _ -> Any() }
    StandAloneContext.loadKoinModules(hsApp + hsViewModel + api + listOf(module {
        single(override = true) { router }
        factory(override = true) { dashboardViewModel } bind ViewModel::class
    }))
    activityRule.activity.setFragment(testFragment)
    EspressoTestUtil.disableProgressBarAnimations(activityRule)
}
@After
fun tearDown() {
    activityRule.finishActivity()
    StandAloneContext.closeKoin()
}
@Test
fun devicesSuccess(){
    val list = listOf(Device(deviceName = "name1Item"), Device(deviceName = "name2"), Device(deviceName = "name3"))
    devicesSuccess.postValue(list)
    onView(withId(R.id.rv_devices)).check(ViewAssertions.matches(ViewMatchers.isCompletelyDisplayed()))
    onView(withId(R.id.rv_devices)).check(matches(hasDescendant(withText("name1Item"))))
    onView(withId(R.id.rv_devices)).check(matches(hasDescendant(withText("name2"))))
    onView(withId(R.id.rv_devices)).check(matches(hasDescendant(withText("name3"))))
}
@Test
fun devicesFailure(){
    devicesFailure.postValue("error")
    onView(withId(R.id.rv_devices)).check(ViewAssertions.matches(ViewMatchers.isCompletelyDisplayed()))
    Mockito.verify(router, times(1)).loginActivity(testFragment.activity!!)
}
@Test
fun devicesCall() {
    onView(withId(R.id.rv_devices)).check(ViewAssertions.matches(ViewMatchers.isCompletelyDisplayed()))
    Mockito.verify(dashboardViewModel, Mockito.times(1)).getDevices()
}

}

在您提供的示例中，您使用 mockito 返回视图模型的特定实例的模拟，而不是每个实例的模拟。

为了完成这项工作，您必须让您的片段使用您创建的确切视图模型模拟。

这很可能来自商店或存储库，所以你可以把你的模拟放在那里？这实际上取决于您如何在片段逻辑中设置视图模型的获取。

建议：1( 模拟构建视图模型的数据源或2( 添加一个 fragment.setViewModel(( 并将其标记为仅用于测试。这有点丑陋，但如果你不想模拟数据源，这种方式很容易。

人们可以轻松地模拟ViewModel和其他没有Dagger的对象，只需：

创建一个包装类，该类可以将调用重新路由到 ViewModelProvider。下面是包装类的生产版本，它只是将调用传递给作为参数传入的实际 ViewModelProvider。
```
class VMProviderInterceptorImpl : VMProviderInterceptor { override fun get(viewModelProvider: ViewModelProvider, x: Class<out ViewModel>): ViewModel {
    return viewModelProvider.get(x)
}
```
}
将此包装对象的 getter 和 setter 添加到应用程序类中。
在活动规则中，在启动活动之前，将实际包装器换成模拟包装器，该包装器不会将 get ViewModel 调用路由到实际 viewModelProvider，而是提供模拟对象。

我意识到这不像dagger那么强大，但简单性很有吸引力。

相关内容

最新更新

热门标签：