如何测试Kotlin中withTimeout代码块的超时分支？

业务代码 #

try {
  withTimeout(200) {
    val response = datafetcher.fetchData(request)
  }
} catch (TimeoutCancellationException e) {
  // 返回附加错误信息的响应
}

错误的测试尝试 #

runTest {
  whenever(mockdatafetcher.fetchData(any())).thenAnswer(
    runBlocking {
      delay(7000)
      defaultResponse
    }
  )
}

问题根源 #

• runBlocking会创建独立的不可取消协程上下文，withTimeout发出的超时取消信号无法传递到这个上下文里的delay，导致代码会一直等到delay执行完毕才返回结果，永远触发不了TimeoutCancellationException。
• 普通的thenAnswer不支持挂起函数逻辑，无法正确衔接测试协程的可取消上下文。

正确的测试方案 #

1. 使用coAnswer处理挂起函数Mock #

引入mockito-kotlin库（支持Kotlin协程的Mock扩展），用coAnswer替代thenAnswer，直接在测试协程上下文里编写挂起逻辑：

runTest {
  // 用coAnswer mock挂起函数，继承测试协程的可取消上下文
  whenever(mockdatafetcher.fetchData(any())).coAnswer {
    delay(7000)
    defaultResponse
  }

  // 执行被测试的业务逻辑
  val result = yourService.handleRequest(request)

  // 断言结果为超时后的错误响应
  assertEquals(expectedTimeoutErrorResponse, result)
}

2. 结合虚拟时间加速测试 #

runTest默认使用虚拟调度器，无需等待真实时间，通过advanceTimeBy主动推进时间触发超时：

runTest {
  whenever(mockdatafetcher.fetchData(any())).coAnswer {
    delay(7000) // 虚拟时间，不会真等待7秒
    defaultResponse
  }

  // 异步启动业务逻辑，避免阻塞测试线程
  val resultDeferred = async { yourService.handleRequest(request) }

  // 推进虚拟时间到超时阈值（200ms），触发withTimeout的取消逻辑
  advanceTimeBy(200)

  // 获取结果并断言
  val result = resultDeferred.await()
  assertEquals(expectedTimeoutErrorResponse, result)
}

关键注意事项 #

• 必须使用coAnswer来Mock挂起函数，普通thenAnswer无法处理协程挂起逻辑。
• 避免在Mock逻辑中使用runBlocking，否则会脱离测试协程的可取消上下文。
• 利用runTest的虚拟时间特性，大幅提升测试效率，无需等待真实耗时。

内容的提问来源于stack exchange，提问作者mulafar p