tide/tests/wildcard.rs

111 lines
3.3 KiB
Rust

mod test_utils;
use test_utils::ServerTestingExt;
use tide::{Error, Request, StatusCode};
async fn add_one(req: Request<()>) -> Result<String, tide::Error> {
let num: i64 = req
.param("num")?
.parse()
.map_err(|err| Error::new(StatusCode::BadRequest, err))?;
Ok((num + 1).to_string())
}
async fn add_two(req: Request<()>) -> Result<String, tide::Error> {
let one: i64 = req
.param("one")?
.parse()
.map_err(|err| Error::new(StatusCode::BadRequest, err))?;
let two: i64 = req
.param("two")?
.parse()
.map_err(|err| Error::new(StatusCode::BadRequest, err))?;
Ok((one + two).to_string())
}
async fn echo_param(req: Request<()>) -> tide::Result<tide::Response> {
match req.param("param") {
Ok(path) => Ok(path.into()),
Err(_) => Ok(StatusCode::NotFound.into()),
}
}
async fn echo_wildcard(req: Request<()>) -> tide::Result<tide::Response> {
match req.wildcard() {
Some(path) => Ok(path.into()),
None => Ok(StatusCode::NotFound.into()),
}
}
#[async_std::test]
async fn param() -> tide::Result<()> {
let mut app = tide::Server::new();
app.at("/add_one/:num").get(add_one);
assert_eq!(app.get("/add_one/3").recv_string().await?, "4");
assert_eq!(app.get("/add_one/-7").recv_string().await?, "-6");
Ok(())
}
#[async_std::test]
async fn invalid_segment_error() -> tide::Result<()> {
let mut app = tide::new();
app.at("/add_one/:num").get(add_one);
assert_eq!(
app.get("/add_one/a").await?.status(),
StatusCode::BadRequest
);
Ok(())
}
#[async_std::test]
async fn not_found_error() -> tide::Result<()> {
let mut app = tide::new();
app.at("/add_one/:num").get(add_one);
assert_eq!(app.get("/add_one/").await?.status(), StatusCode::NotFound);
Ok(())
}
#[async_std::test]
async fn wildcard() -> tide::Result<()> {
let mut app = tide::new();
app.at("/echo/*").get(echo_wildcard);
assert_eq!(app.get("/echo/some_path").recv_string().await?, "some_path");
assert_eq!(
app.get("/echo/multi/segment/path").recv_string().await?,
"multi/segment/path"
);
assert_eq!(app.get("/echo/").await?.status(), StatusCode::Ok);
assert_eq!(app.get("/echo").await?.status(), StatusCode::Ok);
Ok(())
}
#[async_std::test]
async fn multi_param() -> tide::Result<()> {
let mut app = tide::new();
app.at("/add_two/:one/:two/").get(add_two);
assert_eq!(app.get("/add_two/1/2/").recv_string().await?, "3");
assert_eq!(app.get("/add_two/-1/2/").recv_string().await?, "1");
assert_eq!(app.get("/add_two/1").await?.status(), StatusCode::NotFound);
Ok(())
}
#[async_std::test]
async fn wildcard_last_segment() -> tide::Result<()> {
let mut app = tide::new();
app.at("/echo/:param/*").get(echo_param);
assert_eq!(app.get("/echo/one/two").recv_string().await?, "one");
assert_eq!(
app.get("/echo/one/two/three/four").recv_string().await?,
"one"
);
Ok(())
}
#[async_std::test]
async fn ambiguous_router_wildcard_vs_star() -> tide::Result<()> {
let mut app = tide::new();
app.at("/:one/:two").get(|_| async { Ok("one/two") });
app.at("/posts/*").get(|_| async { Ok("posts/*") });
assert_eq!(app.get("/posts/10").recv_string().await?, "posts/*");
Ok(())
}