Fix stripping prefix when nesting at / (#91)

* Fix stripping prefix when nesting at `/` Fixes https://github.com/tokio-rs/axum/issues/88 * changelog
2026-08-26 00:00:23 +02:00 · 2021-08-02 22:40:33 +02:00
parent c6b7ad0f33
commit 6a078ddb71
4 changed files with 135 additions and 101 deletions
@@ -0,0 +1,123 @@
+use super::*;
+
+#[tokio::test]
+async fn nesting_apps() {
+    let api_routes = route(
+        "/users",
+        get(|| async { "users#index" }).post(|| async { "users#create" }),
+    )
+    .route(
+        "/users/:id",
+        get(|params: extract::UrlParamsMap| async move {
+            format!(
+                "{}: users#show ({})",
+                params.get("version").unwrap(),
+                params.get("id").unwrap()
+            )
+        }),
+    )
+    .route(
+        "/games/:id",
+        get(|params: extract::UrlParamsMap| async move {
+            format!(
+                "{}: games#show ({})",
+                params.get("version").unwrap(),
+                params.get("id").unwrap()
+            )
+        }),
+    );
+
+    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");
+}