编写 Kotlin 中 ViewModel 的单元测试案例可以使用以下步骤和示例代码。
首先,确保你的项目中已添加了 ViewModel 和 LiveData 的依赖项。在 Gradle 文件中添加以下依赖项:
androidTestImplementation 'androidx.arch.core:core-testing:$version'
testImplementation 'androidx.test.ext:junit:$version'
接下来,在测试类中创建一个 ViewModel 对象和相关的依赖项。你可以使用 Mockito
或 MockK
等框架来模拟依赖项。下面是一个使用 Mockito
的示例:
import androidx.arch.core.executor.testing.InstantTaskExecutorRule
import org.junit.Assert.assertEquals
import org.junit.Before
import org.junit.Rule
import org.junit.Test
import org.mockito.Mock
import org.mockito.Mockito.`when`
import org.mockito.MockitoAnnotations
class MyViewModelTest {
// 添加 InstantTaskExecutorRule 规则以支持 LiveData 的测试
@get:Rule
val rule = InstantTaskExecutorRule()
// 创建一个模拟的 Repository
@Mock
lateinit var mockRepository: MyRepository
// 要测试的 ViewModel
lateinit var myViewModel: MyViewModel
@Before
fun setup() {
// 初始化 Mockito
MockitoAnnotations.initMocks(this)
// 创建 ViewModel 实例并将模拟的 Repository 传入
myViewModel = MyViewModel(mockRepository)
}
@Test
fun testGetData() {
// 模拟 Repository 的行为
`when`(mockRepository.getData()).thenReturn("Hello, World!")
// 调用 ViewModel 的方法
val result = myViewModel.getData()
// 验证结果是否符合预期
assertEquals("Hello, World!", result)
}
}
在上述示例中,我们使用了 Mockito
框架来创建一个模拟的 MyRepository
对象,并设置其 getData()
方法返回 "Hello, World!"。然后,我们调用 ViewModel 的 getData()
方法,并断言结果是预期的。
请注意,我们还添加了 InstantTaskExecutorRule
规则以支持对 LiveData 的测试。这是因为 LiveData 通常在后台线程上执行操作,如果没有该规则,测试可能会立即返回,而不会等待 LiveData 的结果。
这只是一个简单的示例,你可以根据你的具体需求扩展它,并使用其他框架和工具来模拟和测试 ViewModel。