Rocket/contrib/lib/tests/static_files.rs

#![feature(proc_macro_hygiene)]

#[cfg(feature = "serve")]
mod static_tests {
    use std::{io::Read, fs::File};
    use std::path::{Path, PathBuf};

    use rocket::{self, Rocket, Route};
    use rocket_contrib::serve::{StaticFiles, Options};
    use rocket::http::Status;
    use rocket::local::Client;

    fn static_root() -> PathBuf {
        Path::new(env!("CARGO_MANIFEST_DIR"))
            .join("tests")
            .join("static")
    }

    fn rocket() -> Rocket {
        let root = static_root();
        rocket::ignite()
            .mount("/default", StaticFiles::from(&root))
            .mount("/no_index", StaticFiles::new(&root, Options::None))
            .mount("/dots", StaticFiles::new(&root, Options::DotFiles))
            .mount("/index", StaticFiles::new(&root, Options::Index))
            .mount("/both", StaticFiles::new(&root, Options::DotFiles | Options::Index))
            .mount("/redir", StaticFiles::new(&root, Options::NormalizeDirs))
            .mount("/redir_index", StaticFiles::new(&root, Options::NormalizeDirs | Options::Index))
    }

    static REGULAR_FILES: &[&str] = &[
        "index.html",
        "inner/goodbye",
        "inner/index.html",
        "other/hello.txt",
    ];

    static HIDDEN_FILES: &[&str] = &[
        ".hidden",
        "inner/.hideme",
    ];

    static INDEXED_DIRECTORIES: &[&str] = &[
        "",
        "inner/",
    ];

    async fn assert_file(client: &Client, prefix: &str, path: &str, exists: bool) {
        let full_path = format!("/{}/{}", prefix, path);
        let mut response = client.get(full_path).dispatch().await;
        if exists {
            assert_eq!(response.status(), Status::Ok);

            let mut path = static_root().join(path);
            if path.is_dir() {
                path = path.join("index.html");
            }

            let mut file = File::open(path).expect("open file");
            let mut expected_contents = String::new();
            file.read_to_string(&mut expected_contents).expect("read file");
            assert_eq!(response.body_string().await, Some(expected_contents));
        } else {
            assert_eq!(response.status(), Status::NotFound);
        }
    }

    async fn assert_all(client: &Client, prefix: &str, paths: &[&str], exist: bool) {
        for path in paths.iter() {
            assert_file(client, prefix, path, exist).await;
        }
    }

    #[rocket::async_test]
    async fn test_static_no_index() {
        let client = Client::new(rocket()).expect("valid rocket");
        assert_all(&client, "no_index", REGULAR_FILES, true).await;
        assert_all(&client, "no_index", HIDDEN_FILES, false).await;
        assert_all(&client, "no_index", INDEXED_DIRECTORIES, false).await;
    }

    #[rocket::async_test]
    async fn test_static_hidden() {
        let client = Client::new(rocket()).expect("valid rocket");
        assert_all(&client, "dots", REGULAR_FILES, true).await;
        assert_all(&client, "dots", HIDDEN_FILES, true).await;
        assert_all(&client, "dots", INDEXED_DIRECTORIES, false).await;
    }

    #[rocket::async_test]
    async fn test_static_index() {
        let client = Client::new(rocket()).expect("valid rocket");
        assert_all(&client, "index", REGULAR_FILES, true).await;
        assert_all(&client, "index", HIDDEN_FILES, false).await;
        assert_all(&client, "index", INDEXED_DIRECTORIES, true).await;

        assert_all(&client, "default", REGULAR_FILES, true).await;
        assert_all(&client, "default", HIDDEN_FILES, false).await;
        assert_all(&client, "default", INDEXED_DIRECTORIES, true).await;
    }

    #[rocket::async_test]
    async fn test_static_all() {
        let client = Client::new(rocket()).expect("valid rocket");
        assert_all(&client, "both", REGULAR_FILES, true).await;
        assert_all(&client, "both", HIDDEN_FILES, true).await;
        assert_all(&client, "both", INDEXED_DIRECTORIES, true).await;
    }

    #[test]
    fn test_ranking() {
        let root = static_root();
        for rank in -128..128 {
            let a = StaticFiles::new(&root, Options::None).rank(rank);
            let b = StaticFiles::from(&root).rank(rank);

            for handler in vec![a, b] {
                let routes: Vec<Route> = handler.into();
                assert!(routes.iter().all(|route| route.rank == rank), "{}", rank);
            }
        }
    }

    #[rocket::async_test]
    async fn test_forwarding() {
        use rocket::http::RawStr;
        use rocket::{get, routes};

        #[get("/<value>", rank = 20)]
        fn catch_one(value: String) -> String { value }

        #[get("/<a>/<b>", rank = 20)]
        fn catch_two(a: &RawStr, b: &RawStr) -> String { format!("{}/{}", a, b) }

        let rocket = rocket().mount("/default", routes![catch_one, catch_two]);
        let client = Client::new(rocket).expect("valid rocket");

        let mut response = client.get("/default/ireallydontexist").dispatch().await;
        assert_eq!(response.status(), Status::Ok);
        assert_eq!(response.body_string().await.unwrap(), "ireallydontexist");

        let mut response = client.get("/default/idont/exist").dispatch().await;
        assert_eq!(response.status(), Status::Ok);
        assert_eq!(response.body_string().await.unwrap(), "idont/exist");

        assert_all(&client, "both", REGULAR_FILES, true).await;
        assert_all(&client, "both", HIDDEN_FILES, true).await;
        assert_all(&client, "both", INDEXED_DIRECTORIES, true).await;
    }

    #[test]
    fn test_redirection() {
        let client = Client::new(rocket()).expect("valid rocket");

        // Redirection only happens if enabled, and doesn't affect index behaviour.
        let response = client.get("/no_index/inner").dispatch();
        assert_eq!(response.status(), Status::NotFound);

        let response = client.get("/index/inner").dispatch();
        assert_eq!(response.status(), Status::Ok);

        let response = client.get("/redir/inner").dispatch();
        assert_eq!(response.status(), Status::PermanentRedirect);
        assert_eq!(response.headers().get("Location").next(), Some("/redir/inner/"));

        let response = client.get("/redir/inner?foo=bar").dispatch();
        assert_eq!(response.status(), Status::PermanentRedirect);
        assert_eq!(response.headers().get("Location").next(), Some("/redir/inner/?foo=bar"));

        let response = client.get("/redir_index/inner").dispatch();
        assert_eq!(response.status(), Status::PermanentRedirect);
        assert_eq!(response.headers().get("Location").next(), Some("/redir_index/inner/"));

        // Paths with trailing slash are unaffected.
        let response = client.get("/redir/inner/").dispatch();
        assert_eq!(response.status(), Status::NotFound);

        let response = client.get("/redir_index/inner/").dispatch();
        assert_eq!(response.status(), Status::Ok);

        // Root of route is also redirected.
        let response = client.get("/no_index").dispatch();
        assert_eq!(response.status(), Status::NotFound);

        let response = client.get("/index").dispatch();
        assert_eq!(response.status(), Status::Ok);

        let response = client.get("/redir").dispatch();
        assert_eq!(response.status(), Status::PermanentRedirect);
        assert_eq!(response.headers().get("Location").next(), Some("/redir/"));

        let response = client.get("/redir_index").dispatch();
        assert_eq!(response.status(), Status::PermanentRedirect);
        assert_eq!(response.headers().get("Location").next(), Some("/redir_index/"));
    }
}