Allow route based function middlewares

src/route.rs

@@ -80,11 +80,11 @@ impl<'a, State: 'static> Route<'a, State> {
     /// Apply the given middleware to the current route.
     pub fn middleware<M>(&mut self, middleware: M) -> &mut Self
     where
-        M: Middleware<State> + Debug,
+        M: Middleware<State>,
     {
         log::trace!(
-            "Adding middleware {:?} to route {:?}",
-            middleware,
+            "Adding middleware {} to route {:?}",
+            middleware.name(),
             self.path
         );
         self.middleware.push(Arc::new(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);
+}