1
0
mirror of https://github.com/fafhrd91/actix-web synced 2024-11-24 08:22:59 +01:00
actix-web/src/scope.rs

1124 lines
36 KiB
Rust
Raw Normal View History

2019-03-04 06:02:01 +01:00
use std::cell::RefCell;
use std::fmt;
2020-05-18 04:47:20 +02:00
use std::future::Future;
2019-03-04 06:02:01 +01:00
use std::rc::Rc;
2021-01-09 19:06:49 +01:00
use std::task::Poll;
2019-03-04 06:02:01 +01:00
2021-01-09 19:06:49 +01:00
use actix_http::Extensions;
use actix_router::{ResourceDef, Router};
2019-11-26 06:25:50 +01:00
use actix_service::boxed::{self, BoxService, BoxServiceFactory};
2019-03-04 06:02:01 +01:00
use actix_service::{
apply, apply_fn_factory, IntoServiceFactory, Service, ServiceFactory,
ServiceFactoryExt, Transform,
2019-03-04 06:02:01 +01:00
};
use futures_core::future::LocalBoxFuture;
2021-01-09 19:06:49 +01:00
use futures_util::future::join_all;
2019-03-04 06:02:01 +01:00
use crate::config::ServiceConfig;
use crate::data::Data;
2019-04-15 16:32:49 +02:00
use crate::dev::{AppService, HttpServiceFactory};
use crate::error::Error;
2019-03-04 06:02:01 +01:00
use crate::guard::Guard;
use crate::resource::Resource;
use crate::rmap::ResourceMap;
2019-03-04 06:02:01 +01:00
use crate::route::Route;
use crate::service::{
2019-11-20 18:33:22 +01:00
AppServiceFactory, ServiceFactoryWrapper, ServiceRequest, ServiceResponse,
};
2019-03-04 06:02:01 +01:00
2019-07-17 11:48:37 +02:00
type Guards = Vec<Box<dyn Guard>>;
2019-11-26 06:25:50 +01:00
type HttpService = BoxService<ServiceRequest, ServiceResponse, Error>;
type HttpNewService = BoxServiceFactory<(), ServiceRequest, ServiceResponse, Error, ()>;
2019-03-04 06:02:01 +01:00
/// Resources scope.
2019-03-04 06:02:01 +01:00
///
/// Scope is a set of resources with common root path.
/// Scopes collect multiple paths under a common path prefix.
/// Scope path can contain variable path segments as resources.
/// Scope prefix is always complete path segment, i.e `/app` would
/// be converted to a `/app/` and it would not match `/app` path.
///
/// You can get variable path segments from `HttpRequest::match_info()`.
/// `Path` extractor also is able to extract scope level variable segments.
///
/// ```rust
/// use actix_web::{web, App, HttpResponse};
2019-03-04 06:02:01 +01:00
///
/// fn main() {
/// let app = App::new().service(
/// web::scope("/{project_id}/")
2019-11-21 16:34:04 +01:00
/// .service(web::resource("/path1").to(|| async { HttpResponse::Ok() }))
/// .service(web::resource("/path2").route(web::get().to(|| HttpResponse::Ok())))
/// .service(web::resource("/path3").route(web::head().to(|| HttpResponse::MethodNotAllowed())))
/// );
2019-03-04 06:02:01 +01:00
/// }
/// ```
///
/// In the above example three routes get registered:
/// * /{project_id}/path1 - responds to all http method
2019-03-04 06:02:01 +01:00
/// * /{project_id}/path2 - `GET` requests
/// * /{project_id}/path3 - `HEAD` requests
pub struct Scope<T = ScopeEndpoint> {
2019-03-04 06:02:01 +01:00
endpoint: T,
rdef: String,
2021-01-09 19:06:49 +01:00
app_data: Option<Extensions>,
2019-11-20 18:33:22 +01:00
services: Vec<Box<dyn AppServiceFactory>>,
2019-07-17 11:48:37 +02:00
guards: Vec<Box<dyn Guard>>,
2021-01-09 19:06:49 +01:00
default: Option<Rc<HttpNewService>>,
external: Vec<ResourceDef>,
factory_ref: Rc<RefCell<Option<ScopeFactory>>>,
2019-03-04 06:02:01 +01:00
}
impl Scope {
2019-03-04 06:02:01 +01:00
/// Create a new scope
pub fn new(path: &str) -> Scope {
2019-03-04 06:02:01 +01:00
let fref = Rc::new(RefCell::new(None));
Scope {
endpoint: ScopeEndpoint::new(fref.clone()),
rdef: path.to_string(),
2021-01-09 19:06:49 +01:00
app_data: None,
2019-03-04 20:47:53 +01:00
guards: Vec::new(),
2019-03-04 06:02:01 +01:00
services: Vec::new(),
2021-01-09 19:06:49 +01:00
default: None,
external: Vec::new(),
2019-03-04 06:02:01 +01:00
factory_ref: fref,
}
}
}
impl<T> Scope<T>
2019-03-04 06:02:01 +01:00
where
2019-11-20 18:33:22 +01:00
T: ServiceFactory<
ServiceRequest,
2019-05-12 17:34:51 +02:00
Config = (),
2019-03-04 06:02:01 +01:00
Response = ServiceResponse,
Error = Error,
2019-03-04 06:02:01 +01:00
InitError = (),
>,
{
/// Add match guard to a scope.
2019-03-04 06:02:01 +01:00
///
/// ```rust
/// use actix_web::{web, guard, App, HttpRequest, HttpResponse};
2019-03-04 06:02:01 +01:00
///
2019-11-21 16:34:04 +01:00
/// async fn index(data: web::Path<(String, String)>) -> &'static str {
2019-03-04 06:02:01 +01:00
/// "Welcome!"
/// }
///
/// fn main() {
/// let app = App::new().service(
/// web::scope("/app")
2019-03-04 06:02:01 +01:00
/// .guard(guard::Header("content-type", "text/plain"))
/// .route("/test1", web::get().to(index))
2019-03-04 06:02:01 +01:00
/// .route("/test2", web::post().to(|r: HttpRequest| {
/// HttpResponse::MethodNotAllowed()
/// }))
/// );
2019-03-04 06:02:01 +01:00
/// }
/// ```
pub fn guard<G: Guard + 'static>(mut self, guard: G) -> Self {
2019-03-04 20:47:53 +01:00
self.guards.push(Box::new(guard));
2019-03-04 06:02:01 +01:00
self
}
/// Set or override application data. Application data could be accessed
/// by using `Data<T>` extractor where `T` is data type.
///
/// ```rust
/// use std::cell::Cell;
/// use actix_web::{web, App, HttpResponse, Responder};
///
/// struct MyData {
/// counter: Cell<usize>,
/// }
///
/// async fn index(data: web::Data<MyData>) -> impl Responder {
/// data.counter.set(data.counter.get() + 1);
/// HttpResponse::Ok()
/// }
///
/// fn main() {
/// let app = App::new().service(
/// web::scope("/app")
/// .data(MyData{ counter: Cell::new(0) })
/// .service(
/// web::resource("/index.html").route(
/// web::get().to(index)))
/// );
/// }
/// ```
pub fn data<U: 'static>(self, data: U) -> Self {
self.app_data(Data::new(data))
}
/// Add scope data.
///
/// Data of different types from parent contexts will still be accessible.
pub fn app_data<U: 'static>(mut self, data: U) -> Self {
self.app_data
.get_or_insert_with(Extensions::new)
.insert(data);
self
}
/// Run external configuration as part of the scope building
/// process
///
/// This function is useful for moving parts of configuration to a
/// different module or even library. For example,
/// some of the resource's configuration could be moved to different module.
///
/// ```rust
/// # extern crate actix_web;
/// use actix_web::{web, middleware, App, HttpResponse};
///
/// // this function could be located in different module
/// fn config(cfg: &mut web::ServiceConfig) {
/// cfg.service(web::resource("/test")
/// .route(web::get().to(|| HttpResponse::Ok()))
/// .route(web::head().to(|| HttpResponse::MethodNotAllowed()))
/// );
/// }
///
/// fn main() {
/// let app = App::new()
/// .wrap(middleware::Logger::default())
/// .service(
/// web::scope("/api")
/// .configure(config)
/// )
/// .route("/index.html", web::get().to(|| HttpResponse::Ok()));
/// }
/// ```
pub fn configure<F>(mut self, f: F) -> Self
where
F: FnOnce(&mut ServiceConfig),
{
let mut cfg = ServiceConfig::new();
f(&mut cfg);
self.services.extend(cfg.services);
self.external.extend(cfg.external);
2021-01-09 19:06:49 +01:00
self.app_data
.get_or_insert_with(Extensions::new)
.extend(cfg.app_data);
self
}
2021-02-10 13:12:03 +01:00
/// Register HTTP service.
///
/// This is similar to `App's` service registration.
///
2021-02-10 13:12:03 +01:00
/// Actix Web provides several services implementations:
///
/// * *Resource* is an entry in resource table which corresponds to requested URL.
/// * *Scope* is a set of resources with common root path.
/// * "StaticFiles" is a service for static files support
2019-03-04 06:02:01 +01:00
///
/// ```rust
/// use actix_web::{web, App, HttpRequest};
2019-03-04 06:02:01 +01:00
///
/// struct AppState;
///
2019-11-21 16:34:04 +01:00
/// async fn index(req: HttpRequest) -> &'static str {
2019-03-04 06:02:01 +01:00
/// "Welcome!"
/// }
///
/// fn main() {
/// let app = App::new().service(
/// web::scope("/app").service(
/// web::scope("/v1")
/// .service(web::resource("/test1").to(index)))
/// );
2019-03-04 06:02:01 +01:00
/// }
/// ```
pub fn service<F>(mut self, factory: F) -> Self
2019-03-04 06:02:01 +01:00
where
F: HttpServiceFactory + 'static,
2019-03-04 06:02:01 +01:00
{
self.services
.push(Box::new(ServiceFactoryWrapper::new(factory)));
2019-03-04 06:02:01 +01:00
self
}
/// Configure route for a specific path.
///
/// This is a simplified version of the `Scope::service()` method.
/// This method can be called multiple times, in that case
2019-03-04 06:02:01 +01:00
/// multiple resources with one route would be registered for same resource path.
///
/// ```rust
/// use actix_web::{web, App, HttpResponse};
2019-03-04 06:02:01 +01:00
///
2019-11-21 16:34:04 +01:00
/// async fn index(data: web::Path<(String, String)>) -> &'static str {
2019-03-04 06:02:01 +01:00
/// "Welcome!"
/// }
///
/// fn main() {
/// let app = App::new().service(
/// web::scope("/app")
/// .route("/test1", web::get().to(index))
2019-03-04 06:02:01 +01:00
/// .route("/test2", web::post().to(|| HttpResponse::MethodNotAllowed()))
/// );
2019-03-04 06:02:01 +01:00
/// }
/// ```
pub fn route(self, path: &str, mut route: Route) -> Self {
self.service(
Resource::new(path)
.add_guards(route.take_guards())
.route(route),
)
2019-03-04 06:02:01 +01:00
}
/// Default service to be used if no matching route could be found.
///
/// If default resource is not registered, app's default resource is being used.
pub fn default_service<F, U>(mut self, f: F) -> Self
2019-03-04 06:02:01 +01:00
where
F: IntoServiceFactory<U, ServiceRequest>,
2019-11-20 18:33:22 +01:00
U: ServiceFactory<
ServiceRequest,
2019-05-12 17:34:51 +02:00
Config = (),
2019-03-04 06:02:01 +01:00
Response = ServiceResponse,
Error = Error,
2019-03-04 06:02:01 +01:00
> + 'static,
U::InitError: fmt::Debug,
2019-03-04 06:02:01 +01:00
{
// create and configure default resource
2021-01-09 19:06:49 +01:00
self.default = Some(Rc::new(boxed::factory(f.into_factory().map_init_err(
|e| log::error!("Can not construct default service: {:?}", e),
))));
2019-03-04 06:02:01 +01:00
self
}
2019-04-02 21:51:16 +02:00
/// Registers middleware, in the form of a middleware component (type),
/// that runs during inbound processing in the request
2020-04-21 05:09:35 +02:00
/// life-cycle (request -> response), modifying request as
2019-04-01 21:10:28 +02:00
/// necessary, across all requests managed by the *Scope*. Scope-level
/// middleware is more limited in what it can modify, relative to Route or
2019-04-02 21:51:16 +02:00
/// Application level middleware, in that Scope-level middleware can not modify
2019-04-01 21:10:28 +02:00
/// ServiceResponse.
2019-03-04 06:02:01 +01:00
///
/// Use middleware when you need to read or modify *every* request in some way.
2019-11-20 18:33:22 +01:00
pub fn wrap<M>(
2019-03-04 06:02:01 +01:00
self,
2019-11-20 18:33:22 +01:00
mw: M,
2019-03-04 06:02:01 +01:00
) -> Scope<
2019-11-20 18:33:22 +01:00
impl ServiceFactory<
ServiceRequest,
2019-05-12 17:34:51 +02:00
Config = (),
2019-03-04 06:02:01 +01:00
Response = ServiceResponse,
Error = Error,
2019-03-04 06:02:01 +01:00
InitError = (),
>,
>
where
2019-03-05 06:37:57 +01:00
M: Transform<
2019-03-04 06:02:01 +01:00
T::Service,
ServiceRequest,
2019-03-04 06:02:01 +01:00
Response = ServiceResponse,
Error = Error,
2019-03-04 06:02:01 +01:00
InitError = (),
>,
{
Scope {
2019-11-20 18:33:22 +01:00
endpoint: apply(mw, self.endpoint),
2019-03-04 06:02:01 +01:00
rdef: self.rdef,
2021-01-09 19:06:49 +01:00
app_data: self.app_data,
2019-03-04 06:02:01 +01:00
guards: self.guards,
services: self.services,
default: self.default,
external: self.external,
2019-03-04 06:02:01 +01:00
factory_ref: self.factory_ref,
}
}
2019-03-25 21:43:02 +01:00
/// Registers middleware, in the form of a closure, that runs during inbound
2020-04-21 05:09:35 +02:00
/// processing in the request life-cycle (request -> response), modifying
2019-04-25 20:14:32 +02:00
/// request as necessary, across all requests managed by the *Scope*.
2019-04-01 21:10:28 +02:00
/// Scope-level middleware is more limited in what it can modify, relative
/// to Route or Application level middleware, in that Scope-level middleware
/// can not modify ServiceResponse.
2019-03-25 21:43:02 +01:00
///
/// ```rust
/// use actix_service::Service;
/// use actix_web::{web, App};
/// use actix_web::http::{header::CONTENT_TYPE, HeaderValue};
///
2019-11-21 16:34:04 +01:00
/// async fn index() -> &'static str {
2019-03-25 21:43:02 +01:00
/// "Welcome!"
/// }
///
/// fn main() {
/// let app = App::new().service(
/// web::scope("/app")
2019-11-20 18:33:22 +01:00
/// .wrap_fn(|req, srv| {
/// let fut = srv.call(req);
/// async {
/// let mut res = fut.await?;
2019-03-25 21:43:02 +01:00
/// res.headers_mut().insert(
/// CONTENT_TYPE, HeaderValue::from_static("text/plain"),
/// );
2019-11-20 18:33:22 +01:00
/// Ok(res)
/// }
/// })
2019-03-25 21:43:02 +01:00
/// .route("/index.html", web::get().to(index)));
/// }
/// ```
pub fn wrap_fn<F, R>(
self,
mw: F,
) -> Scope<
2019-11-20 18:33:22 +01:00
impl ServiceFactory<
ServiceRequest,
2019-05-12 17:34:51 +02:00
Config = (),
2019-03-25 21:43:02 +01:00
Response = ServiceResponse,
Error = Error,
InitError = (),
>,
>
where
F: Fn(ServiceRequest, &T::Service) -> R + Clone,
2019-11-20 18:33:22 +01:00
R: Future<Output = Result<ServiceResponse, Error>>,
2019-03-25 21:43:02 +01:00
{
2019-11-20 18:33:22 +01:00
Scope {
endpoint: apply_fn_factory(self.endpoint, mw),
rdef: self.rdef,
2021-01-09 19:06:49 +01:00
app_data: self.app_data,
2019-11-20 18:33:22 +01:00
guards: self.guards,
services: self.services,
default: self.default,
external: self.external,
factory_ref: self.factory_ref,
}
2019-03-25 21:43:02 +01:00
}
2019-03-04 06:02:01 +01:00
}
impl<T> HttpServiceFactory for Scope<T>
2019-03-04 06:02:01 +01:00
where
2019-11-20 18:33:22 +01:00
T: ServiceFactory<
ServiceRequest,
2019-05-12 17:34:51 +02:00
Config = (),
Response = ServiceResponse,
Error = Error,
InitError = (),
> + 'static,
2019-03-04 06:02:01 +01:00
{
fn register(mut self, config: &mut AppService) {
// update default resource if needed
2021-01-09 19:06:49 +01:00
let default = self.default.unwrap_or_else(|| config.default_service());
2019-03-04 06:02:01 +01:00
// register nested services
let mut cfg = config.clone_config();
self.services
.into_iter()
.for_each(|mut srv| srv.register(&mut cfg));
let mut rmap = ResourceMap::new(ResourceDef::root_prefix(&self.rdef));
// external resources
2020-05-17 03:54:42 +02:00
for mut rdef in std::mem::take(&mut self.external) {
rmap.add(&mut rdef, None);
}
// complete scope pipeline creation
2019-03-04 06:02:01 +01:00
*self.factory_ref.borrow_mut() = Some(ScopeFactory {
2021-01-09 19:06:49 +01:00
app_data: self.app_data.take().map(Rc::new),
default,
2020-10-10 02:26:05 +02:00
services: cfg
.into_services()
.1
.into_iter()
.map(|(mut rdef, srv, guards, nested)| {
rmap.add(&mut rdef, nested);
(rdef, srv, RefCell::new(guards))
})
.collect::<Vec<_>>()
.into_boxed_slice()
.into(),
2019-03-04 06:02:01 +01:00
});
// get guards
let guards = if self.guards.is_empty() {
None
} else {
Some(self.guards)
};
// register final service
2019-03-07 04:27:18 +01:00
config.register_service(
ResourceDef::root_prefix(&self.rdef),
guards,
self.endpoint,
Some(Rc::new(rmap)),
2019-03-07 04:27:18 +01:00
)
2019-03-04 06:02:01 +01:00
}
}
pub struct ScopeFactory {
2021-01-09 19:06:49 +01:00
app_data: Option<Rc<Extensions>>,
2020-10-10 02:26:05 +02:00
services: Rc<[(ResourceDef, HttpNewService, RefCell<Option<Guards>>)]>,
2021-01-09 19:06:49 +01:00
default: Rc<HttpNewService>,
2019-03-04 06:02:01 +01:00
}
impl ServiceFactory<ServiceRequest> for ScopeFactory {
2019-03-04 06:02:01 +01:00
type Response = ServiceResponse;
type Error = Error;
2021-01-09 19:06:49 +01:00
type Config = ();
type Service = ScopeService;
2021-01-09 19:06:49 +01:00
type InitError = ();
type Future = LocalBoxFuture<'static, Result<Self::Service, Self::InitError>>;
2019-03-04 06:02:01 +01:00
2019-12-02 16:37:13 +01:00
fn new_service(&self, _: ()) -> Self::Future {
2021-01-09 19:06:49 +01:00
// construct default service factory future
let default_fut = self.default.new_service(());
// construct all services factory future with it's resource def and guards.
let factory_fut =
join_all(self.services.iter().map(|(path, factory, guards)| {
let path = path.clone();
let guards = guards.borrow_mut().take();
let factory_fut = factory.new_service(());
async move {
let service = factory_fut.await?;
Ok((path, guards, service))
}
}));
2019-03-04 06:02:01 +01:00
2021-01-09 19:06:49 +01:00
let app_data = self.app_data.clone();
2019-03-04 06:02:01 +01:00
2021-01-09 19:06:49 +01:00
Box::pin(async move {
let default = default_fut.await?;
2019-03-04 06:02:01 +01:00
2021-01-09 19:06:49 +01:00
// build router from the factory future result.
let router = factory_fut
.await
.into_iter()
.collect::<Result<Vec<_>, _>>()?
2019-03-04 06:02:01 +01:00
.drain(..)
2021-01-09 19:06:49 +01:00
.fold(Router::build(), |mut router, (path, guards, service)| {
router.rdef(path, service).2 = guards;
2019-03-04 06:02:01 +01:00
router
2021-01-09 19:06:49 +01:00
})
.finish();
Ok(ScopeService {
app_data,
router,
default,
})
})
2019-03-04 06:02:01 +01:00
}
}
pub struct ScopeService {
2021-01-09 19:06:49 +01:00
app_data: Option<Rc<Extensions>>,
2019-07-17 11:48:37 +02:00
router: Router<HttpService, Vec<Box<dyn Guard>>>,
2021-01-09 19:06:49 +01:00
default: HttpService,
2019-03-04 06:02:01 +01:00
}
impl Service<ServiceRequest> for ScopeService {
2019-03-04 06:02:01 +01:00
type Response = ServiceResponse;
type Error = Error;
type Future = LocalBoxFuture<'static, Result<Self::Response, Self::Error>>;
2019-03-04 06:02:01 +01:00
2021-01-09 19:06:49 +01:00
actix_service::always_ready!();
2019-03-04 06:02:01 +01:00
fn call(&self, mut req: ServiceRequest) -> Self::Future {
let res = self.router.recognize_checked(&mut req, |req, guards| {
2019-03-04 20:47:53 +01:00
if let Some(ref guards) = guards {
for f in guards {
if !f.check(req.head()) {
return false;
}
}
}
true
});
2021-01-09 19:06:49 +01:00
if let Some(ref app_data) = self.app_data {
req.add_data_container(app_data.clone());
}
2019-03-04 20:47:53 +01:00
if let Some((srv, _info)) = res {
srv.call(req)
2019-03-04 06:02:01 +01:00
} else {
2021-01-09 19:06:49 +01:00
self.default.call(req)
2019-03-04 06:02:01 +01:00
}
}
}
#[doc(hidden)]
pub struct ScopeEndpoint {
factory: Rc<RefCell<Option<ScopeFactory>>>,
2019-03-04 06:02:01 +01:00
}
impl ScopeEndpoint {
fn new(factory: Rc<RefCell<Option<ScopeFactory>>>) -> Self {
2019-03-04 06:02:01 +01:00
ScopeEndpoint { factory }
}
}
impl ServiceFactory<ServiceRequest> for ScopeEndpoint {
2019-03-04 06:02:01 +01:00
type Response = ServiceResponse;
type Error = Error;
type Config = ();
type Service = ScopeService;
type InitError = ();
2021-01-09 19:06:49 +01:00
type Future = LocalBoxFuture<'static, Result<Self::Service, Self::InitError>>;
2019-03-04 06:02:01 +01:00
2019-12-02 16:37:13 +01:00
fn new_service(&self, _: ()) -> Self::Future {
self.factory.borrow_mut().as_mut().unwrap().new_service(())
2019-03-04 06:02:01 +01:00
}
}
#[cfg(test)]
mod tests {
use actix_service::Service;
2019-03-04 06:02:01 +01:00
use bytes::Bytes;
2020-05-18 04:47:20 +02:00
use futures_util::future::ok;
2019-03-04 06:02:01 +01:00
2019-03-08 00:51:24 +01:00
use crate::dev::{Body, ResponseBody};
2019-03-25 21:02:10 +01:00
use crate::http::{header, HeaderValue, Method, StatusCode};
2019-11-20 18:33:22 +01:00
use crate::middleware::DefaultHeaders;
2019-11-26 06:25:50 +01:00
use crate::service::ServiceRequest;
use crate::test::{call_service, init_service, read_body, TestRequest};
use crate::{guard, web, App, HttpRequest, HttpResponse};
#[actix_rt::test]
async fn test_scope() {
let srv = init_service(App::new().service(
web::scope("/app").service(web::resource("/path1").to(HttpResponse::Ok)),
))
2019-11-26 06:25:50 +01:00
.await;
2019-03-04 06:02:01 +01:00
2019-11-26 06:25:50 +01:00
let req = TestRequest::with_uri("/app/path1").to_request();
let resp = srv.call(req).await.unwrap();
assert_eq!(resp.status(), StatusCode::OK);
2019-03-04 06:02:01 +01:00
}
2019-11-26 06:25:50 +01:00
#[actix_rt::test]
async fn test_scope_root() {
let srv = init_service(
2019-11-26 06:25:50 +01:00
App::new().service(
web::scope("/app")
.service(web::resource("").to(HttpResponse::Ok))
.service(web::resource("/").to(HttpResponse::Created)),
2019-11-26 06:25:50 +01:00
),
)
.await;
2019-03-04 06:02:01 +01:00
2019-11-26 06:25:50 +01:00
let req = TestRequest::with_uri("/app").to_request();
let resp = srv.call(req).await.unwrap();
assert_eq!(resp.status(), StatusCode::OK);
2019-03-04 06:02:01 +01:00
2019-11-26 06:25:50 +01:00
let req = TestRequest::with_uri("/app/").to_request();
let resp = srv.call(req).await.unwrap();
assert_eq!(resp.status(), StatusCode::CREATED);
2019-03-04 06:02:01 +01:00
}
2019-11-26 06:25:50 +01:00
#[actix_rt::test]
async fn test_scope_root2() {
let srv = init_service(App::new().service(
web::scope("/app/").service(web::resource("").to(HttpResponse::Ok)),
2019-11-26 06:25:50 +01:00
))
.await;
2019-11-20 18:33:22 +01:00
2019-11-26 06:25:50 +01:00
let req = TestRequest::with_uri("/app").to_request();
let resp = srv.call(req).await.unwrap();
assert_eq!(resp.status(), StatusCode::NOT_FOUND);
2019-11-20 18:33:22 +01:00
2019-11-26 06:25:50 +01:00
let req = TestRequest::with_uri("/app/").to_request();
let resp = srv.call(req).await.unwrap();
assert_eq!(resp.status(), StatusCode::OK);
2019-03-04 06:02:01 +01:00
}
2019-11-26 06:25:50 +01:00
#[actix_rt::test]
async fn test_scope_root3() {
let srv = init_service(App::new().service(
web::scope("/app/").service(web::resource("/").to(HttpResponse::Ok)),
2019-11-26 06:25:50 +01:00
))
.await;
2019-03-04 06:02:01 +01:00
2019-11-26 06:25:50 +01:00
let req = TestRequest::with_uri("/app").to_request();
let resp = srv.call(req).await.unwrap();
assert_eq!(resp.status(), StatusCode::NOT_FOUND);
2019-03-04 06:02:01 +01:00
2019-11-26 06:25:50 +01:00
let req = TestRequest::with_uri("/app/").to_request();
let resp = srv.call(req).await.unwrap();
assert_eq!(resp.status(), StatusCode::NOT_FOUND);
2019-03-04 06:02:01 +01:00
}
2019-11-26 06:25:50 +01:00
#[actix_rt::test]
async fn test_scope_route() {
let srv = init_service(
2019-11-26 06:25:50 +01:00
App::new().service(
web::scope("app")
.route("/path1", web::get().to(HttpResponse::Ok))
.route("/path1", web::delete().to(HttpResponse::Ok)),
2019-11-26 06:25:50 +01:00
),
)
.await;
let req = TestRequest::with_uri("/app/path1").to_request();
let resp = srv.call(req).await.unwrap();
assert_eq!(resp.status(), StatusCode::OK);
let req = TestRequest::with_uri("/app/path1")
.method(Method::DELETE)
.to_request();
let resp = srv.call(req).await.unwrap();
assert_eq!(resp.status(), StatusCode::OK);
let req = TestRequest::with_uri("/app/path1")
.method(Method::POST)
.to_request();
let resp = srv.call(req).await.unwrap();
assert_eq!(resp.status(), StatusCode::NOT_FOUND);
2019-03-04 06:02:01 +01:00
}
2019-11-26 06:25:50 +01:00
#[actix_rt::test]
async fn test_scope_route_without_leading_slash() {
let srv = init_service(
2019-11-26 06:25:50 +01:00
App::new().service(
web::scope("app").service(
web::resource("path1")
.route(web::get().to(HttpResponse::Ok))
.route(web::delete().to(HttpResponse::Ok)),
),
2019-11-26 06:25:50 +01:00
),
)
.await;
let req = TestRequest::with_uri("/app/path1").to_request();
let resp = srv.call(req).await.unwrap();
assert_eq!(resp.status(), StatusCode::OK);
let req = TestRequest::with_uri("/app/path1")
.method(Method::DELETE)
.to_request();
let resp = srv.call(req).await.unwrap();
assert_eq!(resp.status(), StatusCode::OK);
let req = TestRequest::with_uri("/app/path1")
.method(Method::POST)
.to_request();
let resp = srv.call(req).await.unwrap();
assert_eq!(resp.status(), StatusCode::METHOD_NOT_ALLOWED);
2019-03-04 06:02:01 +01:00
}
2019-11-26 06:25:50 +01:00
#[actix_rt::test]
async fn test_scope_guard() {
let srv = init_service(
2019-11-26 06:25:50 +01:00
App::new().service(
web::scope("/app")
.guard(guard::Get())
.service(web::resource("/path1").to(HttpResponse::Ok)),
2019-11-26 06:25:50 +01:00
),
)
.await;
let req = TestRequest::with_uri("/app/path1")
.method(Method::POST)
.to_request();
let resp = srv.call(req).await.unwrap();
assert_eq!(resp.status(), StatusCode::NOT_FOUND);
let req = TestRequest::with_uri("/app/path1")
.method(Method::GET)
.to_request();
let resp = srv.call(req).await.unwrap();
assert_eq!(resp.status(), StatusCode::OK);
2019-03-04 20:47:53 +01:00
}
2019-03-04 06:02:01 +01:00
2019-11-26 06:25:50 +01:00
#[actix_rt::test]
async fn test_scope_variable_segment() {
let srv = init_service(App::new().service(web::scope("/ab-{project}").service(
web::resource("/path1").to(|r: HttpRequest| {
HttpResponse::Ok()
.body(format!("project: {}", &r.match_info()["project"]))
}),
)))
.await;
2019-11-20 18:33:22 +01:00
2019-11-26 06:25:50 +01:00
let req = TestRequest::with_uri("/ab-project1/path1").to_request();
let resp = srv.call(req).await.unwrap();
assert_eq!(resp.status(), StatusCode::OK);
2019-11-20 18:33:22 +01:00
2019-11-26 06:25:50 +01:00
match resp.response().body() {
ResponseBody::Body(Body::Bytes(ref b)) => {
let bytes = b.clone();
2019-11-26 06:25:50 +01:00
assert_eq!(bytes, Bytes::from_static(b"project: project1"));
2019-03-04 06:02:01 +01:00
}
2019-11-26 06:25:50 +01:00
_ => panic!(),
}
2019-03-04 06:02:01 +01:00
2019-11-26 06:25:50 +01:00
let req = TestRequest::with_uri("/aa-project1/path1").to_request();
let resp = srv.call(req).await.unwrap();
assert_eq!(resp.status(), StatusCode::NOT_FOUND);
2019-03-04 06:02:01 +01:00
}
2019-11-26 06:25:50 +01:00
#[actix_rt::test]
async fn test_nested_scope() {
let srv = init_service(App::new().service(web::scope("/app").service(
web::scope("/t1").service(web::resource("/path1").to(HttpResponse::Created)),
)))
2019-11-26 06:25:50 +01:00
.await;
2019-03-04 06:02:01 +01:00
2019-11-26 06:25:50 +01:00
let req = TestRequest::with_uri("/app/t1/path1").to_request();
let resp = srv.call(req).await.unwrap();
assert_eq!(resp.status(), StatusCode::CREATED);
2019-03-04 06:02:01 +01:00
}
2019-11-26 06:25:50 +01:00
#[actix_rt::test]
async fn test_nested_scope_no_slash() {
let srv = init_service(App::new().service(web::scope("/app").service(
web::scope("t1").service(web::resource("/path1").to(HttpResponse::Created)),
)))
2019-11-26 06:25:50 +01:00
.await;
2019-03-04 06:02:01 +01:00
2019-11-26 06:25:50 +01:00
let req = TestRequest::with_uri("/app/t1/path1").to_request();
let resp = srv.call(req).await.unwrap();
assert_eq!(resp.status(), StatusCode::CREATED);
2019-03-04 06:02:01 +01:00
}
2019-11-26 06:25:50 +01:00
#[actix_rt::test]
async fn test_nested_scope_root() {
let srv = init_service(
2019-11-26 06:25:50 +01:00
App::new().service(
web::scope("/app").service(
web::scope("/t1")
.service(web::resource("").to(HttpResponse::Ok))
.service(web::resource("/").to(HttpResponse::Created)),
),
2019-11-26 06:25:50 +01:00
),
)
.await;
2019-03-04 06:02:01 +01:00
2019-11-26 06:25:50 +01:00
let req = TestRequest::with_uri("/app/t1").to_request();
let resp = srv.call(req).await.unwrap();
assert_eq!(resp.status(), StatusCode::OK);
2019-03-04 06:02:01 +01:00
2019-11-26 06:25:50 +01:00
let req = TestRequest::with_uri("/app/t1/").to_request();
let resp = srv.call(req).await.unwrap();
assert_eq!(resp.status(), StatusCode::CREATED);
2019-03-04 06:02:01 +01:00
}
2019-11-26 06:25:50 +01:00
#[actix_rt::test]
async fn test_nested_scope_filter() {
let srv = init_service(
2019-11-26 06:25:50 +01:00
App::new().service(
web::scope("/app").service(
web::scope("/t1")
.guard(guard::Get())
.service(web::resource("/path1").to(HttpResponse::Ok)),
),
2019-11-26 06:25:50 +01:00
),
)
.await;
let req = TestRequest::with_uri("/app/t1/path1")
.method(Method::POST)
.to_request();
let resp = srv.call(req).await.unwrap();
assert_eq!(resp.status(), StatusCode::NOT_FOUND);
let req = TestRequest::with_uri("/app/t1/path1")
.method(Method::GET)
.to_request();
let resp = srv.call(req).await.unwrap();
assert_eq!(resp.status(), StatusCode::OK);
2019-03-04 20:47:53 +01:00
}
2019-03-04 06:02:01 +01:00
2019-11-26 06:25:50 +01:00
#[actix_rt::test]
async fn test_nested_scope_with_variable_segment() {
let srv = init_service(App::new().service(web::scope("/app").service(
2019-11-26 06:25:50 +01:00
web::scope("/{project_id}").service(web::resource("/path1").to(
|r: HttpRequest| {
HttpResponse::Created()
.body(format!("project: {}", &r.match_info()["project_id"]))
2019-11-26 06:25:50 +01:00
},
)),
)))
.await;
let req = TestRequest::with_uri("/app/project_1/path1").to_request();
let resp = srv.call(req).await.unwrap();
assert_eq!(resp.status(), StatusCode::CREATED);
match resp.response().body() {
ResponseBody::Body(Body::Bytes(ref b)) => {
let bytes = b.clone();
2019-11-26 06:25:50 +01:00
assert_eq!(bytes, Bytes::from_static(b"project: project_1"));
2019-03-04 06:02:01 +01:00
}
2019-11-26 06:25:50 +01:00
_ => panic!(),
}
2019-03-04 06:02:01 +01:00
}
2019-11-26 06:25:50 +01:00
#[actix_rt::test]
async fn test_nested2_scope_with_variable_segment() {
let srv = init_service(App::new().service(web::scope("/app").service(
2019-11-26 06:25:50 +01:00
web::scope("/{project}").service(web::scope("/{id}").service(
web::resource("/path1").to(|r: HttpRequest| {
HttpResponse::Created().body(format!(
"project: {} - {}",
&r.match_info()["project"],
&r.match_info()["id"],
))
2019-11-26 06:25:50 +01:00
}),
)),
)))
.await;
let req = TestRequest::with_uri("/app/test/1/path1").to_request();
let resp = srv.call(req).await.unwrap();
assert_eq!(resp.status(), StatusCode::CREATED);
match resp.response().body() {
ResponseBody::Body(Body::Bytes(ref b)) => {
let bytes = b.clone();
2019-11-26 06:25:50 +01:00
assert_eq!(bytes, Bytes::from_static(b"project: test - 1"));
2019-03-04 06:02:01 +01:00
}
2019-11-26 06:25:50 +01:00
_ => panic!(),
}
2019-03-04 06:02:01 +01:00
2019-11-26 06:25:50 +01:00
let req = TestRequest::with_uri("/app/test/1/path2").to_request();
let resp = srv.call(req).await.unwrap();
assert_eq!(resp.status(), StatusCode::NOT_FOUND);
2019-03-04 06:02:01 +01:00
}
2019-11-26 06:25:50 +01:00
#[actix_rt::test]
async fn test_default_resource() {
let srv = init_service(
2019-11-26 06:25:50 +01:00
App::new().service(
web::scope("/app")
.service(web::resource("/path1").to(HttpResponse::Ok))
2019-11-26 06:25:50 +01:00
.default_service(|r: ServiceRequest| {
ok(r.into_response(HttpResponse::BadRequest()))
}),
),
)
.await;
2019-03-04 06:02:01 +01:00
2019-11-26 06:25:50 +01:00
let req = TestRequest::with_uri("/app/path2").to_request();
let resp = srv.call(req).await.unwrap();
assert_eq!(resp.status(), StatusCode::BAD_REQUEST);
2019-03-04 06:02:01 +01:00
2019-11-26 06:25:50 +01:00
let req = TestRequest::with_uri("/path2").to_request();
let resp = srv.call(req).await.unwrap();
assert_eq!(resp.status(), StatusCode::NOT_FOUND);
2019-03-04 06:02:01 +01:00
}
2019-11-26 06:25:50 +01:00
#[actix_rt::test]
async fn test_default_resource_propagation() {
let srv = init_service(
2019-11-26 06:25:50 +01:00
App::new()
.service(
web::scope("/app1")
.default_service(web::resource("").to(HttpResponse::BadRequest)),
)
2019-11-26 06:25:50 +01:00
.service(web::scope("/app2"))
.default_service(|r: ServiceRequest| {
ok(r.into_response(HttpResponse::MethodNotAllowed()))
}),
)
.await;
2019-11-20 18:33:22 +01:00
2019-11-26 06:25:50 +01:00
let req = TestRequest::with_uri("/non-exist").to_request();
let resp = srv.call(req).await.unwrap();
assert_eq!(resp.status(), StatusCode::METHOD_NOT_ALLOWED);
2019-11-20 18:33:22 +01:00
2019-11-26 06:25:50 +01:00
let req = TestRequest::with_uri("/app1/non-exist").to_request();
let resp = srv.call(req).await.unwrap();
assert_eq!(resp.status(), StatusCode::BAD_REQUEST);
2019-03-25 21:02:10 +01:00
2019-11-26 06:25:50 +01:00
let req = TestRequest::with_uri("/app2/non-exist").to_request();
let resp = srv.call(req).await.unwrap();
assert_eq!(resp.status(), StatusCode::METHOD_NOT_ALLOWED);
2019-03-25 21:02:10 +01:00
}
2019-11-26 06:25:50 +01:00
#[actix_rt::test]
async fn test_middleware() {
let srv = init_service(
App::new().service(
web::scope("app")
.wrap(
DefaultHeaders::new().header(
2019-11-20 18:33:22 +01:00
header::CONTENT_TYPE,
HeaderValue::from_static("0001"),
),
)
.service(
web::resource("/test").route(web::get().to(HttpResponse::Ok)),
),
),
)
.await;
2019-11-20 18:33:22 +01:00
2019-11-26 06:25:50 +01:00
let req = TestRequest::with_uri("/app/test").to_request();
let resp = call_service(&srv, req).await;
2019-11-26 06:25:50 +01:00
assert_eq!(resp.status(), StatusCode::OK);
assert_eq!(
resp.headers().get(header::CONTENT_TYPE).unwrap(),
HeaderValue::from_static("0001")
);
2019-03-25 21:02:10 +01:00
}
2019-03-25 21:43:02 +01:00
2019-11-26 06:25:50 +01:00
#[actix_rt::test]
async fn test_middleware_fn() {
let srv = init_service(
2019-11-26 06:25:50 +01:00
App::new().service(
web::scope("app")
.wrap_fn(|req, srv| {
let fut = srv.call(req);
async move {
let mut res = fut.await?;
res.headers_mut().insert(
header::CONTENT_TYPE,
HeaderValue::from_static("0001"),
);
Ok(res)
}
})
.route("/test", web::get().to(HttpResponse::Ok)),
2019-11-26 06:25:50 +01:00
),
)
.await;
let req = TestRequest::with_uri("/app/test").to_request();
let resp = call_service(&srv, req).await;
2019-11-26 06:25:50 +01:00
assert_eq!(resp.status(), StatusCode::OK);
assert_eq!(
resp.headers().get(header::CONTENT_TYPE).unwrap(),
HeaderValue::from_static("0001")
);
2019-03-25 21:43:02 +01:00
}
2019-11-26 06:25:50 +01:00
#[actix_rt::test]
async fn test_override_data() {
let srv = init_service(App::new().data(1usize).service(
2019-11-26 06:25:50 +01:00
web::scope("app").data(10usize).route(
"/t",
web::get().to(|data: web::Data<usize>| {
2019-12-20 12:45:35 +01:00
assert_eq!(**data, 10);
2019-11-26 06:25:50 +01:00
HttpResponse::Ok()
}),
),
))
.await;
2019-11-26 06:25:50 +01:00
let req = TestRequest::with_uri("/app/t").to_request();
let resp = call_service(&srv, req).await;
2019-11-26 06:25:50 +01:00
assert_eq!(resp.status(), StatusCode::OK);
}
#[actix_rt::test]
async fn test_override_data_default_service() {
let srv = init_service(App::new().data(1usize).service(
web::scope("app").data(10usize).default_service(web::to(
|data: web::Data<usize>| {
assert_eq!(**data, 10);
HttpResponse::Ok()
},
)),
))
.await;
let req = TestRequest::with_uri("/app/t").to_request();
let resp = call_service(&srv, req).await;
assert_eq!(resp.status(), StatusCode::OK);
}
2019-11-26 06:25:50 +01:00
#[actix_rt::test]
async fn test_override_app_data() {
let srv = init_service(App::new().app_data(web::Data::new(1usize)).service(
web::scope("app").app_data(web::Data::new(10usize)).route(
"/t",
web::get().to(|data: web::Data<usize>| {
2019-12-20 12:45:35 +01:00
assert_eq!(**data, 10);
HttpResponse::Ok()
}),
2019-11-26 06:25:50 +01:00
),
))
2019-11-26 06:25:50 +01:00
.await;
2019-11-26 06:25:50 +01:00
let req = TestRequest::with_uri("/app/t").to_request();
let resp = call_service(&srv, req).await;
2019-11-26 06:25:50 +01:00
assert_eq!(resp.status(), StatusCode::OK);
}
2019-11-26 06:25:50 +01:00
#[actix_rt::test]
async fn test_scope_config() {
let srv = init_service(App::new().service(web::scope("/app").configure(|s| {
s.route("/path1", web::get().to(HttpResponse::Ok));
})))
.await;
2019-11-26 06:25:50 +01:00
let req = TestRequest::with_uri("/app/path1").to_request();
let resp = srv.call(req).await.unwrap();
assert_eq!(resp.status(), StatusCode::OK);
}
2019-11-26 06:25:50 +01:00
#[actix_rt::test]
async fn test_scope_config_2() {
let srv = init_service(App::new().service(web::scope("/app").configure(|s| {
s.service(web::scope("/v1").configure(|s| {
s.route("/", web::get().to(HttpResponse::Ok));
}));
})))
.await;
2019-11-26 06:25:50 +01:00
let req = TestRequest::with_uri("/app/v1/").to_request();
let resp = srv.call(req).await.unwrap();
assert_eq!(resp.status(), StatusCode::OK);
}
2019-11-26 06:25:50 +01:00
#[actix_rt::test]
async fn test_url_for_external() {
let srv = init_service(App::new().service(web::scope("/app").configure(|s| {
s.service(web::scope("/v1").configure(|s| {
s.external_resource("youtube", "https://youtube.com/watch/{video_id}");
s.route(
"/",
web::get().to(|req: HttpRequest| {
HttpResponse::Ok().body(
req.url_for("youtube", &["xxxxxx"]).unwrap().to_string(),
)
}),
);
}));
})))
.await;
2019-11-20 18:33:22 +01:00
2019-11-26 06:25:50 +01:00
let req = TestRequest::with_uri("/app/v1/").to_request();
let resp = srv.call(req).await.unwrap();
assert_eq!(resp.status(), StatusCode::OK);
let body = read_body(resp).await;
assert_eq!(body, &b"https://youtube.com/watch/xxxxxx"[..]);
}
#[actix_rt::test]
async fn test_url_for_nested() {
let srv = init_service(App::new().service(web::scope("/a").service(
2019-11-26 06:25:50 +01:00
web::scope("/b").service(web::resource("/c/{stuff}").name("c").route(
web::get().to(|req: HttpRequest| {
HttpResponse::Ok()
.body(format!("{}", req.url_for("c", &["12345"]).unwrap()))
2019-11-26 06:25:50 +01:00
}),
)),
)))
.await;
let req = TestRequest::with_uri("/a/b/c/test").to_request();
let resp = call_service(&srv, req).await;
2019-11-26 06:25:50 +01:00
assert_eq!(resp.status(), StatusCode::OK);
let body = read_body(resp).await;
assert_eq!(
body,
Bytes::from_static(b"http://localhost:8080/a/b/c/12345")
);
}
2019-03-04 06:02:01 +01:00
}