rust - 如何设计具有测试友好性的 Axum 服务器？

Question

当我尝试使用 axum 构建应用程序时，我未能将框架与我的处理程序分开。使用 Go，经典的方式是定义一个 Interface，实现它并将处理程序注册到框架。通过这种方式，很容易提供一个模拟处理程序来进行测试。但是，我无法使其与 Axum 一起使用。我定义了一个 trait 就像上面一样，但它不会编译:

use std::net::ToSocketAddrs;
use std::sync::{Arc, Mutex};
use serde_derive::{Serialize, Deserialize};
use serde_json::json;
use axum::{Server, Router, Json};

use axum::extract::Extension;
use axum::routing::BoxRoute;
use axum::handler::get;

#[tokio::main]
async fn main() {
    let app = new_router(
        Foo{}
    );
    axum::Server::bind(&"0.0.0.0:3000".parse().unwrap())
        .serve(app.into_make_service())
        .await
        .unwrap();
}

trait Handler {
    fn get(&self, get: GetRequest) -> Result<GetResponse, String>;
}

struct Foo {}

impl Handler for Foo {
    fn get(&self, req: GetRequest) -> Result<GetResponse, String> {
        Ok(GetResponse{ message: "It works.".to_owned()})
    }
}

fn new_router<T:Handler>(handler: T) -> Router<BoxRoute> {
    Router::new()
        .route("/", get(helper))
        .boxed()
}

fn helper<T:Handler>(
    Extension(mut handler): Extension<T>,
    Json(req): Json<GetRequest>
) -> Json<GetResponse> {
    Json(handler.get(req).unwrap())
}

#[derive(Debug, Serialize, Deserialize)]
struct GetRequest {
    // omited
}

#[derive(Debug, Serialize, Deserialize)]
struct GetResponse {
    message: String
    // omited
}

error[E0599]: the method `boxed` exists for struct `Router<axum::routing::Layered<Trace<axum::routing::Layered<AddExtension<Nested<Router<BoxRoute>, Route<axum::handler::OnMethod<fn() -> impl Future {direct}, _, (), EmptyRouter>, EmptyRouter<_>>>, T>>, SharedClassifier<ServerErrorsAsFailures>>>>`, but its trait bounds were not satisfied
   --> src/router.rs:25:10
    |
25  |         .boxed()
    |          ^^^^^ method cannot be called on `Router<axum::routing::Layered<Trace<axum::routing::Layered<AddExtension<Nested<Router<BoxRoute>, Route<axum::handler::OnMethod<fn() -> impl Future {direct}, _, (), EmptyRouter>, EmptyRouter<_>>>, T>>, SharedClassifier<ServerErrorsAsFailures>>>>` due to unsatisfied trait bounds
    | 
   ::: /Users/lebrancebw/.cargo/registry/src/github.com-1ecc6299db9ec823/axum-0.2.5/src/routing/mod.rs:876:1
    |
876 | pub struct Layered<S> {
    | --------------------- doesn't satisfy `<_ as tower_service::Service<Request<_>>>::Error = _`
    |
    = note: the following trait bounds were not satisfied:
            `<axum::routing::Layered<Trace<axum::routing::Layered<AddExtension<Nested<Router<BoxRoute>, Route<axum::handler::OnMethod<fn() -> impl Future {direct}, _, (), EmptyRouter>, EmptyRouter<_>>>, T>>, SharedClassifier<ServerErrorsAsFailures>>> as tower_service::Service<Request<_>>>::Error = _`

我想关键是我的设计显然不是“土气”。有没有办法构建一个易于测试的 Axum 项目？

Answer 1

问题是您要测试什么。我会假设你有一些核心逻辑和一个 HTTP 层。并且您要确保:

1. 请求路由正确；
2. 请求被正确解析；
3. 使用预期参数调用核心逻辑；
4. 并且来自核心的返回值被正确格式化为 HTTP 响应。

要对其进行测试，您需要生成一个模拟了核心逻辑的服务器实例。@lukemathwalker 在他的博客和“Rust 中的零生产”一书中很好地描述了如何通过实际的 TCP 端口生成用于测试的应用程序。它是为 Actix-Web 编写的，但这个想法也适用于 Axum。

你不应该使用 axum::Server::bind，而应该使用 axum::Server::from_tcp 来传递一个 std::net::TcpListner 允许您使用 `TcpListener::bind("127.0.0.1:0") 在任何可用端口上生成测试服务器。

为了使核心逻辑可注入(inject)(和可模拟)，我将其声明为结构并在其上实现所有业务方法。像这样:

pub struct Core {
    public_url: Url,
    secret_key: String,
    storage: Storage,
    client: SomeThirdPartyClient,
}

impl Core {
    pub async fn create_something(
        &self,
        new_something: NewSomething,
    ) -> Result<Something, BusinessErr> {
    ...
}

有了所有这些部分，您就可以编写一个函数来启动服务器:

pub async fn run(listener: TcpListener, core: Core)

这个函数应该封装路由配置、服务器日志配置等等。

可以使用扩展层机制将核心提供给处理程序，如下所示:

...
let shared_core = Arc::new(core);
...
let app = Router::new()
    .route("/something", post(post_something))
    ...
    .layer(AddExtensionLayer::new(shared_core));

在处理程序中可以使用扩展提取器在参数列表中声明:

async fn post_something(
    Extension(core): Extension<Arc<Core>>,
    Json(new_something): Json<NewSomething>,
) -> impl IntoResponse {
    core
        .create_something(new_something)
        .await
}

Axum 示例包含一个关于错误处理和依赖注入(inject)的示例。您可以查看here .

最后但同样重要的是，现在您可以使用 mockall 之类的库来模拟 Core，编写 spawn_app 函数，该函数将返回运行服务器的主机和端口，运行一些针对它的请求并进行断言。

video来自 Let's Get Rusty channel 的 Bogdan 为 mockall 提供了一个良好的开端。

如果您觉得答案中缺少某些内容，我很乐意提供更多详细信息。