firefox/third_party/rust/warp/tests/fs.rs

#![deny(warnings)]
use std::fs;

#[tokio::test]
async fn file() {
    let _ = pretty_env_logger::try_init();

    let file = warp::fs::file("README.md");

    let req = warp::test::request();
    let res = req.reply(&file).await;

    assert_eq!(res.status(), 200);

    let contents = fs::read("README.md").expect("fs::read README.md");
    assert_eq!(res.headers()["content-length"], contents.len().to_string());
    assert_eq!(res.headers()["accept-ranges"], "bytes");

    let ct = &res.headers()["content-type"];
    assert!(
        ct == "text/x-markdown" || ct == "text/markdown",
        "content-type is not markdown: {:?}",
        ct,
    );

    assert_eq!(res.body(), &*contents);
}

#[tokio::test]
async fn dir() {
    let _ = pretty_env_logger::try_init();

    let file = warp::fs::dir("examples");

    let req = warp::test::request().path("/todos.rs");
    let res = req.reply(&file).await;

    assert_eq!(res.status(), 200);

    let contents = fs::read("examples/todos.rs").expect("fs::read");
    assert_eq!(res.headers()["content-length"], contents.len().to_string());
    assert_eq!(res.headers()["content-type"], "text/x-rust");
    assert_eq!(res.headers()["accept-ranges"], "bytes");

    assert_eq!(res.body(), &*contents);

    let malformed_req = warp::test::request().path("todos.rs");
    assert_eq!(malformed_req.reply(&file).await.status(), 404);
}

#[tokio::test]
async fn dir_encoded() {
    let _ = pretty_env_logger::try_init();

    let file = warp::fs::dir("examples");

    let req = warp::test::request().path("/todos%2ers");
    let res = req.reply(&file).await;

    assert_eq!(res.status(), 200);

    let contents = fs::read("examples/todos.rs").expect("fs::read");
    assert_eq!(res.headers()["content-length"], contents.len().to_string());

    assert_eq!(res.body(), &*contents);
}

#[tokio::test]
async fn dir_not_found() {
    let _ = pretty_env_logger::try_init();

    let file = warp::fs::dir("examples");

    let req = warp::test::request().path("/definitely-not-found");
    let res = req.reply(&file).await;

    assert_eq!(res.status(), 404);
}

#[tokio::test]
async fn dir_bad_path() {
    let _ = pretty_env_logger::try_init();

    let file = warp::fs::dir("examples");

    let req = warp::test::request().path("/../README.md");
    let res = req.reply(&file).await;

    assert_eq!(res.status(), 404);
}

#[tokio::test]
async fn dir_bad_encoded_path() {
    let _ = pretty_env_logger::try_init();

    let file = warp::fs::dir("examples");

    let req = warp::test::request().path("/%2E%2e/README.md");
    let res = req.reply(&file).await;

    assert_eq!(res.status(), 404);
}

#[tokio::test]
async fn dir_fallback_index_on_dir() {
    let _ = pretty_env_logger::try_init();

    let file = warp::fs::dir("examples");
    let req = warp::test::request().path("/dir");
    let res = req.reply(&file).await;
    let contents = fs::read("examples/dir/index.html").expect("fs::read");
    assert_eq!(res.headers()["content-length"], contents.len().to_string());
    assert_eq!(res.status(), 200);
    let req = warp::test::request().path("/dir/");
    let res = req.reply(&file).await;
    assert_eq!(res.headers()["content-length"], contents.len().to_string());
    assert_eq!(res.status(), 200);
}

#[tokio::test]
async fn not_modified() {
    let _ = pretty_env_logger::try_init();

    let file = warp::fs::file("README.md");

    let req = warp::test::request();
    let body = fs::read("README.md").unwrap();
    let res1 = req.reply(&file).await;
    assert_eq!(res1.status(), 200);
    assert_eq!(res1.headers()["content-length"], body.len().to_string());

    // if-modified-since
    let res = warp::test::request()
        .header("if-modified-since", &res1.headers()["last-modified"])
        .reply(&file)
        .await;
    assert_eq!(res.headers().get("content-length"), None);
    assert_eq!(res.status(), 304);
    assert_eq!(res.body(), "");

    // clearly too old
    let res = warp::test::request()
        .header("if-modified-since", "Mon, 07 Nov 1994 01:00:00 GMT")
        .reply(&file)
        .await;
    assert_eq!(res.status(), 200);
    assert_eq!(res.body(), &body);
    assert_eq!(res1.headers()["content-length"], body.len().to_string());
}

