Myujth

When I add a coroutine delay() in my view model, the remaining part of the code will not be executed.

This is my demo code:

class SimpleViewModel : ViewModel(), CoroutineScope 

 override val coroutineContext: CoroutineContext
 get() = Dispatchers.Unconfined

 var data = 0

 fun doSomething() 
 launch 
 delay(1000)
 data = 1
 
 


class ScopedViewModelTest 

 @Test
 fun coroutineDelay() 
 // Arrange
 val viewModel = SimpleViewModel()

 // ActTes
 viewModel.doSomething()

 // Assert
 Assert.assertEquals(1, viewModel.data)
 

I got the assertion result:

java.lang.AssertionError: 
Expected :1
Actual :0

Any idea how to fix this?

You start a coroutine which suspends for 1 second before setting data to 1. Your test just invokes doSomething but does not wait until data is actually being set. If you add another, longer delay, to the test it will, work:

@Test 
fun coroutineDelay() = runBlocking 
 ...
 viewModel.doSomething()
 delay(1100)
 ...

You can also make the coroutine return a Deferred which you can wait on:

fun doSomething(): Deferred<Unit> 
 return async 
 delay(1000)
 data = 1
 

With await there's no need to delay your code anymore:

val model = SimpleViewModel()
model.doSomething().await()

The first issue in your code is that SimpleViewModel.coroutineContext has no Job associated with it. The whole point of making your view model a CoroutineScope is the ability to centralize the cancelling of all coroutines it starts. So add the job as follows (note the absence of a custom getter):

class SimpleViewModel : ViewModel(), CoroutineScope 

 override val coroutineContext = Job() + Dispatchers.Unconfined

 var data = 0

 fun doSomething() 
 launch 
 delay(1000)
 data = 1
 
 

Now your test code can ensure it proceeds to the assertions only after all the jobs your view model launched are done:

class ScopedViewModelTest 

 @Test
 fun coroutineDelay() 
 // Arrange
 val viewModel = SimpleViewModel()

 // ActTes
 viewModel.doSomething()

 // Assert
 runBlocking 
 viewModel.coroutineContext[Job]!!.children.forEach it.join() 
 
 Assert.assertEquals(1, viewModel.data)