impl Service for MethodRouter<()>

tokio-rs · Jul 11, 2022 · 3beae1f · 3beae1f
1 parent 18bb70b
commit 3beae1f
Showing 1 changed file with 31 additions and 0 deletions.
diff --git a/axum/src/routing/method_routing.rs b/axum/src/routing/method_routing.rs
@@ -1048,6 +1048,24 @@ fn append_allow_header(allow_header: &mut AllowHeader, method: &'static str) {
     }
 }
 
+impl<B, E> Service<Request<B>> for MethodRouter<(), B, E>
+where
+    B: HttpBody,
+{
+    type Response = Response;
+    type Error = E;
+    type Future = RouteFuture<B, E>;
+
+    #[inline]
+    fn poll_ready(&mut self, _cx: &mut Context<'_>) -> Poll<Result<(), Self::Error>> {
+        Poll::Ready(Ok(()))
+    }
+
+    fn call(&mut self, req: Request<B>) -> Self::Future {
+        self.clone().with_state(()).call(req)
+    }
+}
+
 impl<S, B, E> Clone for MethodRouter<S, B, E> {
     fn clone(&self) -> Self {
         Self {
@@ -1207,6 +1225,19 @@ mod tests {
         assert!(body.is_empty());
     }
 
+    #[tokio::test]
+    async fn get_service_fn() {
+        async fn handle(_req: Request<Body>) -> Result<Response<Body>, Infallible> {
+            Ok(Response::new(Body::from("ok")))
+        }
+
+        let mut svc = get_service(service_fn(handle));
+
+        let (status, _, body) = call(Method::GET, &mut svc).await;
+        assert_eq!(status, StatusCode::OK);
+        assert_eq!(body, "ok");
+    }
+
     #[tokio::test]
     async fn get_handler() {
         let mut svc = MethodRouter::new().get(ok).with_state(());