#[tokio::test]
async fn precondition() {
    let _ = pretty_env_logger::try_init();

    let file = warp::fs::file("README.md");

    let req = warp::test::request();
    let res1 = req.reply(&file).await;
    assert_eq!(res1.status(), 200);

    // if-unmodified-since
    let res = warp::test::request()
        .header("if-unmodified-since", &res1.headers()["last-modified"])
        .reply(&file)
        .await;
    assert_eq!(res.status(), 200);

    // clearly too old
    let res = warp::test::request()
        .header("if-unmodified-since", "Mon, 07 Nov 1994 01:00:00 GMT")
        .reply(&file)
        .await;
    assert_eq!(res.status(), 412);
    assert_eq!(res.body(), "");
}

#[tokio::test]
async fn byte_ranges() {
    let _ = pretty_env_logger::try_init();

    let contents = fs::read("README.md").expect("fs::read README.md");
    let file = warp::fs::file("README.md");

    let res = warp::test::request()
        .header("range", "bytes=100-200")
        .reply(&file)
        .await;
    assert_eq!(res.status(), 206);
    assert_eq!(
        res.headers()["content-range"],
        format!("bytes 100-200/{}", contents.len())
    );
    assert_eq!(res.headers()["content-length"], "101");
    assert_eq!(res.body(), &contents[100..=200]);

    // bad range
    let res = warp::test::request()
        .header("range", "bytes=100-10")
        .reply(&file)
        .await;
    assert_eq!(res.status(), 416);
    assert_eq!(
        res.headers()["content-range"],
        format!("bytes */{}", contents.len())
    );
    assert_eq!(res.headers().get("content-length"), None);
    assert_eq!(res.body(), "");

    // out of range
    let res = warp::test::request()
        .header("range", "bytes=100-100000")
        .reply(&file)
        .await;
    assert_eq!(res.status(), 416);
    assert_eq!(
        res.headers()["content-range"],
        format!("bytes */{}", contents.len())
    );
    assert_eq!(res.headers().get("content-length"), None);
    assert_eq!(res.body(), "");

    // if-range too old
    let res = warp::test::request()
        .header("range", "bytes=100-200")
        .header("if-range", "Mon, 07 Nov 1994 01:00:00 GMT")
        .reply(&file)
        .await;
    assert_eq!(res.status(), 200);
    assert_eq!(res.headers()["content-length"], contents.len().to_string());
    assert_eq!(res.headers().get("content-range"), None);
}

#[tokio::test]
async fn byte_ranges_with_excluded_file_size() {
    let _ = pretty_env_logger::try_init();

    let contents = fs::read("README.md").expect("fs::read README.md");
    let file = warp::fs::file("README.md");

    // range including end of file (non-inclusive result)
    let res = warp::test::request()
        .header("range", format!("bytes=100-{}", contents.len()))
        .reply(&file)
        .await;
    assert_eq!(res.status(), 206);
    assert_eq!(
        res.headers()["content-range"],
        format!("bytes 100-{}/{}", contents.len() - 1, contents.len())
    );
    assert_eq!(
        res.headers()["content-length"],
        format!("{}", contents.len() - 100)
    );
    assert_eq!(res.body(), &contents[100..=contents.len() - 1]);

    // range with 1 byte to end yields same result as above. (inclusive result)
    let res = warp::test::request()
        .header("range", format!("bytes=100-{}", contents.len() - 1))
        .reply(&file)
        .await;
    assert_eq!(res.status(), 206);
    assert_eq!(
        res.headers()["content-range"],
        format!("bytes 100-{}/{}", contents.len() - 1, contents.len())
    );
    assert_eq!(
        res.headers()["content-length"],
        format!("{}", contents.len() - 100)
    );
    assert_eq!(res.body(), &contents[100..=contents.len() - 1]);
}