Skip to main content

actix_web/
resource.rs

1use std::{cell::RefCell, fmt, future::Future, rc::Rc};
2
3use actix_http::Extensions;
4use actix_router::{IntoPatterns, Patterns};
5use actix_service::{
6    apply, apply_fn_factory, boxed, fn_service, IntoServiceFactory, Service, ServiceFactory,
7    ServiceFactoryExt, Transform,
8};
9use futures_core::future::LocalBoxFuture;
10use futures_util::future::join_all;
11
12use crate::{
13    body::MessageBody,
14    data::Data,
15    dev::{ensure_leading_slash, AppService, ResourceDef},
16    guard::{self, Guard},
17    handler::Handler,
18    http::header,
19    route::{Route, RouteService},
20    service::{
21        BoxedHttpService, BoxedHttpServiceFactory, HttpServiceFactory, ServiceRequest,
22        ServiceResponse,
23    },
24    web, Error, FromRequest, HttpResponse, Responder,
25};
26
27/// A collection of [`Route`]s that respond to the same path pattern.
28///
29/// Resource in turn has at least one route. Route consists of an handlers objects and list of
30/// guards (objects that implement `Guard` trait). Resources and routes uses builder-like pattern
31/// for configuration. During request handling, the resource object iterates through all routes
32/// and checks guards for the specific route, if the request matches all the guards, then the route
33/// is considered matched and the route handler gets called.
34///
35/// # Examples
36/// ```
37/// use actix_web::{web, App, HttpResponse};
38///
39/// let app = App::new().service(
40///     web::resource("/")
41///         .get(|| HttpResponse::Ok())
42///         .post(|| async { "Hello World!" })
43/// );
44/// ```
45///
46/// If no matching route is found, an empty 405 response is returned which includes an
47/// [appropriate Allow header][RFC 9110 §15.5.6]. This default behavior can be overridden using
48/// [`default_service()`](Self::default_service).
49///
50/// [RFC 9110 §15.5.6]: https://www.rfc-editor.org/rfc/rfc9110.html#section-15.5.6
51pub struct Resource<T = ResourceEndpoint> {
52    endpoint: T,
53    rdef: Patterns,
54    name: Option<String>,
55    routes: Vec<Route>,
56    app_data: Option<Extensions>,
57    guards: Vec<Box<dyn Guard>>,
58    default: BoxedHttpServiceFactory,
59    factory_ref: Rc<RefCell<Option<ResourceFactory>>>,
60}
61
62impl Resource {
63    /// Constructs new resource that matches a `path` pattern.
64    pub fn new<T: IntoPatterns>(path: T) -> Resource {
65        let factory_ref = Rc::new(RefCell::new(None));
66
67        Resource {
68            routes: Vec::new(),
69            rdef: path.patterns(),
70            name: None,
71            endpoint: ResourceEndpoint::new(Rc::clone(&factory_ref)),
72            factory_ref,
73            guards: Vec::new(),
74            app_data: None,
75            default: boxed::factory(fn_service(|req: ServiceRequest| async {
76                use crate::HttpMessage as _;
77
78                let allowed = req.extensions().get::<guard::RegisteredMethods>().cloned();
79
80                if let Some(methods) = allowed {
81                    Ok(req.into_response(
82                        HttpResponse::MethodNotAllowed()
83                            .insert_header(header::Allow(methods.0))
84                            .finish(),
85                    ))
86                } else {
87                    Ok(req.into_response(HttpResponse::MethodNotAllowed()))
88                }
89            })),
90        }
91    }
92}
93
94impl<T> Resource<T>
95where
96    T: ServiceFactory<ServiceRequest, Config = (), Error = Error, InitError = ()>,
97{
98    /// Set resource name.
99    ///
100    /// Name is used for url generation.
101    pub fn name(mut self, name: &str) -> Self {
102        self.name = Some(name.to_string());
103        self
104    }
105
106    /// Add match guard to a resource.
107    ///
108    /// ```
109    /// use actix_web::{web, guard, App, HttpResponse};
110    ///
111    /// async fn index(data: web::Path<(String, String)>) -> &'static str {
112    ///     "Welcome!"
113    /// }
114    ///
115    /// let app = App::new()
116    ///     .service(
117    ///         web::resource("/app")
118    ///             .guard(guard::Header("content-type", "text/plain"))
119    ///             .route(web::get().to(index))
120    ///     )
121    ///     .service(
122    ///         web::resource("/app")
123    ///             .guard(guard::Header("content-type", "text/json"))
124    ///             .route(web::get().to(|| HttpResponse::MethodNotAllowed()))
125    ///     );
126    /// ```
127    pub fn guard<G: Guard + 'static>(mut self, guard: G) -> Self {
128        self.guards.push(Box::new(guard));
129        self
130    }
131
132    pub(crate) fn add_guards(mut self, guards: Vec<Box<dyn Guard>>) -> Self {
133        self.guards.extend(guards);
134        self
135    }
136
137    /// Register a new route.
138    ///
139    /// ```
140    /// use actix_web::{web, guard, App, HttpResponse};
141    ///
142    /// let app = App::new().service(
143    ///     web::resource("/").route(
144    ///         web::route()
145    ///             .guard(guard::Any(guard::Get()).or(guard::Put()))
146    ///             .guard(guard::Header("Content-Type", "text/plain"))
147    ///             .to(|| HttpResponse::Ok()))
148    /// );
149    /// ```
150    ///
151    /// Multiple routes could be added to a resource. Resource object uses
152    /// match guards for route selection.
153    ///
154    /// ```
155    /// use actix_web::{web, guard, App};
156    ///
157    /// let app = App::new().service(
158    ///     web::resource("/container/")
159    ///          .route(web::get().to(get_handler))
160    ///          .route(web::post().to(post_handler))
161    ///          .route(web::delete().to(delete_handler))
162    /// );
163    ///
164    /// # async fn get_handler() -> impl actix_web::Responder { actix_web::HttpResponse::Ok() }
165    /// # async fn post_handler() -> impl actix_web::Responder { actix_web::HttpResponse::Ok() }
166    /// # async fn delete_handler() -> impl actix_web::Responder { actix_web::HttpResponse::Ok() }
167    /// ```
168    pub fn route(mut self, route: Route) -> Self {
169        self.routes.push(route);
170        self
171    }
172
173    /// Add resource data.
174    ///
175    /// Data of different types from parent contexts will still be accessible. Any `Data<T>` types
176    /// set here can be extracted in handlers using the `Data<T>` extractor.
177    ///
178    /// # Examples
179    /// ```
180    /// use std::cell::Cell;
181    /// use actix_web::{web, App, HttpRequest, HttpResponse, Responder};
182    ///
183    /// struct MyData {
184    ///     count: std::cell::Cell<usize>,
185    /// }
186    ///
187    /// async fn handler(req: HttpRequest, counter: web::Data<MyData>) -> impl Responder {
188    ///     // note this cannot use the Data<T> extractor because it was not added with it
189    ///     let incr = *req.app_data::<usize>().unwrap();
190    ///     assert_eq!(incr, 3);
191    ///
192    ///     // update counter using other value from app data
193    ///     counter.count.set(counter.count.get() + incr);
194    ///
195    ///     HttpResponse::Ok().body(counter.count.get().to_string())
196    /// }
197    ///
198    /// let app = App::new().service(
199    ///     web::resource("/")
200    ///         .app_data(3usize)
201    ///         .app_data(web::Data::new(MyData { count: Default::default() }))
202    ///         .route(web::get().to(handler))
203    /// );
204    /// ```
205    #[doc(alias = "manage")]
206    pub fn app_data<U: 'static>(mut self, data: U) -> Self {
207        self.app_data
208            .get_or_insert_with(Extensions::new)
209            .insert(data);
210
211        self
212    }
213
214    /// Add resource data after wrapping in `Data<T>`.
215    ///
216    /// Deprecated in favor of [`app_data`](Self::app_data).
217    #[deprecated(since = "4.0.0", note = "Use `.app_data(Data::new(val))` instead.")]
218    pub fn data<U: 'static>(self, data: U) -> Self {
219        self.app_data(Data::new(data))
220    }
221
222    /// Register a new route and add handler. This route matches all requests.
223    ///
224    /// ```
225    /// use actix_web::{App, HttpRequest, HttpResponse, web};
226    ///
227    /// async fn index(req: HttpRequest) -> HttpResponse {
228    ///     todo!()
229    /// }
230    ///
231    /// App::new().service(web::resource("/").to(index));
232    /// ```
233    ///
234    /// This is shortcut for:
235    ///
236    /// ```
237    /// # use actix_web::*;
238    /// # async fn index(req: HttpRequest) -> HttpResponse { todo!() }
239    /// App::new().service(web::resource("/").route(web::route().to(index)));
240    /// ```
241    pub fn to<F, Args>(mut self, handler: F) -> Self
242    where
243        F: Handler<Args>,
244        Args: FromRequest + 'static,
245        F::Output: Responder + 'static,
246    {
247        self.routes.push(Route::new().to(handler));
248        self
249    }
250
251    /// Registers a resource middleware.
252    ///
253    /// `mw` is a middleware component (type), that can modify the request and response across all
254    /// routes managed by this `Resource`.
255    ///
256    /// See [`App::wrap`](crate::App::wrap) for more details.
257    #[doc(alias = "middleware")]
258    #[doc(alias = "use")] // nodejs terminology
259    pub fn wrap<M, B>(
260        self,
261        mw: M,
262    ) -> Resource<
263        impl ServiceFactory<
264            ServiceRequest,
265            Config = (),
266            Response = ServiceResponse<B>,
267            Error = Error,
268            InitError = (),
269        >,
270    >
271    where
272        M: Transform<
273                T::Service,
274                ServiceRequest,
275                Response = ServiceResponse<B>,
276                Error = Error,
277                InitError = (),
278            > + 'static,
279        B: MessageBody,
280    {
281        Resource {
282            endpoint: apply(mw, self.endpoint),
283            rdef: self.rdef,
284            name: self.name,
285            guards: self.guards,
286            routes: self.routes,
287            default: self.default,
288            app_data: self.app_data,
289            factory_ref: self.factory_ref,
290        }
291    }
292
293    /// Registers a resource function middleware.
294    ///
295    /// `mw` is a closure that runs during inbound and/or outbound processing in the request
296    /// life-cycle (request -> response), modifying request/response as necessary, across all
297    /// requests handled by the `Resource`.
298    ///
299    /// See [`App::wrap_fn`](crate::App::wrap_fn) for examples and more details.
300    #[doc(alias = "middleware")]
301    #[doc(alias = "use")] // nodejs terminology
302    pub fn wrap_fn<F, R, B>(
303        self,
304        mw: F,
305    ) -> Resource<
306        impl ServiceFactory<
307            ServiceRequest,
308            Config = (),
309            Response = ServiceResponse<B>,
310            Error = Error,
311            InitError = (),
312        >,
313    >
314    where
315        F: Fn(ServiceRequest, &T::Service) -> R + Clone + 'static,
316        R: Future<Output = Result<ServiceResponse<B>, Error>>,
317        B: MessageBody,
318    {
319        Resource {
320            endpoint: apply_fn_factory(self.endpoint, mw),
321            rdef: self.rdef,
322            name: self.name,
323            guards: self.guards,
324            routes: self.routes,
325            default: self.default,
326            app_data: self.app_data,
327            factory_ref: self.factory_ref,
328        }
329    }
330
331    /// Sets the default service to be used if no matching route is found.
332    ///
333    /// Unlike [`Scope`]s, a `Resource` does _not_ inherit its parent's default service. You can
334    /// use a [`Route`] as default service.
335    ///
336    /// If a custom default service is not registered, an empty `405 Method Not Allowed` response
337    /// with an appropriate Allow header will be sent instead.
338    ///
339    /// # Examples
340    /// ```
341    /// use actix_web::{App, HttpResponse, web};
342    ///
343    /// let resource = web::resource("/test")
344    ///     .route(web::get().to(HttpResponse::Ok))
345    ///     .default_service(web::to(|| {
346    ///         HttpResponse::BadRequest()
347    ///     }));
348    ///
349    /// App::new().service(resource);
350    /// ```
351    ///
352    /// [`Scope`]: crate::Scope
353    pub fn default_service<F, U>(mut self, f: F) -> Self
354    where
355        F: IntoServiceFactory<U, ServiceRequest>,
356        U: ServiceFactory<ServiceRequest, Config = (), Response = ServiceResponse, Error = Error>
357            + 'static,
358        U::InitError: fmt::Debug,
359    {
360        // create and configure default resource
361        self.default = boxed::factory(f.into_factory().map_init_err(|err| {
362            log::error!("Can not construct default service: {err:?}");
363        }));
364
365        self
366    }
367}
368
369macro_rules! route_shortcut {
370    ($method_fn:ident, $method_upper:literal) => {
371        #[doc = concat!(" Adds a ", $method_upper, " route.")]
372        ///
373        /// Use [`route`](Self::route) if you need to add additional guards.
374        ///
375        /// # Examples
376        ///
377        /// ```
378        /// # use actix_web::web;
379        /// web::resource("/")
380        #[doc = concat!("    .", stringify!($method_fn), "(|| async { \"Hello World!\" })")]
381        /// # ;
382        /// ```
383        pub fn $method_fn<F, Args>(self, handler: F) -> Self
384        where
385            F: Handler<Args>,
386            Args: FromRequest + 'static,
387            F::Output: Responder + 'static,
388        {
389            self.route(web::$method_fn().to(handler))
390        }
391    };
392}
393
394/// Concise routes for well-known HTTP methods.
395impl<T> Resource<T>
396where
397    T: ServiceFactory<ServiceRequest, Config = (), Error = Error, InitError = ()>,
398{
399    route_shortcut!(get, "GET");
400    route_shortcut!(post, "POST");
401    route_shortcut!(put, "PUT");
402    route_shortcut!(patch, "PATCH");
403    route_shortcut!(delete, "DELETE");
404    route_shortcut!(head, "HEAD");
405    route_shortcut!(trace, "TRACE");
406}
407
408impl<T, B> HttpServiceFactory for Resource<T>
409where
410    T: ServiceFactory<
411            ServiceRequest,
412            Config = (),
413            Response = ServiceResponse<B>,
414            Error = Error,
415            InitError = (),
416        > + 'static,
417    B: MessageBody + 'static,
418{
419    fn register(mut self, config: &mut AppService) {
420        let routes = std::mem::take(&mut self.routes);
421
422        let guards = if self.guards.is_empty() {
423            None
424        } else {
425            Some(std::mem::take(&mut self.guards))
426        };
427
428        let mut rdef = if config.is_root() || !self.rdef.is_empty() {
429            ResourceDef::new(ensure_leading_slash(self.rdef.clone()))
430        } else {
431            ResourceDef::new(self.rdef.clone())
432        };
433        #[cfg(feature = "experimental-introspection")]
434        {
435            use crate::http::Method;
436
437            let full_paths = crate::introspection::expand_patterns(&config.current_prefix, &rdef);
438            let patterns = rdef
439                .pattern_iter()
440                .map(|pattern| pattern.to_string())
441                .collect::<Vec<_>>();
442            let guards_routes = routes.iter().map(|r| r.guards()).collect::<Vec<_>>();
443            let scope_id = config.scope_id_stack.last().copied();
444            let resource_guards: &[Box<dyn Guard>] = guards.as_deref().unwrap_or(&[]);
445            let resource_name = self.name.clone();
446
447            for route_guards in guards_routes {
448                // Log the guards and methods for introspection
449                let mut guard_names = Vec::new();
450                let mut methods = Vec::new();
451
452                for guard in resource_guards.iter().chain(route_guards.iter()) {
453                    guard_names.push(guard.name());
454                    methods.extend(
455                        guard
456                            .details()
457                            .unwrap_or_default()
458                            .into_iter()
459                            .flat_map(|d| {
460                                if let crate::guard::GuardDetail::HttpMethods(v) = d {
461                                    v.into_iter()
462                                        .filter_map(|s| s.parse::<Method>().ok())
463                                        .collect::<Vec<_>>()
464                                } else {
465                                    Vec::new()
466                                }
467                            }),
468                    );
469                }
470
471                let guard_details = crate::introspection::guard_reports_from_iter(
472                    resource_guards.iter().chain(route_guards.iter()),
473                );
474
475                for full_path in &full_paths {
476                    let info = crate::introspection::RouteInfo::new(
477                        full_path.clone(),
478                        methods.clone(),
479                        guard_names.clone(),
480                        guard_details.clone(),
481                        patterns.clone(),
482                        resource_name.clone(),
483                    );
484                    config
485                        .introspector
486                        .borrow_mut()
487                        .register_route(info, scope_id);
488                }
489            }
490        }
491
492        if let Some(ref name) = self.name {
493            rdef.set_name(name);
494        }
495
496        *self.factory_ref.borrow_mut() = Some(ResourceFactory {
497            routes,
498            default: self.default,
499        });
500
501        let resource_data = self.app_data.map(Rc::new);
502
503        // wraps endpoint service (including middleware) call and injects app data for this scope
504        let endpoint = apply_fn_factory(self.endpoint, move |mut req: ServiceRequest, srv| {
505            if let Some(ref data) = resource_data {
506                req.add_data_container(Rc::clone(data));
507            }
508
509            let fut = srv.call(req);
510
511            async { Ok(fut.await?.map_into_boxed_body()) }
512        });
513
514        config.register_service(rdef, guards, endpoint, None)
515    }
516}
517
518pub struct ResourceFactory {
519    routes: Vec<Route>,
520    default: BoxedHttpServiceFactory,
521}
522
523impl ServiceFactory<ServiceRequest> for ResourceFactory {
524    type Response = ServiceResponse;
525    type Error = Error;
526    type Config = ();
527    type Service = ResourceService;
528    type InitError = ();
529    type Future = LocalBoxFuture<'static, Result<Self::Service, Self::InitError>>;
530
531    fn new_service(&self, _: ()) -> Self::Future {
532        // construct default service factory future.
533        let default_fut = self.default.new_service(());
534
535        // construct route service factory futures
536        let factory_fut = join_all(self.routes.iter().map(|route| route.new_service(())));
537
538        Box::pin(async move {
539            let default = default_fut.await?;
540            let routes = factory_fut
541                .await
542                .into_iter()
543                .collect::<Result<Vec<_>, _>>()?;
544
545            Ok(ResourceService { routes, default })
546        })
547    }
548}
549
550pub struct ResourceService {
551    routes: Vec<RouteService>,
552    default: BoxedHttpService,
553}
554
555impl Service<ServiceRequest> for ResourceService {
556    type Response = ServiceResponse;
557    type Error = Error;
558    type Future = LocalBoxFuture<'static, Result<Self::Response, Self::Error>>;
559
560    actix_service::always_ready!();
561
562    fn call(&self, mut req: ServiceRequest) -> Self::Future {
563        for route in &self.routes {
564            if route.check(&mut req) {
565                return route.call(req);
566            }
567        }
568
569        self.default.call(req)
570    }
571}
572
573#[doc(hidden)]
574pub struct ResourceEndpoint {
575    factory: Rc<RefCell<Option<ResourceFactory>>>,
576}
577
578impl ResourceEndpoint {
579    fn new(factory: Rc<RefCell<Option<ResourceFactory>>>) -> Self {
580        ResourceEndpoint { factory }
581    }
582}
583
584impl ServiceFactory<ServiceRequest> for ResourceEndpoint {
585    type Response = ServiceResponse;
586    type Error = Error;
587    type Config = ();
588    type Service = ResourceService;
589    type InitError = ();
590    type Future = LocalBoxFuture<'static, Result<Self::Service, Self::InitError>>;
591
592    fn new_service(&self, _: ()) -> Self::Future {
593        self.factory.borrow().as_ref().unwrap().new_service(())
594    }
595}
596
597#[cfg(test)]
598mod tests {
599    use std::time::Duration;
600
601    use actix_rt::time::sleep;
602    use actix_utils::future::ok;
603
604    use super::*;
605    use crate::{
606        http::{header::HeaderValue, Method, StatusCode},
607        middleware::DefaultHeaders,
608        test::{call_service, init_service, TestRequest},
609        App, HttpMessage,
610    };
611
612    #[test]
613    fn can_be_returned_from_fn() {
614        fn my_resource_1() -> Resource {
615            web::resource("/test1").route(web::get().to(|| async { "hello" }))
616        }
617
618        fn my_resource_2() -> Resource<
619            impl ServiceFactory<
620                ServiceRequest,
621                Config = (),
622                Response = ServiceResponse<impl MessageBody>,
623                Error = Error,
624                InitError = (),
625            >,
626        > {
627            web::resource("/test2")
628                .wrap_fn(|req, srv| {
629                    let fut = srv.call(req);
630                    async { Ok(fut.await?.map_into_right_body::<()>()) }
631                })
632                .route(web::get().to(|| async { "hello" }))
633        }
634
635        fn my_resource_3() -> impl HttpServiceFactory {
636            web::resource("/test3").route(web::get().to(|| async { "hello" }))
637        }
638
639        App::new()
640            .service(my_resource_1())
641            .service(my_resource_2())
642            .service(my_resource_3());
643    }
644
645    #[actix_rt::test]
646    async fn test_middleware() {
647        let srv = init_service(
648            App::new().service(
649                web::resource("/test")
650                    .name("test")
651                    .wrap(
652                        DefaultHeaders::new()
653                            .add((header::CONTENT_TYPE, HeaderValue::from_static("0001"))),
654                    )
655                    .route(web::get().to(HttpResponse::Ok)),
656            ),
657        )
658        .await;
659        let req = TestRequest::with_uri("/test").to_request();
660        let resp = call_service(&srv, req).await;
661        assert_eq!(resp.status(), StatusCode::OK);
662        assert_eq!(
663            resp.headers().get(header::CONTENT_TYPE).unwrap(),
664            HeaderValue::from_static("0001")
665        );
666    }
667
668    #[actix_rt::test]
669    async fn test_middleware_fn() {
670        let srv = init_service(
671            App::new().service(
672                web::resource("/test")
673                    .wrap_fn(|req, srv| {
674                        let fut = srv.call(req);
675                        async {
676                            fut.await.map(|mut res| {
677                                res.headers_mut()
678                                    .insert(header::CONTENT_TYPE, HeaderValue::from_static("0001"));
679                                res
680                            })
681                        }
682                    })
683                    .route(web::get().to(HttpResponse::Ok)),
684            ),
685        )
686        .await;
687        let req = TestRequest::with_uri("/test").to_request();
688        let resp = call_service(&srv, req).await;
689        assert_eq!(resp.status(), StatusCode::OK);
690        assert_eq!(
691            resp.headers().get(header::CONTENT_TYPE).unwrap(),
692            HeaderValue::from_static("0001")
693        );
694    }
695
696    #[actix_rt::test]
697    async fn test_to() {
698        let srv = init_service(App::new().service(web::resource("/test").to(|| async {
699            sleep(Duration::from_millis(100)).await;
700            Ok::<_, Error>(HttpResponse::Ok())
701        })))
702        .await;
703        let req = TestRequest::with_uri("/test").to_request();
704        let resp = call_service(&srv, req).await;
705        assert_eq!(resp.status(), StatusCode::OK);
706    }
707
708    #[actix_rt::test]
709    async fn test_pattern() {
710        let srv = init_service(App::new().service(
711            web::resource(["/test", "/test2"]).to(|| async { Ok::<_, Error>(HttpResponse::Ok()) }),
712        ))
713        .await;
714        let req = TestRequest::with_uri("/test").to_request();
715        let resp = call_service(&srv, req).await;
716        assert_eq!(resp.status(), StatusCode::OK);
717        let req = TestRequest::with_uri("/test2").to_request();
718        let resp = call_service(&srv, req).await;
719        assert_eq!(resp.status(), StatusCode::OK);
720    }
721
722    #[actix_rt::test]
723    async fn test_default_resource() {
724        let srv = init_service(
725            App::new()
726                .service(
727                    web::resource("/test")
728                        .route(web::get().to(HttpResponse::Ok))
729                        .route(web::delete().to(HttpResponse::Ok)),
730                )
731                .default_service(|r: ServiceRequest| {
732                    ok(r.into_response(HttpResponse::BadRequest()))
733                }),
734        )
735        .await;
736        let req = TestRequest::with_uri("/test").to_request();
737        let resp = call_service(&srv, req).await;
738        assert_eq!(resp.status(), StatusCode::OK);
739
740        let req = TestRequest::with_uri("/test")
741            .method(Method::POST)
742            .to_request();
743        let resp = call_service(&srv, req).await;
744        assert_eq!(resp.status(), StatusCode::METHOD_NOT_ALLOWED);
745        assert_eq!(
746            resp.headers().get(header::ALLOW).unwrap().as_bytes(),
747            b"GET, DELETE"
748        );
749
750        let srv = init_service(
751            App::new().service(
752                web::resource("/test")
753                    .route(web::get().to(HttpResponse::Ok))
754                    .default_service(|r: ServiceRequest| {
755                        ok(r.into_response(HttpResponse::BadRequest()))
756                    }),
757            ),
758        )
759        .await;
760
761        let req = TestRequest::with_uri("/test").to_request();
762        let resp = call_service(&srv, req).await;
763        assert_eq!(resp.status(), StatusCode::OK);
764
765        let req = TestRequest::with_uri("/test")
766            .method(Method::POST)
767            .to_request();
768        let resp = call_service(&srv, req).await;
769        assert_eq!(resp.status(), StatusCode::BAD_REQUEST);
770    }
771
772    #[actix_rt::test]
773    async fn test_resource_guards() {
774        let srv = init_service(
775            App::new()
776                .service(
777                    web::resource("/test/{p}")
778                        .guard(guard::Get())
779                        .to(HttpResponse::Ok),
780                )
781                .service(
782                    web::resource("/test/{p}")
783                        .guard(guard::Put())
784                        .to(HttpResponse::Created),
785                )
786                .service(
787                    web::resource("/test/{p}")
788                        .guard(guard::Delete())
789                        .to(HttpResponse::NoContent),
790                ),
791        )
792        .await;
793
794        let req = TestRequest::with_uri("/test/it")
795            .method(Method::GET)
796            .to_request();
797        let resp = call_service(&srv, req).await;
798        assert_eq!(resp.status(), StatusCode::OK);
799
800        let req = TestRequest::with_uri("/test/it")
801            .method(Method::PUT)
802            .to_request();
803        let resp = call_service(&srv, req).await;
804        assert_eq!(resp.status(), StatusCode::CREATED);
805
806        let req = TestRequest::with_uri("/test/it")
807            .method(Method::DELETE)
808            .to_request();
809        let resp = call_service(&srv, req).await;
810        assert_eq!(resp.status(), StatusCode::NO_CONTENT);
811    }
812
813    // allow deprecated `{App, Resource}::data`
814    #[allow(deprecated)]
815    #[actix_rt::test]
816    async fn test_data() {
817        let srv = init_service(
818            App::new()
819                .data(1.0f64)
820                .data(1usize)
821                .app_data(web::Data::new('-'))
822                .service(
823                    web::resource("/test")
824                        .data(10usize)
825                        .app_data(web::Data::new('*'))
826                        .guard(guard::Get())
827                        .to(
828                            |data1: web::Data<usize>,
829                             data2: web::Data<char>,
830                             data3: web::Data<f64>| {
831                                assert_eq!(**data1, 10);
832                                assert_eq!(**data2, '*');
833                                let error = f64::EPSILON;
834                                assert!((**data3 - 1.0).abs() < error);
835                                HttpResponse::Ok()
836                            },
837                        ),
838                ),
839        )
840        .await;
841
842        let req = TestRequest::get().uri("/test").to_request();
843        let resp = call_service(&srv, req).await;
844        assert_eq!(resp.status(), StatusCode::OK);
845    }
846
847    // allow deprecated `{App, Resource}::data`
848    #[allow(deprecated)]
849    #[actix_rt::test]
850    async fn test_data_default_service() {
851        let srv =
852            init_service(
853                App::new().data(1usize).service(
854                    web::resource("/test")
855                        .data(10usize)
856                        .default_service(web::to(|data: web::Data<usize>| {
857                            assert_eq!(**data, 10);
858                            HttpResponse::Ok()
859                        })),
860                ),
861            )
862            .await;
863
864        let req = TestRequest::get().uri("/test").to_request();
865        let resp = call_service(&srv, req).await;
866        assert_eq!(resp.status(), StatusCode::OK);
867    }
868
869    #[actix_rt::test]
870    async fn test_middleware_app_data() {
871        let srv = init_service(
872            App::new().service(
873                web::resource("test")
874                    .app_data(1usize)
875                    .wrap_fn(|req, srv| {
876                        assert_eq!(req.app_data::<usize>(), Some(&1usize));
877                        req.extensions_mut().insert(1usize);
878                        srv.call(req)
879                    })
880                    .route(web::get().to(HttpResponse::Ok))
881                    .default_service(|req: ServiceRequest| async move {
882                        let (req, _) = req.into_parts();
883
884                        assert_eq!(req.extensions().get::<usize>(), Some(&1));
885
886                        Ok(ServiceResponse::new(
887                            req,
888                            HttpResponse::BadRequest().finish(),
889                        ))
890                    }),
891            ),
892        )
893        .await;
894
895        let req = TestRequest::get().uri("/test").to_request();
896        let resp = call_service(&srv, req).await;
897        assert_eq!(resp.status(), StatusCode::OK);
898
899        let req = TestRequest::post().uri("/test").to_request();
900        let resp = call_service(&srv, req).await;
901        assert_eq!(resp.status(), StatusCode::BAD_REQUEST);
902    }
903
904    #[actix_rt::test]
905    async fn test_middleware_body_type() {
906        let srv = init_service(
907            App::new().service(
908                web::resource("/test")
909                    .wrap_fn(|req, srv| {
910                        let fut = srv.call(req);
911                        async { Ok(fut.await?.map_into_right_body::<()>()) }
912                    })
913                    .route(web::get().to(|| async { "hello" })),
914            ),
915        )
916        .await;
917
918        // test if `try_into_bytes()` is preserved across scope layer
919        use actix_http::body::MessageBody as _;
920        let req = TestRequest::with_uri("/test").to_request();
921        let resp = call_service(&srv, req).await;
922        let body = resp.into_body();
923        assert_eq!(body.try_into_bytes().unwrap(), b"hello".as_ref());
924    }
925}