时间戳

首页 > 解决方案 > 在挂起函数中异步调用 ktor http 客户端

kotlin - 在挂起函数中异步调用 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，但我不知道如何为我的场景实现它。如果你能指出我正确的方向，我将不胜感激。

标签： kotlinktorktor-client

解决方案

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

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

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

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

推荐阅读