Android插桩测试中如何获取Lifecycle Owner并创建ViewModel？

针对你的场景，有两种直接可行的方法解决Lifecycle Owner/ViewModelStoreOwner的问题，无需更换测试策略：

方法1：直接使用ViewModelStore构造ViewModelProvider #

ViewModelProvider的第一个构造函数允许你直接传入ViewModelStore和Factory，不需要依赖ViewModelStoreOwner。这种方式简单直接，适合测试不依赖生命周期回调的ViewModel逻辑（比如请求功能）。

代码示例：

// 获取测试上下文
val appContext = InstrumentationRegistry.getInstrumentation().targetContext

// 构建ViewModel依赖链
val retrofitService = RetrofitService.getInstance(appContext)
val mainRepository = MainRepository(retrofitService)
val factory = AppVMFactory(mainRepository)

// 手动创建ViewModelStore并初始化ViewModel
val viewModelStore = ViewModelStore()
val viewVM = ViewModelProvider(viewModelStore, factory).get(AppViewModel::class.java)

// 测试完成后清理资源（可选但推荐）
@After
fun tearDown() {
    viewModelStore.clear()
}

方法2：自定义实现ViewModelStoreOwner + LifecycleOwner #

如果你的ViewModel逻辑依赖生命周期状态（比如需要触发onCleared、或LiveData观察需要活跃生命周期），可以自己实现一个兼具ViewModelStoreOwner和LifecycleOwner的测试类：

class TestViewModelStoreOwner : ViewModelStoreOwner, LifecycleOwner {
    private val viewModelStore = ViewModelStore()
    private val lifecycleRegistry = LifecycleRegistry(this).apply {
        // 设置测试需要的生命周期状态，比如RESUMED
        currentState = Lifecycle.State.RESUMED
    }

    override fun getViewModelStore(): ViewModelStore = viewModelStore
    override fun getLifecycle(): Lifecycle = lifecycleRegistry
}

然后在测试中使用这个类作为Owner：

// 获取上下文并构建依赖
val appContext = InstrumentationRegistry.getInstrumentation().targetContext
val retrofitService = RetrofitService.getInstance(appContext)
val mainRepository = MainRepository(retrofitService)
val factory = AppVMFactory(mainRepository)

// 创建测试用的Owner并初始化ViewModel
val testOwner = TestViewModelStoreOwner()
val viewVM = ViewModelProvider(testOwner, factory).get(AppViewModel::class.java)

额外建议 #

如果你的核心需求是测试ViewModel的业务逻辑而非Retrofit的网络请求，其实可以考虑Mock RetrofitService（比如用Mockito），这样不需要真实Context，甚至可以用本地单元测试而非插桩测试，效率更高。但既然你已经选择插桩测试，上述两种方法都能满足你的需求。

内容的提问来源于stack exchange，提问作者Andrew