如果通过启动(调度程序){...}

时间:2018-04-11 22:26:43

标签: kotlin executorservice kotlinx.coroutines

我们最近开始在Android应用程序上使用协同程序。一切都很好,花花公子,直到有人写了大致相当于以下功能的东西:

fun example(dispatcher: CoroutineDispatcher, block: () -> Unit) {
    launch(dispatcher) {
        block()
    }
}

我们想编写一个测试,验证block是否使用dispatcher执行。

我们已尝试过以下操作,但这不起作用,因为我们得到NullPointerException,因为dispatcher.parentContext未被模拟。我不喜欢嘲笑它,因为我们不关心它执行的上下文,只是它被执行了:

@Test
fun `test that doesn't work`() {
    val dispatcher: CoroutineDispatcher = mock()
    val block: () -> Unit = mock()

    // fails here, specifically on the call to `launch` in example()
    example(dispatcher, block)

    val captor = argumentCaptor<Runnable>()
    verify(dispatcher).dispatch(any(), captor.capture())
    verify(block, never()).invoke()

    captor.firstArgument.run()
    verify(block).invoke()
}

我们认为这是一个选择,但它仍然感觉不对:

@Test
fun `test that works but doesn't feel right`() {
    val executor: ExecutorService = mock()
    val block: () -> Unit = mock()

    example(executor.asCoroutineDispatcher(), block)

    val captor = argumentCaptor<Runnable>()
    verify(executor).execute(captor.capture())
    verify(block, never()).invoke()

    captor.firstArgument.run()
    verify(block).invoke()
}

任何人都能想出更好的东西吗?

编辑:

值得注意的依赖关系如下:

com.nhaarman:mockito-kotlin:1.5.0
org.jetbrains.kotlinx:kotlinx-coroutines-core:0.22.5
junit:junit:4.12
org.jetbrains.kotlin:kotlin-stdlib:1.2.30

1 个答案:

答案 0 :(得分:0)

由于launch返回了Job,因此您需要等待其完成才能验证已完成的所有操作。

您可以通过两种方式做到这一点:

由于您正在launch进行协程,因此实际的单元测试可能会在启动执行完成之前完成。您需要等待其执行,以确保测试没有尽早完成。您可以通过多种方式做到这一点。

  1. 更改函数签名以使启动作业退回(例如fun example(dispatcher: CoroutineDispatcher, block: () -> Unit) = launch(dispatcher) { ...。这样,您可以在.join()函数上调用example以等待其完成。

  2. >
    1. 保留内部var exampleJob:Job? = null,并通过执行启动来进行设置,例如exampleJob = launch { ...}。这样,在测试中,您可以使用exampleJob.join()等待其完成并随后进行验证。

奖金解决方案:

您还可以使用MockKverifyWithTimeout进行单元测试。这样,您的单元测试将等待验证完成。