具有延迟的Kotlin协式的单元测试

Question

我正在尝试对使用delay()的Kotlin coroutine进行单元测试。对于单元测试，我不关心delay()，它只是放慢测试的速度。我希望以某种方式运行测试，在调用delay()时不会实际延迟。

class TestUiContext : CoroutineDispatcher(), Delay {
    suspend override fun delay(time: Long, unit: TimeUnit) {
        // I'd like it to call this
    }

    override fun scheduleResumeAfterDelay(time: Long, unit: TimeUnit, continuation: CancellableContinuation<Unit>) {
        // but instead it calls this
    }

    override fun dispatch(context: CoroutineContext, block: Runnable) {
        CommonPool.dispatch(context, block)
    }
}

Answer 1

如果您不希望有任何延迟，为什么不简单地恢复日程呼叫中的继续？：

class TestUiContext : CoroutineDispatcher(), Delay {
    override fun scheduleResumeAfterDelay(time: Long, unit: TimeUnit, continuation: CancellableContinuation<Unit>) {
        continuation.resume(Unit)
    }

    override fun dispatch(context: CoroutineContext, block: Runnable) {
        //CommonPool.dispatch(context, block)  // dispatch on CommonPool
        block.run()  // dispatch on calling thread
    }
}

这样delay()就不会延迟地恢复。请注意，这仍然会延迟挂起，因此其他协程仍然可以运行（如yield())

@Test
fun `test with delay`() {
    runBlocking(TestUiContext()) {
        launch { println("launched") }
        println("start")
        delay(5000)
        println("stop")
    }
}

不延迟运行并打印：

start
launched
stop

编辑：

您可以通过自定义dispatch函数来控制继续运行的位置。