src/tests/nest.rs

use super::*;
use crate::body::box_body;
use std::collections::HashMap;

#[tokio::test]
async fn nesting_apps() {
    let api_routes = Router::new()
        .route(
            "/users",
            get(|| async { "users#index" }).post(|| async { "users#create" }),
        )
        .route(
            "/users/:id",
            get(
                |params: extract::Path<HashMap<String, String>>| async move {
                    format!(
                        "{}: users#show ({})",
                        params.get("version").unwrap(),
                        params.get("id").unwrap()
                    )
                },
            ),
        )
        .route(
            "/games/:id",
            get(
                |params: extract::Path<HashMap<String, String>>| async move {
                    format!(
                        "{}: games#show ({})",
                        params.get("version").unwrap(),
                        params.get("id").unwrap()
                    )
                },
            ),
        );

    let app = Router::new()
        .route("/", get(|| async { "hi" }))
        .nest("/:version/api", api_routes);

    let client = TestClient::new(app);

    let res = client.get("/").send().await;
    assert_eq!(res.status(), StatusCode::OK);
    assert_eq!(res.text().await, "hi");

    let res = client.get("/v0/api/users").send().await;
    assert_eq!(res.status(), StatusCode::OK);
    assert_eq!(res.text().await, "users#index");

    let res = client.get("/v0/api/users/123").send().await;
    assert_eq!(res.status(), StatusCode::OK);
    assert_eq!(res.text().await, "v0: users#show (123)");

    let res = client.get("/v0/api/games/123").send().await;
    assert_eq!(res.status(), StatusCode::OK);
    assert_eq!(res.text().await, "v0: games#show (123)");
}

#[tokio::test]
async fn wrong_method_nest() {
    let nested_app = Router::new().route("/", get(|| async {}));
    let app = Router::new().nest("/", nested_app);

    let client = TestClient::new(app);

    let res = client.get("/").send().await;
    assert_eq!(res.status(), StatusCode::OK);

    let res = client.post("/").send().await;
    assert_eq!(res.status(), StatusCode::METHOD_NOT_ALLOWED);

    let res = client.patch("/foo").send().await;
    assert_eq!(res.status(), StatusCode::NOT_FOUND);
}

#[tokio::test]
async fn nesting_at_root() {
    let app = Router::new().nest("/", get(|uri: Uri| async move { uri.to_string() }));

    let client = TestClient::new(app);

    let res = client.get("/").send().await;
    assert_eq!(res.status(), StatusCode::OK);
    assert_eq!(res.text().await, "/");

    let res = client.get("/foo").send().await;
    assert_eq!(res.status(), StatusCode::OK);
    assert_eq!(res.text().await, "/foo");

    let res = client.get("/foo/bar").send().await;
    assert_eq!(res.status(), StatusCode::OK);
    assert_eq!(res.text().await, "/foo/bar");
}

#[tokio::test]
async fn nested_url_extractor() {
    let app = Router::new().nest(
        "/foo",
        Router::new().nest(
            "/bar",
            Router::new()
                .route("/baz", get(|uri: Uri| async move { uri.to_string() }))
                .route(
                    "/qux",
                    get(|req: Request<Body>| async move { req.uri().to_string() }),
                ),
        ),
    );

    let client = TestClient::new(app);

    let res = client.get("/foo/bar/baz").send().await;
    assert_eq!(res.status(), StatusCode::OK);
    assert_eq!(res.text().await, "/baz");

    let res = client.get("/foo/bar/qux").send().await;
    assert_eq!(res.status(), StatusCode::OK);
    assert_eq!(res.text().await, "/qux");
}

#[tokio::test]
async fn nested_url_original_extractor() {
    let app = Router::new().nest(
        "/foo",
        Router::new().nest(
            "/bar",
            Router::new().route(
                "/baz",
                get(|uri: extract::OriginalUri| async move { uri.0.to_string() }),
            ),
        ),
    );

    let client = TestClient::new(app);

    let res = client.get("/foo/bar/baz").send().await;
    assert_eq!(res.status(), StatusCode::OK);
    assert_eq!(res.text().await, "/foo/bar/baz");
}

