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);
+}