berkus
/
actix-web
mirror of https://github.com/fafhrd91/actix-web
1
0
Fork 0
Code Issues Releases Wiki Activity

fix body change test

This commit is contained in:
Rob Ede 2021-12-14 18:53:59 +00:00
parent d877367912
commit 27af320f27
No known key found for this signature in database
GPG Key ID: 97C636207D3EF933
1 changed files with 14 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
src/middleware/err_handlers.rs
View File

@ -189,12 +189,13 @@ mod tests {
use futures_util::future::FutureExt as _; use futures_util::future::FutureExt as _;
use super::*; use super::*;
use crate::http::{ use crate::{
header::{HeaderValue, CONTENT_TYPE}, http::{
StatusCode, header::{HeaderValue, CONTENT_TYPE},
StatusCode,
},
test::{self, TestRequest},
}; };
use crate::test::{self, TestRequest};
use crate::HttpResponse;
#[actix_rt::test] #[actix_rt::test]
async fn add_header_error_handler() { async fn add_header_error_handler() {
@ -207,9 +208,7 @@ mod tests {
Ok(ErrorHandlerResponse::Response(res.map_into_left_body())) Ok(ErrorHandlerResponse::Response(res.map_into_left_body()))
} }
let srv = |req: ServiceRequest| { let srv = test::default_service(StatusCode::INTERNAL_SERVER_ERROR);
ok(req.into_response(HttpResponse::InternalServerError().finish()))
};
let mw = ErrorHandlers::new() let mw = ErrorHandlers::new()
.handler(StatusCode::INTERNAL_SERVER_ERROR, error_handler) .handler(StatusCode::INTERNAL_SERVER_ERROR, error_handler)
@ -236,9 +235,7 @@ mod tests {
)) ))
} }
let srv = |req: ServiceRequest| { let srv = test::default_service(StatusCode::INTERNAL_SERVER_ERROR);
ok(req.into_response(HttpResponse::InternalServerError().finish()))
};
let mw = ErrorHandlers::new() let mw = ErrorHandlers::new()
.handler(StatusCode::INTERNAL_SERVER_ERROR, error_handler) .handler(StatusCode::INTERNAL_SERVER_ERROR, error_handler)
@ -262,12 +259,11 @@ mod tests {
let res = ServiceResponse::new(req, res) let res = ServiceResponse::new(req, res)
.map_into_boxed_body() .map_into_boxed_body()
.map_into_right_body(); .map_into_right_body();
Ok(ErrorHandlerResponse::Response(res)) Ok(ErrorHandlerResponse::Response(res))
} }
let srv = |req: ServiceRequest| { let srv = test::default_service(StatusCode::INTERNAL_SERVER_ERROR);
ok(req.into_response(HttpResponse::InternalServerError().finish()))
};
let mw = ErrorHandlers::new() let mw = ErrorHandlers::new()
.handler(StatusCode::INTERNAL_SERVER_ERROR, error_handler) .handler(StatusCode::INTERNAL_SERVER_ERROR, error_handler)
@ -275,7 +271,9 @@ mod tests {
.await .await
.unwrap(); .unwrap();
let resp = test::call_service(&mw, TestRequest::default().to_srv_request()).await; let res = test::call_service(&mw, TestRequest::default().to_srv_request()).await;
assert_eq!(resp.headers().get(CONTENT_TYPE).unwrap(), "0001"); assert_eq!(test::read_body(res).await, "sorry, that's no bueno");
} }
// TODO: test where error is thrown
} }