tokio-rs/axum
1
0
Fork 0
mirror of https://github.com/tokio-rs/axum.git synced 2026-08-24 00:00:16 +02:00
Code Issues Packages Projects Releases Wiki Activity
Files
8ef96f219958d07e92f1147029de7eb7d3be8cfd
axum/src/tests/nest.rs
T

219 lines
5.6 KiB
Rust
Raw Normal View History

Fix stripping prefix when nesting at / (#91)
2021-08-02 22:40:33 +02:00
use super::*;
Make sure nested services still see full URI (#166)
2021-08-08 17:27:23 +02:00
use crate::body::box_body;
Implement path extractor (#124)
2021-08-06 16:17:57 +08:00
use std::collections::HashMap;
Fix stripping prefix when nesting at / (#91)
2021-08-02 22:40:33 +02:00
#[tokio::test]
async fn nesting_apps() {
let api_routes = route(
"/users",
get(|| async { "users#index" }).post(|| async { "users#create" }),
)
.route(
"/users/:id",
Implement path extractor (#124)
2021-08-06 16:17:57 +08:00
get(
|params: extract::Path<HashMap<String, String>>| async move {
format!(
"{}: users#show ({})",
params.get("version").unwrap(),
params.get("id").unwrap()
)
},
),
Fix stripping prefix when nesting at / (#91)
2021-08-02 22:40:33 +02:00
)
.route(
"/games/:id",
Implement path extractor (#124)
2021-08-06 16:17:57 +08:00
get(
|params: extract::Path<HashMap<String, String>>| async move {
format!(
"{}: games#show ({})",
params.get("version").unwrap(),
params.get("id").unwrap()
)
},
),
Fix stripping prefix when nesting at / (#91)
2021-08-02 22:40:33 +02:00
);
let app = route("/", get(|| async { "hi" })).nest("/:version/api", api_routes);
let addr = run_in_background(app).await;
let client = reqwest::Client::new();
let res = client
.get(format!("http://{}/", addr))
.send()
.await
.unwrap();
assert_eq!(res.status(), StatusCode::OK);
assert_eq!(res.text().await.unwrap(), "hi");
let res = client
.get(format!("http://{}/v0/api/users", addr))
.send()
.await
.unwrap();
assert_eq!(res.status(), StatusCode::OK);
assert_eq!(res.text().await.unwrap(), "users#index");
let res = client
.get(format!("http://{}/v0/api/users/123", addr))
.send()
.await
.unwrap();
assert_eq!(res.status(), StatusCode::OK);
assert_eq!(res.text().await.unwrap(), "v0: users#show (123)");
let res = client
.get(format!("http://{}/v0/api/games/123", addr))
.send()
.await
.unwrap();
assert_eq!(res.status(), StatusCode::OK);
assert_eq!(res.text().await.unwrap(), "v0: games#show (123)");
}
#[tokio::test]
async fn wrong_method_nest() {
let nested_app = route("/", get(|| async {}));
let app = crate::routing::nest("/", nested_app);
let addr = run_in_background(app).await;
let client = reqwest::Client::new();
let res = client.get(format!("http://{}", addr)).send().await.unwrap();
assert_eq!(res.status(), StatusCode::OK);
let res = client
.post(format!("http://{}", addr))
.send()
.await
.unwrap();
assert_eq!(res.status(), StatusCode::METHOD_NOT_ALLOWED);
let res = client
.patch(format!("http://{}/foo", addr))
.send()
.await
.unwrap();
assert_eq!(res.status(), StatusCode::NOT_FOUND);
}
#[tokio::test]
async fn nesting_at_root() {
let app = nest("/", get(|uri: Uri| async move { uri.to_string() }));
let addr = run_in_background(app).await;
let client = reqwest::Client::new();
let res = client.get(format!("http://{}", addr)).send().await.unwrap();
assert_eq!(res.status(), StatusCode::OK);
assert_eq!(res.text().await.unwrap(), "/");
let res = client
.get(format!("http://{}/foo", addr))
.send()
.await
.unwrap();
assert_eq!(res.status(), StatusCode::OK);
assert_eq!(res.text().await.unwrap(), "/foo");
let res = client
.get(format!("http://{}/foo/bar", addr))
.send()
.await
.unwrap();
assert_eq!(res.status(), StatusCode::OK);
assert_eq!(res.text().await.unwrap(), "/foo/bar");
}
Fix Uri extractor not being the full URI if using nest (#156)
2021-08-07 22:07:50 +02:00
#[tokio::test]
async fn nested_url_extractor() {
let app = nest(
"/foo",
nest(
"/bar",
route("/baz", get(|uri: Uri| async move { uri.to_string() })).route(
"/qux",
get(|req: Request<Body>| async move { req.uri().to_string() }),
),
),
);
let addr = run_in_background(app).await;
let client = reqwest::Client::new();
let res = client
.get(format!("http://{}/foo/bar/baz", addr))
.send()
.await
.unwrap();
Make sure nested services still see full URI (#166)
2021-08-08 17:27:23 +02:00
assert_eq!(res.status(), StatusCode::OK);
Fix Uri extractor not being the full URI if using nest (#156)
2021-08-07 22:07:50 +02:00
assert_eq!(res.text().await.unwrap(), "/foo/bar/baz");
let res = client
.get(format!("http://{}/foo/bar/qux", addr))
.send()
.await
.unwrap();
Make sure nested services still see full URI (#166)
2021-08-08 17:27:23 +02:00
assert_eq!(res.status(), StatusCode::OK);
Fix Uri extractor not being the full URI if using nest (#156)
2021-08-07 22:07:50 +02:00
assert_eq!(res.text().await.unwrap(), "/foo/bar/qux");
}
Add NestedUri (#161)
2021-08-08 14:45:31 +02:00
#[tokio::test]
async fn nested_url_nested_extractor() {
let app = nest(
"/foo",
nest(
"/bar",
route(
"/baz",
get(|uri: extract::NestedUri| async move { uri.0.to_string() }),
),
),
);
let addr = run_in_background(app).await;
let client = reqwest::Client::new();
let res = client
.get(format!("http://{}/foo/bar/baz", addr))
.send()
.await
.unwrap();
Make sure nested services still see full URI (#166)
2021-08-08 17:27:23 +02:00
assert_eq!(res.status(), StatusCode::OK);
Add NestedUri (#161)
2021-08-08 14:45:31 +02:00
assert_eq!(res.text().await.unwrap(), "/baz");
}
Make sure nested services still see full URI (#166)
2021-08-08 17:27:23 +02:00
#[tokio::test]
async fn nested_service_sees_original_uri() {
let app = nest(
"/foo",
nest(
"/bar",
route(
"/baz",
service_fn(|req: Request<Body>| async move {
let body = box_body(Body::from(req.uri().to_string()));
Ok::<_, Infallible>(Response::new(body))
}),
),
),
);
let addr = run_in_background(app).await;
let client = reqwest::Client::new();
let res = client
.get(format!("http://{}/foo/bar/baz", addr))
.send()
.await
.unwrap();
assert_eq!(res.status(), StatusCode::OK);
assert_eq!(res.text().await.unwrap(), "/foo/bar/baz");
}
Reference in New Issue Copy Permalink