如何測試我注入 Koin 的 viewModel？

Question

我有一個視圖模型。 它在我的數據倉庫中調用了一個函數並返回了狗 object 的列表。

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中的數據。 幫助將不勝感激。

Answer 1

將所有依賴注入 (DI) 框架排除在 ViewModel 之外是一種很好的做法。 這使您的 ViewModel 獨立並簡化了您的單元測試。

視圖模型

因此，您可以通過構造函數傳遞它們，而不是在 ViewModel 中注入您的依賴項：

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

Koin 模塊

在您的 Koin 模塊中，您可以定義依賴項並提供 ViewModel 的實例：

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

然后，在您的 Activity 或 Fragment 中，您可以像往常一樣注入您的 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
    }
}

有了這個，大大簡化了編寫單元測試。 您可以稍后將 Koin 與任何其他 DI 框架交換，而無需接觸您的 ViewModel 和測試。

我希望這對你有所幫助。