android - Android アーキテクチャ コンポーネント ViewModel - テスト アクティビティで ViewModel をモックする方法は?

Question

GithubBrowserSampleに似た UI テストをセットアップしようとしていますが、サンプル プロジェクトには のモックのみViewModelがFragmentあり、の例はないようですActivity。

Activityをモックしてテストしようとしているコードを次に示しViewModelます。しかし、アクティビティでViewModel前に設定されていません。onCreate()

@RunWith(AndroidJUnit4::class)
class MainActivityTest {

    val viewModel = mock(MainViewModel::class.java)

    @Rule
    @JvmField
    val activityRule = ActivityTestRule<MainActivity>(MainActivity::class.java, true, true)

    private val liveData = MutableLiveData<Resource<Object>>()

    @Before
    open fun setUp() {
        activityRule.activity.viewModelFactory = createViewModelFor(viewModel)
        `when`(viewModel.liveData).thenReturn(liveData)
        viewModel.liveData?.observeForever(mock(Observer::class.java) as Observer<Resource<Object>>)
        liveData.postValue(Resource.success(Object()))
    }

    fun <T : ViewModel> createViewModelFor(model: T): ViewModelProvider.Factory =
        object : ViewModelProvider.Factory {
            override fun <T : ViewModel> create(modelClass: Class<T>): T {
                if (modelClass.isAssignableFrom(model.javaClass)) {
                    return model as T
                }
                throw IllegalArgumentException("unexpected model class " + modelClass)
            }
        }
}

誰かがこの問題について私を助けてくれますか?