Add tests for function middleware

tests/function_middleware.rs

@@ -0,0 +1,84 @@
+use test_utils::BoxFuture;
+use tide::http::{self, url::Url, Method};
+
+mod test_utils;
+
+fn auth_middleware<'a>(
+    request: tide::Request<()>,
+    next: tide::Next<'a, ()>,
+) -> BoxFuture<'a, tide::Result> {
+    Box::pin(async {
+        let authenticated = match request.header("X-Auth") {
+            Some(header) => header == "secret_key",
+            None => false,
+        };
+
+        if authenticated {
+            next.run(request).await
+        } else {
+            Ok(tide::Response::new(tide::StatusCode::Unauthorized))
+        }
+    })
+}
+
+async fn echo_path<State>(req: tide::Request<State>) -> tide::Result<String> {
+    Ok(req.url().path().to_string())
+}
+
+#[async_std::test]
+async fn route_middleware() {
+    let mut app = tide::new();
+    app.at("/protected")
+        .middleware(auth_middleware)
+        .get(echo_path);
+    app.at("/unprotected").get(echo_path);
+
+    // Protected
+    let req = http::Request::new(
+        Method::Get,
+        Url::parse("http://localhost/protected").unwrap(),
+    );
+    let res: http::Response = app.respond(req).await.unwrap();
+    assert_eq!(res.status(), tide::StatusCode::Unauthorized);
+
+    let mut req = http::Request::new(
+        Method::Get,
+        Url::parse("http://localhost/protected").unwrap(),
+    );
+    req.insert_header("X-Auth", "secret_key");
+    let res: http::Response = app.respond(req).await.unwrap();
+    assert_eq!(res.status(), tide::StatusCode::Ok);
+
+    // Unprotected
+    let req = http::Request::new(
+        Method::Get,
+        Url::parse("http://localhost/unprotected").unwrap(),
+    );
+    let res: http::Response = app.respond(req).await.unwrap();
+    assert_eq!(res.status(), tide::StatusCode::Ok);
+
+    let mut req = http::Request::new(
+        Method::Get,
+        Url::parse("http://localhost/unprotected").unwrap(),
+    );
+    req.insert_header("X-Auth", "secret_key");
+    let res: http::Response = app.respond(req).await.unwrap();
+    assert_eq!(res.status(), tide::StatusCode::Ok);
+}
+
+#[async_std::test]
+async fn app_middleware() {
+    let mut app = tide::new();
+    app.middleware(auth_middleware);
+    app.at("/foo").get(echo_path);
+
+    // Foo
+    let req = http::Request::new(Method::Get, Url::parse("http://localhost/foo").unwrap());
+    let res: http::Response = app.respond(req).await.unwrap();
+    assert_eq!(res.status(), tide::StatusCode::Unauthorized);
+
+    let mut req = http::Request::new(Method::Get, Url::parse("http://localhost/foo").unwrap());
+    req.insert_header("X-Auth", "secret_key");
+    let res: http::Response = app.respond(req).await.unwrap();
+    assert_eq!(res.status(), tide::StatusCode::Ok);
+}