Android блок тестирования потока с несколькими излучателями

Question

У меня есть сценарий использования, который обращается к хранилищу и выдает несколько данных, как показано ниже -

class MyUseCase(private val myRepoInterface: MyRepoInterface) : MyUseCaseInterface {

    override fun performAction(action: MyAction) = flow {
        emit(MyResult.Loading)
        emit(myRepoInterface.getData(action))
    }
}

Модульный тест для того же выглядит ниже -

class MyUseCaseTest {

    @get:Rule
    val instantExecutorRule = InstantTaskExecutorRule()

    @Mock
    private lateinit var myRepoInterface: MyRepoInterface

    private lateinit var myUseCaseInterface: MyUseCaseInterface

    @Before
    fun setUp() {
        MockitoAnnotations.initMocks(this)
        myUseCaseInterface = MyUseCase(myRepoInterface)
    }

    @After
    fun tearDown() {
    }

    @Test
    fun test_myUseCase_returnDataSuccess() {
        runBlocking {
            `when`(myRepoInterface.getData(MyAction.GetData))
                .thenReturn(MyResult.Data("data"))

            // Test
            val flow = myUseCaseInterface.performAction(MyAction.GetData)

            flow.collect { data ->
                Assert.assertEquals(data, MyResult.Loading)
                Assert.assertEquals(data, MyResult.Data("data"))
            }
        }
    }
}

После запуска тест Я получаю эту ошибку -

java.lang.AssertionError: 
Expected : MyResult$Loading@4f32a3ad
Actual   : MyResult.Data("data")

Как проверить вариант использования, который испускает несколько вещей?

Answer 1

Вы можете создать список с помощью оператора toList. Тогда вы можете утверждать в этом списке:

@Test
fun test_myUseCase_returnDataSuccess() = runBlockingTest {
    //Prepare your test here
    //...

    val list = myUseCaseInterface.performAction(MyAction.GetData).toList()
    assertEquals(list, listOf(MyResult.Loading, MyResult.Data("data")))
}