MVVMTesting7.kt

class MainViewModelTest {

    @get:Rule
    val mainCoroutineRule = MainCoroutineRule()

    private val userServices = mock<UserServices>()

    private lateinit var viewModel: MainViewModel

    private val testDispatcher = AppDispatchers(
        IO = TestCoroutineDispatcher(),
        MAIN = Dispatchers.Unconfined
    )

    @Test
    fun `Failure state works`() = runBlocking {
        whenever(userServices.getUsers()).thenThrow(RuntimeException("Error"))
        viewModel = MainViewModel(userServices, testDispatcher)
        Assert.assertEquals(State.FAILURE("Error"), viewModel.state.value)
    }
}