scala - Akka Http 测试 Gzip 响应

Question

所以，我有一组用 scala 编写的 Akka Http 路由。看起来像这样

val route: Route = {
  handleRejections(PrimaryRejectionHandler.handler) {
    handleExceptions(PrimaryExceptionHandler.handler) {
      cors() {
        encodeResponseWith(Gzip) {
          pathPrefix("v1") {
            new v1Routes().handler
          } ~
            path("ping") {
              complete("pong")
            }
        }
      }
    }
  }
}

现在我想使用 scala-test 和 akka testkit 来测试它。

class HttpRouteTest extends WordSpec with Matchers with ScalatestRouteTest {

  "GET /ping" should {

    "return 200 pong" in new Context {
      Get("/ping") ~> httpRoute ~> check {
        responseAs[String] shouldBe "pong"
        status.intValue() shouldBe 200
      }
    }
  }

  trait Context {
    val httpRoute: Route = new HttpRoute().route
  }

}

现在，我在路由中使用 gzip 对我的响应进行编码，当它试图转换为字符串时，测试变得乱码。结果测试没有通过。

有什么解决办法吗？提前致谢。

Answer 1

对于任何碰到这个的人。

这就是我解决问题的方法。首先，我构建了与它正在测试的模块名称相同的单元测试包。

我做了一个 BaseService 将在所有测试中使用，看起来像这样

trait BaseServiceTest extends WordSpec with Matchers with ScalatestRouteTest with MockitoSugar {

  def awaitForResult[T](futureResult: Future[T]): T =
    Await.result(futureResult, 5.seconds)

  def decodeResponse(response: HttpResponse): HttpResponse = {
    val decoder = response.encoding match {
      case HttpEncodings.gzip ⇒
        Gzip
      case HttpEncodings.deflate ⇒
        Deflate
      case HttpEncodings.identity ⇒
        NoCoding
    }

    decoder.decodeMessage(response)
  }
}

然后使用它，我像这样写了我的测试

class UserTest extends BaseServiceTest {

  "GET /user" should {

    "return user details with 200 code" in new Context {

      Get("/") ~> userRoute ~> check {
        val decodedResponse = getBody(decodeResponse(response))

        decodedResponse.user.name.isDefined shouldBe true
        decodedResponse.user.age.isDefined shouldBe true
        decodedResponse.user.city.isDefined shouldBe true
        status.intValue() shouldBe 200
      }
    }
  }

  trait Context {
    val userRoute: Route = UserRoute.route
  }

  def getBody(resp: HttpResponse): UserResponse = {

    import UserResponseJsonProtocol._ // Using spray-json for marshalling protocols

    Await.result(Unmarshal(resp).to[UserResponse], 10.seconds)
  }
}

希望这可以帮助。谢谢！

Answer 2

目前akka-http不提供响应客户端的自动解码，测试套件似乎也是如此。

这意味着如果您需要自己添加解压处理。也就是说，大部分实际的解码代码已经捆绑在 akka 中，您只需要 [ Akka HTTP 文档] 中描述的一些胶水代码。