在挂起函数中异步调用 ktor http 客户端

问题描述

我有以下使用 ktor 的 HttpClient 的函数。并且 tokenService getAccesstoken 函数也使用 ktor 客户端并获取令牌。

suspend fun addUser(user: User): User {
    return client.post("${Base}api/v1/adduser") {
        contentType(ContentType.Application.Json)
        header("Authorization","Bearer ${tokenService.getAccesstoken()}")
        body = user
    }
}

这是我的简单测试

@Test
fun `should add user`() = runBlocking {
    delay(3000)
    val user = userService.addUser(User("john doe"))
    user.firstName shouldBe "john doe"
}

现在测试由于延迟而通过，延迟为首先获取令牌提供了时间，但是如果我删除延迟，则会调用 addUser 函数而不先获取令牌。我的问题是，是否有更好的方法来做到这一点，以避免增加测试延迟。我查看了 Kotlin Flows，但我不知道如何在我的场景中实现它。如果您能为我指明正确的方向，我将不胜感激。

解决方法

这通常是单元测试的常见问题。想想当您必须向数据库发出请求时的场景，我们不想启动整个数据库只是为了对其进行单元测试，这会使测试变得缓慢而繁琐。我们希望从我们的单元测试中获得快速反馈，而不是到处添加延迟。

一种常见的方法是使用模拟库（如 Mockito 或 Mockk）来模拟网络服务（在您的情况下为 tokenService）。如果您使用 Mockito reference，您可以执行以下操作：

val mockTokenService = Mockito.mock(TokenService::class.java)
Mockito.`when`(mockTokenService.getAccessToken()).thenReturn("sampleToken123")

这样您就可以从单元测试中删除 delay(3000)。然后，如果测试通过，您就会知道您的 addUser 方法业务逻辑将是正确的，并且您可以单独测试令牌服务，如果您愿意，可以使用更复杂的集成测试或其他模拟。>

kotlin kotlin ktor ktor-client