Unit test StateFlow mới của chương trình Kotlin

Gần đây đã được giới thiệu lớp học StateFlownhư một phần của chương trình đăng quang Kotlin.

Tôi hiện đang dùng thử nó và gặp sự cố khi cố gắng kiểm tra đơn vị ViewModel của mình. Điều tôi muốn đạt được: kiểm tra xem của tôi StateFlowcó đang nhận tất cả các giá trị trạng thái theo đúng thứ tự trong ViewModel của tôi không.

Mã của tôi như sau:

ViewModel:

class WalletViewModel(private val getUserWallets: GetUersWallets) : ViewModel() {

val userWallet: StateFlow<State<UserWallets>> get() = _userWallets
private val _userWallets: MutableStateFlow<State<UserWallets>> =
    MutableStateFlow(State.Init)

fun getUserWallets() {
    viewModelScope.launch {
        getUserWallets.getUserWallets()
            .onStart { _userWallets.value = State.Loading }
            .collect { _userWallets.value = it }
    }
}

Bài kiểm tra của tôi:

    @Test
fun `observe user wallets ok`() = runBlockingTest {
    Mockito.`when`(api.getAssetWallets()).thenReturn(TestUtils.getAssetsWalletResponseOk())
    Mockito.`when`(api.getFiatWallets()).thenReturn(TestUtils.getFiatWalletResponseOk())

    viewModel.getUserWallets()

    val res = arrayListOf<State<UserWallets>>()
    viewModel.userWallet.toList(res) //doesn't works

    Assertions.assertThat(viewModel.userWallet.value is State.Success).isTrue() //works, last value enmited
}

Truy cập giá trị cuối cùng được phát ra hoạt động. Nhưng những gì tôi muốn kiểm tra là tất cả các giá trị phát ra được phát ra theo đúng thứ tự. với đoạn mã này: viewModel.userWallet.toList(res) //doesn't worksTôi gặp lỗi sau:

java.lang.IllegalStateException: This job has not completed yet
    at kotlinx.coroutines.JobSupport.getCompletionExceptionOrNull(JobSupport.kt:1189)
    at kotlinx.coroutines.test.TestBuildersKt.runBlockingTest(TestBuilders.kt:53)
    at kotlinx.coroutines.test.TestBuildersKt.runBlockingTest$default(TestBuilders.kt:45)
    at WalletViewModelTest.observe user wallets ok(WalletViewModelTest.kt:52)
....

Tôi đoán tôi đang thiếu một cái gì đó rõ ràng. Nhưng không chắc tại sao vì tôi mới bắt đầu với Coroutine và Flow và lỗi này dường như xảy ra khi không sử dụng runBlockingTest mà tôi đã sử dụng.

CHỈNH SỬA: Là một giải pháp tạm thời, tôi đang thử nghiệm nó dưới dạng dữ liệu trực tiếp:

    @Captor
    lateinit var captor: ArgumentCaptor<State<UserWallets>>

    @Mock
    lateinit var walletsObserver: Observer<State<UserWallets>>

    @Test
    fun `observe user wallets ok`() = runBlockingTest {
        viewModel.userWallet.asLiveData().observeForever(walletsObserver)

        viewModel.getUserWallets()

        captor.run {
            Mockito.verify(walletsObserver, Mockito.times(3)).onChanged(capture())
            Assertions.assertThat(allValues[0] is State.Init).isTrue()
            Assertions.assertThat(allValues[1] is State.Loading).isTrue()
            Assertions.assertThat(allValues[2] is State.Success).isTrue()
        }
    }

runBlockingTestchỉ bỏ qua sự chậm trễ trong trường hợp của bạn nhưng không ghi đè người điều phối được sử dụng trong ViewModel bằng người điều phối thử nghiệm của bạn. Bạn cần phải đưa TestCoroutineDispatchervào ViewModel của mình hoặc vì bạn đang sử dụng viewModelScope.launch {}mà đã sử dụng Dispatchers.Maintheo mặc định, bạn cần ghi đè người điều phối chính qua Dispatchers.setMain(testCoroutineDispatcher). Bạn có thể tạo và thêm quy tắc sau vào tệp thử nghiệm của mình.

class MainCoroutineRule(
        val testDispatcher: TestCoroutineDispatcher = TestCoroutineDispatcher()
) : TestWatcher() {

    override fun starting(description: Description?) {
        super.starting(description)
        Dispatchers.setMain(testDispatcher)
    }

    override fun finished(description: Description?) {
        super.finished(description)
        Dispatchers.resetMain()
        testDispatcher.cleanupTestCoroutines()
    }
} 

Và trong tệp thử nghiệm của bạn

@get:Rule
var mainCoroutineRule = MainCoroutineRule()

@Test
fun `observe user wallets ok`() = mainCoroutineRule.testDispatcher.runBlockingTest {
}

Btw nó luôn luôn là một thực hành tốt để tiêm điều phối viên. Ví dụ: nếu bạn đang sử dụng một điều phối viên khác với Dispatchers.Mainphạm vi quy trình đăng ký viewModelScope.launch(Dispatchers.Default)của bạn, thì bài kiểm tra của bạn sẽ lại thất bại ngay cả khi bạn đang sử dụng một điều phối viên kiểm tra. Lý do là bạn chỉ có thể ghi đè điều phối viên chính Dispatchers.setMain()vì nó có thể được hiểu từ tên của nó nhưng không Dispatchers.IOhoặc Dispatchers.Default. Trong trường hợp đó, bạn cần phải đưa mainCoroutineRule.testDispatchervào mô hình chế độ xem của mình và sử dụng trình điều phối được tiêm vào thay vì mã hóa cứng nó.