如何测试注入koin的viewmodel？

我有一个viewmodel。它调用我的数据repo中的函数，并返回dog对象的列表。

class MainViewModel() : ViewModel() {
    private val dataRepo: DataRepo by inject(DataRepo::class.java) //dataRepo
    private var limit = 10
    private val _dogListLiveData = MutableLiveData<List<Dog>>()
    private var dogList = mutableListOf<Dog>()

    val dogListLiveData: MutableLiveData<List<Dog>>
        get() = _dogListLiveData

    fun searchByBreed(queryText: String) {
        dataRepo.searchByBreed(
            queryText,
            object : DataSource.OnResponseCallback<List<Dog>, String> {
                override fun onSuccess(obj: List<Dog>?) {
                    dogList = mutableListOf()
                    if(!obj.isNullOrEmpty()){
                    dogList.addAll(obj)
                    dogListLiveData.value = dogList.take(limit)
                    }

                }

                override fun onError(error: String) {
                    Log.i("Calling Network Service", error)
                }
            })

    }

    fun loadPaginateBreed() : Boolean{
        return if ((limit+10) < dogList.size) {
            limit += 10
            Log.i("Pagination new Limit", limit.toString())
            dogListLiveData.value = dogList.take(limit)
            false
        }else{
            limit += dogList.size%limit
            dogListLiveData.value = dogList.take(limit)
            true
        }
    }
}

我需要为它编写一个简单的单元测试。我写过这个，也尝试过很多其他的迭代。但似乎什么都不管用。

package com.example.koinapplication.ui.main

import androidx.lifecycle.Observer
import com.example.koinapplication.custom.adpaters.GranularErrorCallAdapterFactory
import com.example.koinapplication.models.Dog
import com.example.koinapplication.models.Height
import com.example.koinapplication.models.Weight
import com.example.koinapplication.repo.*
import org.junit.After
import org.junit.Before
import org.junit.Test

import org.junit.Assert.*
import org.koin.android.ext.koin.androidContext
import org.koin.androidx.viewmodel.dsl.viewModel
import org.koin.core.context.startKoin
import org.koin.core.context.stopKoin
import org.koin.dsl.module
import org.koin.java.KoinJavaComponent.inject
import org.mockito.Mock
import org.mockito.Mockito
import org.mockito.MockitoAnnotations
import java.util.*

class MainViewModelTest {

    private val mainViewModel: MainViewModel by inject(MainViewModel::class.java)

    lateinit var obserserData : Observer<List<Dog>>

    private val networkModule = module {
        factory { AuthInterceptor() }
        factory { provideOkHttpClient(get()) }
        factory { GranularErrorCallAdapterFactory<Any>() }
        single { providesNetworkClient(get(), get()) }
        single { DataRepo(get()) }
        single { NetworkRepo(get()) }
    }

    @Before
    fun setUp() {
        MockitoAnnotations.initMocks(this)
        startKoin {
            modules(arrayListOf(networkModule))
        }
    }

    @After
    fun tearDown() {
        stopKoin()
    }

    @Test
    fun searchByBreed() {
        mainViewModel.dogListLiveData.observeForever { obserserData }
        mainViewModel.searchByBreed("dal")
        Mockito.verify(obserserData).onChanged(mainViewModel.dogListLiveData.value)
    }
}

请帮助我编写一个简单的测试来测试viewmodel中的数据。我们将非常感谢您的帮助。

最好将所有依赖注入（di）框架都排除在viewmodel之外。这使viewmodel独立，并简化了单元测试。
视图模型
因此，您可以通过构造函数传递依赖项，而不是在viewmodel中注入依赖项：

class MainViewModel(
    private val dataRepo: dataRepo
) : ViewModel() {
    // ...
}

koin模块
在koin模块中，您可以定义依赖项并提供viewmodel的示例：

module {
  single { DataRepo(get()) }
  factory { MainViewModel(dataRepo = get())}
}

在活动或片段中，可以像往常一样注入viewmodel。
单元测试
在你的 MainViewModelTest koin不需要任何代码：

class MainViewModelTest {
    private val dataRepo: DataRepo = mockk() // I used Mockk for mocking, but you can use any other mocking framework

    private val mainViewModel = MainViewModel(dataRepo) // your class under test

    @Test
    fun yourTest() {
        // prepare
        every { dataRepo.searchByBreed(...)} returns ...

        mainViewModel.searchByBreed(queryText = "...")

        // do assertions
    }
}

有了它，编写单元测试就简单多了。而且您可以稍后与任何其他di框架交换koin，而无需接触viewmodel和测试。
我希望这对你有点帮助。

如何测试注入koin的viewmodel？

1条答案

相关问题

热门标签

最新问答