#[tokio::test]
async fn nested_service_sees_stripped_uri() {
    let app = Router::new().nest(
        "/foo",
        Router::new().nest(
            "/bar",
            Router::new().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 client = TestClient::new(app);

    let res = client.get("/foo/bar/baz").send().await;
    assert_eq!(res.status(), StatusCode::OK);
    assert_eq!(res.text().await, "/baz");
}

#[tokio::test]
async fn nest_static_file_server() {
    let app = Router::new().nest(
        "/static",
        service::get(tower_http::services::ServeDir::new(".")).handle_error(|error| {
            Ok::<_, Infallible>((
                StatusCode::INTERNAL_SERVER_ERROR,
                format!("Unhandled internal error: {}", error),
            ))
        }),
    );

    let client = TestClient::new(app);

    let res = client.get("/static/README.md").send().await;
    assert_eq!(res.status(), StatusCode::OK);
}
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() {`
Replace `route` with `Router::new().route()` (#215 ) 2021-08-19 22:37:48 +02:00			`let api_routes = Router::new()`
			`.route(`
			`"/users",`
			`get(\|\| async { "users#index" }).post(\|\| async { "users#create" }),`
			`)`
			`.route(`
			`"/users/:id",`
			`get(`
			`\|params: extract::Path<HashMap<String, String>>\| async move {`
			`format!(`
			`"{}: users#show ({})",`
			`params.get("version").unwrap(),`
			`params.get("id").unwrap()`
			`)`
			`},`
			`),`
			`)`
			`.route(`
			`"/games/:id",`
			`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
Replace `route` with `Router::new().route()` (#215 ) 2021-08-19 22:37:48 +02:00			`let app = Router::new()`
			`.route("/", get(\|\| async { "hi" }))`
			`.nest("/:version/api", api_routes);`
Fix stripping prefix when nesting at `/` (#91 ) 2021-08-02 22:40:33 +02:00
Refactor internal testing setup (#338 ) 2021-09-19 11:38:34 +02:00			`let client = TestClient::new(app);`
Fix stripping prefix when nesting at `/` (#91 ) 2021-08-02 22:40:33 +02:00
Refactor internal testing setup (#338 ) 2021-09-19 11:38:34 +02:00			`let res = client.get("/").send().await;`
Fix stripping prefix when nesting at `/` (#91 ) 2021-08-02 22:40:33 +02:00			`assert_eq!(res.status(), StatusCode::OK);`
Refactor internal testing setup (#338 ) 2021-09-19 11:38:34 +02:00			`assert_eq!(res.text().await, "hi");`
Fix stripping prefix when nesting at `/` (#91 ) 2021-08-02 22:40:33 +02:00
Refactor internal testing setup (#338 ) 2021-09-19 11:38:34 +02:00			`let res = client.get("/v0/api/users").send().await;`
Fix stripping prefix when nesting at `/` (#91 ) 2021-08-02 22:40:33 +02:00			`assert_eq!(res.status(), StatusCode::OK);`
Refactor internal testing setup (#338 ) 2021-09-19 11:38:34 +02:00			`assert_eq!(res.text().await, "users#index");`
Fix stripping prefix when nesting at `/` (#91 ) 2021-08-02 22:40:33 +02:00
Refactor internal testing setup (#338 ) 2021-09-19 11:38:34 +02:00			`let res = client.get("/v0/api/users/123").send().await;`
Fix stripping prefix when nesting at `/` (#91 ) 2021-08-02 22:40:33 +02:00			`assert_eq!(res.status(), StatusCode::OK);`
Refactor internal testing setup (#338 ) 2021-09-19 11:38:34 +02:00			`assert_eq!(res.text().await, "v0: users#show (123)");`
Fix stripping prefix when nesting at `/` (#91 ) 2021-08-02 22:40:33 +02:00
Refactor internal testing setup (#338 ) 2021-09-19 11:38:34 +02:00			`let res = client.get("/v0/api/games/123").send().await;`
Fix stripping prefix when nesting at `/` (#91 ) 2021-08-02 22:40:33 +02:00			`assert_eq!(res.status(), StatusCode::OK);`
Refactor internal testing setup (#338 ) 2021-09-19 11:38:34 +02:00			`assert_eq!(res.text().await, "v0: games#show (123)");`
Fix stripping prefix when nesting at `/` (#91 ) 2021-08-02 22:40:33 +02:00			`}`

			`#[tokio::test]`
			`async fn wrong_method_nest() {`
Replace `route` with `Router::new().route()` (#215 ) 2021-08-19 22:37:48 +02:00			`let nested_app = Router::new().route("/", get(\|\| async {}));`
			`let app = Router::new().nest("/", nested_app);`
Fix stripping prefix when nesting at `/` (#91 ) 2021-08-02 22:40:33 +02:00
Refactor internal testing setup (#338 ) 2021-09-19 11:38:34 +02:00			`let client = TestClient::new(app);`
Fix stripping prefix when nesting at `/` (#91 ) 2021-08-02 22:40:33 +02:00
Refactor internal testing setup (#338 ) 2021-09-19 11:38:34 +02:00			`let res = client.get("/").send().await;`
Fix stripping prefix when nesting at `/` (#91 ) 2021-08-02 22:40:33 +02:00			`assert_eq!(res.status(), StatusCode::OK);`

Refactor internal testing setup (#338 ) 2021-09-19 11:38:34 +02:00			`let res = client.post("/").send().await;`
Fix stripping prefix when nesting at `/` (#91 ) 2021-08-02 22:40:33 +02:00			`assert_eq!(res.status(), StatusCode::METHOD_NOT_ALLOWED);`

Refactor internal testing setup (#338 ) 2021-09-19 11:38:34 +02:00			`let res = client.patch("/foo").send().await;`
Fix stripping prefix when nesting at `/` (#91 ) 2021-08-02 22:40:33 +02:00			`assert_eq!(res.status(), StatusCode::NOT_FOUND);`
			`}`

			`#[tokio::test]`
			`async fn nesting_at_root() {`
Replace `route` with `Router::new().route()` (#215 ) 2021-08-19 22:37:48 +02:00			`let app = Router::new().nest("/", get(\|uri: Uri\| async move { uri.to_string() }));`
Fix stripping prefix when nesting at `/` (#91 ) 2021-08-02 22:40:33 +02:00
Refactor internal testing setup (#338 ) 2021-09-19 11:38:34 +02:00			`let client = TestClient::new(app);`
Fix stripping prefix when nesting at `/` (#91 ) 2021-08-02 22:40:33 +02:00
Refactor internal testing setup (#338 ) 2021-09-19 11:38:34 +02:00			`let res = client.get("/").send().await;`
Fix stripping prefix when nesting at `/` (#91 ) 2021-08-02 22:40:33 +02:00			`assert_eq!(res.status(), StatusCode::OK);`
Refactor internal testing setup (#338 ) 2021-09-19 11:38:34 +02:00			`assert_eq!(res.text().await, "/");`
Fix stripping prefix when nesting at `/` (#91 ) 2021-08-02 22:40:33 +02:00
Refactor internal testing setup (#338 ) 2021-09-19 11:38:34 +02:00			`let res = client.get("/foo").send().await;`
Fix stripping prefix when nesting at `/` (#91 ) 2021-08-02 22:40:33 +02:00			`assert_eq!(res.status(), StatusCode::OK);`
Refactor internal testing setup (#338 ) 2021-09-19 11:38:34 +02:00			`assert_eq!(res.text().await, "/foo");`
Fix stripping prefix when nesting at `/` (#91 ) 2021-08-02 22:40:33 +02:00
Refactor internal testing setup (#338 ) 2021-09-19 11:38:34 +02:00			`let res = client.get("/foo/bar").send().await;`
Fix stripping prefix when nesting at `/` (#91 ) 2021-08-02 22:40:33 +02:00			`assert_eq!(res.status(), StatusCode::OK);`
Refactor internal testing setup (#338 ) 2021-09-19 11:38:34 +02:00			`assert_eq!(res.text().await, "/foo/bar");`
Fix stripping prefix when nesting at `/` (#91 ) 2021-08-02 22:40:33 +02:00			`}`
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() {`
Replace `route` with `Router::new().route()` (#215 ) 2021-08-19 22:37:48 +02:00			`let app = Router::new().nest(`
Fix `Uri` extractor not being the full URI if using `nest` (#156 ) 2021-08-07 22:07:50 +02:00			`"/foo",`
Replace `route` with `Router::new().route()` (#215 ) 2021-08-19 22:37:48 +02:00			`Router::new().nest(`
Fix `Uri` extractor not being the full URI if using `nest` (#156 ) 2021-08-07 22:07:50 +02:00			`"/bar",`
Replace `route` with `Router::new().route()` (#215 ) 2021-08-19 22:37:48 +02:00			`Router::new()`
			`.route("/baz", get(\|uri: Uri\| async move { uri.to_string() }))`
			`.route(`
			`"/qux",`
			`get(\|req: Request<Body>\| async move { req.uri().to_string() }),`
			`),`
Fix `Uri` extractor not being the full URI if using `nest` (#156 ) 2021-08-07 22:07:50 +02:00			`),`
			`);`

Refactor internal testing setup (#338 ) 2021-09-19 11:38:34 +02:00			`let client = TestClient::new(app);`
Fix `Uri` extractor not being the full URI if using `nest` (#156 ) 2021-08-07 22:07:50 +02:00
Refactor internal testing setup (#338 ) 2021-09-19 11:38:34 +02:00			`let res = client.get("/foo/bar/baz").send().await;`
Make sure nested services still see full URI (#166 ) 2021-08-08 17:27:23 +02:00			`assert_eq!(res.status(), StatusCode::OK);`
Refactor internal testing setup (#338 ) 2021-09-19 11:38:34 +02:00			`assert_eq!(res.text().await, "/baz");`
Fix `Uri` extractor not being the full URI if using `nest` (#156 ) 2021-08-07 22:07:50 +02:00
Refactor internal testing setup (#338 ) 2021-09-19 11:38:34 +02:00			`let res = client.get("/foo/bar/qux").send().await;`
Make sure nested services still see full URI (#166 ) 2021-08-08 17:27:23 +02:00			`assert_eq!(res.status(), StatusCode::OK);`
Refactor internal testing setup (#338 ) 2021-09-19 11:38:34 +02:00			`assert_eq!(res.text().await, "/qux");`
Fix `Uri` extractor not being the full URI if using `nest` (#156 ) 2021-08-07 22:07:50 +02:00			`}`
Add `NestedUri` (#161 ) 2021-08-08 14:45:31 +02:00
			`#[tokio::test]`
Change nested routes to see the URI with prefix stripped (#197 ) 2021-08-18 09:48:36 +02:00			`async fn nested_url_original_extractor() {`
Replace `route` with `Router::new().route()` (#215 ) 2021-08-19 22:37:48 +02:00			`let app = Router::new().nest(`
Add `NestedUri` (#161 ) 2021-08-08 14:45:31 +02:00			`"/foo",`
Replace `route` with `Router::new().route()` (#215 ) 2021-08-19 22:37:48 +02:00			`Router::new().nest(`
Add `NestedUri` (#161 ) 2021-08-08 14:45:31 +02:00			`"/bar",`
Replace `route` with `Router::new().route()` (#215 ) 2021-08-19 22:37:48 +02:00			`Router::new().route(`
Add `NestedUri` (#161 ) 2021-08-08 14:45:31 +02:00			`"/baz",`
Change nested routes to see the URI with prefix stripped (#197 ) 2021-08-18 09:48:36 +02:00			`get(\|uri: extract::OriginalUri\| async move { uri.0.to_string() }),`
Add `NestedUri` (#161 ) 2021-08-08 14:45:31 +02:00			`),`
			`),`
			`);`

Refactor internal testing setup (#338 ) 2021-09-19 11:38:34 +02:00			`let client = TestClient::new(app);`
Add `NestedUri` (#161 ) 2021-08-08 14:45:31 +02:00
Refactor internal testing setup (#338 ) 2021-09-19 11:38:34 +02:00			`let res = client.get("/foo/bar/baz").send().await;`
Make sure nested services still see full URI (#166 ) 2021-08-08 17:27:23 +02:00			`assert_eq!(res.status(), StatusCode::OK);`
Refactor internal testing setup (#338 ) 2021-09-19 11:38:34 +02:00			`assert_eq!(res.text().await, "/foo/bar/baz");`
Add `NestedUri` (#161 ) 2021-08-08 14:45:31 +02:00			`}`
Make sure nested services still see full URI (#166 ) 2021-08-08 17:27:23 +02:00
			`#[tokio::test]`
Change nested routes to see the URI with prefix stripped (#197 ) 2021-08-18 09:48:36 +02:00			`async fn nested_service_sees_stripped_uri() {`
Replace `route` with `Router::new().route()` (#215 ) 2021-08-19 22:37:48 +02:00			`let app = Router::new().nest(`
Make sure nested services still see full URI (#166 ) 2021-08-08 17:27:23 +02:00			`"/foo",`
Replace `route` with `Router::new().route()` (#215 ) 2021-08-19 22:37:48 +02:00			`Router::new().nest(`
Make sure nested services still see full URI (#166 ) 2021-08-08 17:27:23 +02:00			`"/bar",`
Replace `route` with `Router::new().route()` (#215 ) 2021-08-19 22:37:48 +02:00			`Router::new().route(`
Make sure nested services still see full URI (#166 ) 2021-08-08 17:27:23 +02:00			`"/baz",`
			`service_fn(\|req: Request<Body>\| async move {`
			`let body = box_body(Body::from(req.uri().to_string()));`
			`Ok::<_, Infallible>(Response::new(body))`
			`}),`
			`),`
			`),`
			`);`

Refactor internal testing setup (#338 ) 2021-09-19 11:38:34 +02:00			`let client = TestClient::new(app);`
Make sure nested services still see full URI (#166 ) 2021-08-08 17:27:23 +02:00
Refactor internal testing setup (#338 ) 2021-09-19 11:38:34 +02:00			`let res = client.get("/foo/bar/baz").send().await;`
Make sure nested services still see full URI (#166 ) 2021-08-08 17:27:23 +02:00			`assert_eq!(res.status(), StatusCode::OK);`
Refactor internal testing setup (#338 ) 2021-09-19 11:38:34 +02:00			`assert_eq!(res.text().await, "/baz");`
Change nested routes to see the URI with prefix stripped (#197 ) 2021-08-18 09:48:36 +02:00			`}`

			`#[tokio::test]`
			`async fn nest_static_file_server() {`
Replace `route` with `Router::new().route()` (#215 ) 2021-08-19 22:37:48 +02:00			`let app = Router::new().nest(`
Change nested routes to see the URI with prefix stripped (#197 ) 2021-08-18 09:48:36 +02:00			`"/static",`
			`service::get(tower_http::services::ServeDir::new(".")).handle_error(\|error\| {`
			`Ok::<_, Infallible>((`
			`StatusCode::INTERNAL_SERVER_ERROR,`
			`format!("Unhandled internal error: {}", error),`
			`))`
			`}),`
			`);`

Refactor internal testing setup (#338 ) 2021-09-19 11:38:34 +02:00			`let client = TestClient::new(app);`
Change nested routes to see the URI with prefix stripped (#197 ) 2021-08-18 09:48:36 +02:00
Refactor internal testing setup (#338 ) 2021-09-19 11:38:34 +02:00			`let res = client.get("/static/README.md").send().await;`
Change nested routes to see the URI with prefix stripped (#197 ) 2021-08-18 09:48:36 +02:00			`assert_eq!(res.status(), StatusCode::OK);`
Make sure nested services still see full URI (#166 ) 2021-08-08 17:27:23 +02:00			`}`