如何将协程作为单元测试的阻塞？

Question

我已经开始为我的MVP Android项目编写单元测试，但是我的依赖于协程的测试间歇性地失败了（通过记录和调试，我确认有时会提前进行验证delay，当然要添加此修复程序）

我已经尝试过使用，runBlocking并且Dispatchers.setMain(mainThreadSurrogate)从中发现了org.jetbrains.kotlinx:kotlinx-coroutines-test，但是到目前为止，尝试进行如此多的组合并没有取得任何成功。

abstract class CoroutinePresenter : Presenter, CoroutineScope {
    private lateinit var job: Job

    override val coroutineContext: CoroutineContext
        get() = job + Dispatchers.Main

    override fun onCreate() {
        super.onCreate()
        job = Job()
    }

    override fun onDestroy() {
        super.onDestroy()
        job.cancel()
    }
}

class MainPresenter @Inject constructor(private val getInfoUsecase: GetInfoUsecase) : CoroutinePresenter() {
    lateinit var view: View

    fun inject(view: View) {
        this.view = view
    }

    override fun onResume() {
        super.onResume()

        refreshInfo()
    }

    fun refreshInfo() = launch {
        view.showLoading()
        view.showInfo(getInfoUsecase.getInfo())
        view.hideLoading()
    }

    interface View {
        fun showLoading()
        fun hideLoading()

        fun showInfo(info: Info)
    }
}

class MainPresenterTest {
    private val mainThreadSurrogate = newSingleThreadContext("Mocked UI thread")

    private lateinit var presenter: MainPresenter
    private lateinit var view: MainPresenter.View

    val expectedInfo = Info()

    @Before
    fun setUp() {
        Dispatchers.setMain(mainThreadSurrogate)

        view = mock()

        val mockInfoUseCase = mock<GetInfoUsecase> {
            on { runBlocking { getInfo() } } doReturn expectedInfo
        }

        presenter = MainPresenter(mockInfoUseCase)
        presenter.inject(view)
        presenter.onCreate()
    }

    @Test
    fun onResume_RefreshView() {
        presenter.onResume()

        verify(view).showLoading()
        verify(view).showInfo(expectedInfo)
        verify(view).hideLoading()
    }

    @After
    fun tearDown() {
        Dispatchers.resetMain()
        mainThreadSurrogate.close()
    }
}

我认为这些runBlocking块应强制所有子代coroutineScopes在同一线程上运行，并迫使它们在继续验证之前完成。

Answer 1

在CoroutinePresenter课堂上，您正在使用Dispatchers.Main。您应该能够在测试中进行更改。尝试执行以下操作：

1. 将uiContext: CoroutineContext参数添加到演示者的构造函数中：

   abstract class CoroutinePresenter(private val uiContext: CoroutineContext = Dispatchers.Main) : CoroutineScope {
   private lateinit var job: Job
   
   override val coroutineContext: CoroutineContext
       get() = uiContext + job
   
   //...
   }
   
   class MainPresenter(private val getInfoUsecase: GetInfoUsecase, 
                       private val uiContext: CoroutineContext = Dispatchers.Main 
   ) : CoroutinePresenter(uiContext) { ... }
   

2. 更改MainPresenterTest类以注入另一个CoroutineContext：

   class MainPresenterTest {
       private lateinit var presenter: MainPresenter
   
       @Mock
       private lateinit var view: MainPresenter.View
   
       @Mock
       private lateinit var mockInfoUseCase: GetInfoUsecase
   
       val expectedInfo = Info()
   
       @Before
       fun setUp() {
           // Mockito has a very convenient way to inject mocks by using the @Mock annotation. To
           // inject the mocks in the test the initMocks method needs to be called.
           MockitoAnnotations.initMocks(this)
   
           presenter = MainPresenter(mockInfoUseCase, Dispatchers.Unconfined) // here another CoroutineContext is injected 
           presenter.inject(view)
           presenter.onCreate()
   }
   
       @Test
       fun onResume_RefreshView() = runBlocking {
           Mockito.`when`(mockInfoUseCase.getInfo()).thenReturn(expectedInfo)
   
           presenter.onResume()
   
           verify(view).showLoading()
           verify(view).showInfo(expectedInfo)
           verify(view).hideLoading()
       }
   }