使用 Dagger2 (Robolectric) 进行单元测试的模拟 jetpack ViewModel

Question

所以我正在尝试使用 Robolectric 为我的 Activity 编写一个单元测试，但是我不知道如何提供一个模拟视图模型来查看我的虚拟机是如何在类中直接实例化的。这是因为 jetpack 的生命周期感知 ViewModel 需要一个 Provider 类来实例化。因此，我实质上是在注入自定义提供程序，然后使用它来创建我的 ViewModel。我看过其他示例，但它们似乎都非常令人困惑。我如何做到这一点？

class ActivityEpisodeList : AppCompatActivity() {

        @Inject
        lateinit var vmFactory: ViewModelProvider.Factory

        private lateinit var vm: ActivityViewModel


        override fun onCreate(savedInstanceState: Bundle?) {
            super.onCreate(savedInstanceState)
            setContentView(R.layout.activity_episode_list)

            MvvmDaggerApp.get(this).appComponent.inject(this)

            vm = ViewModelProviders.of(this, vmFactory)[ActivityViewModel::class.java]
    }
}

这就是我创建 ViewModel 的方式：

@Module
abstract class ViewModelModule {

    @Binds
    internal abstract fun bindViewModelFactory(factory: ViewModelFactory): ViewModelProvider.Factory

    @Binds
    @IntoMap
    @ViewModelKey(ActivityViewModel::class)
    internal abstract fun postListViewModel(viewModel: ActivityViewModel): ViewModel

}


@Singleton
class ViewModelFactory @Inject constructor(private val viewModels: MutableMap<Class<out ViewModel>, Provider<ViewModel>>) : ViewModelProvider.Factory {

    override fun <T : ViewModel> create(modelClass: Class<T>): T = viewModels[modelClass]?.get() as T
}

@Target(AnnotationTarget.FUNCTION, AnnotationTarget.PROPERTY_GETTER, AnnotationTarget.PROPERTY_SETTER)
@kotlin.annotation.Retention(AnnotationRetention.RUNTIME)
@MapKey
internal annotation class ViewModelKey(val value: KClass<out ViewModel>)

Answer 1

您应该从 appComponent 用于在此处创建依赖项的模块中注入一个测试视图模型。

不要自己创建视图模型。为 appComponent 创建 2 个模块，一个提供原始依赖项，另一个提供测试/模拟依赖项。像这样 -

@Module
public AppModule {
      public ViewModel appViewModel() { // return original here}
}

  @Module
public TestAppModule extends AppModule {
      public ViewModel appViewModel() { // return test/mock here}
}

在您的测试中，当您创建您的 AppComponent 时通过 TestAppModule 而不是 AppModule，那么您将获得模拟依赖项。