update return-from-fn tests

This commit is contained in:
Rob Ede 2021-12-22 08:12:08 +00:00
parent b71ca411ab
commit 8bdf04016c
No known key found for this signature in database
GPG Key ID: 97C636207D3EF933
3 changed files with 19 additions and 10 deletions

View File

@ -38,6 +38,15 @@ pub struct Compat<T> {
transform: T, transform: T,
} }
#[cfg(test)]
impl Compat<super::Noop> {
pub(crate) fn noop() -> Self {
Self {
transform: super::Noop,
}
}
}
impl<T> Compat<T> { impl<T> Compat<T> {
/// Wrap a middleware to give it broader compatibility. /// Wrap a middleware to give it broader compatibility.
pub fn new(middleware: T) -> Self { pub fn new(middleware: T) -> Self {

View File

@ -546,11 +546,11 @@ mod tests {
>, >,
> { > {
web::resource("/test-compat") web::resource("/test-compat")
// .wrap_fn(|req, srv| { .wrap_fn(|req, srv| {
// let fut = srv.call(req); let fut = srv.call(req);
// async { Ok(fut.await?.map_into_right_body::<()>()) } async { Ok(fut.await?.map_into_right_body::<()>()) }
// }) })
.wrap(Compat::new(DefaultHeaders::new())) .wrap(Compat::noop())
.route(web::get().to(|| async { "hello" })) .route(web::get().to(|| async { "hello" }))
} }

View File

@ -616,11 +616,11 @@ mod tests {
>, >,
> { > {
web::scope("/test-compat") web::scope("/test-compat")
// .wrap_fn(|req, srv| { .wrap_fn(|req, srv| {
// let fut = srv.call(req); let fut = srv.call(req);
// async { Ok(fut.await?.map_into_right_body::<()>()) } async { Ok(fut.await?.map_into_right_body::<()>()) }
// }) })
.wrap(Compat::new(DefaultHeaders::new())) .wrap(Compat::noop())
.service(web::resource("").route(web::get().to(|| async { "hello" }))) .service(web::resource("").route(web::get().to(|| async { "hello" })))
} }