mirrors/ntex
1
0
Fork 0
mirror of https://github.com/ntex-rs/ntex.git synced 2025-04-03 21:07:39 +03:00
Code Issues Projects Releases Packages Wiki Activity

wip

Browse source
This commit is contained in:
Nikolay Kim 2024-11-28 14:36:42 -08:00
parent d33eda04ab
commit 1101dcb9be
4 changed files with 36 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

39
ntex/src/http/response.rs
View file

@ -227,6 +227,20 @@ impl<B> Response<B> {
}
}
#[cfg(test)]
impl Response<Body> {
pub(crate) fn get_body_ref(&self) -> &[u8] {
let b = match *self.body() {
ResponseBody::Body(ref b) => b,
ResponseBody::Other(ref b) => b,
};
match b {
Body::Bytes(bin) => bin,
_ => panic!(),
}
}
}
impl<B: MessageBody> fmt::Debug for Response<B> {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
let res = writeln!(
@ -925,7 +939,7 @@ mod tests {
let resp = Response::build(StatusCode::OK).json(&vec!["v1", "v2", "v3"]);
let ct = resp.headers().get(CONTENT_TYPE).unwrap();
assert_eq!(ct, HeaderValue::from_static("application/json"));
assert_eq!(resp.body().get_ref(), b"[\"v1\",\"v2\",\"v3\"]");
assert_eq!(resp.get_body_ref(), b"[\"v1\",\"v2\",\"v3\"]");
}
#[test]
@ -935,14 +949,7 @@ mod tests {
.json(&vec!["v1", "v2", "v3"]);
let ct = resp.headers().get(CONTENT_TYPE).unwrap();
assert_eq!(ct, HeaderValue::from_static("text/json"));
assert_eq!(resp.body().get_ref(), b"[\"v1\",\"v2\",\"v3\"]");
}
#[test]
fn test_serde_json_in_body() {
use serde_json::json;
let resp = Response::build(StatusCode::OK).body(json!({"test-key":"test-value"}));
assert_eq!(resp.body().get_ref(), br#"{"test-key":"test-value"}"#);
assert_eq!(resp.get_body_ref(), b"[\"v1\",\"v2\",\"v3\"]");
}
#[test]
@ -955,7 +962,7 @@ mod tests {
HeaderValue::from_static("text/plain; charset=utf-8")
);
assert_eq!(resp.status(), StatusCode::OK);
assert_eq!(resp.body().get_ref(), b"test");
assert_eq!(resp.get_body_ref(), b"test");
let resp: Response = b"test".as_ref().into();
assert_eq!(resp.status(), StatusCode::OK);
@ -964,7 +971,7 @@ mod tests {
HeaderValue::from_static("application/octet-stream")
);
assert_eq!(resp.status(), StatusCode::OK);
assert_eq!(resp.body().get_ref(), b"test");
assert_eq!(resp.get_body_ref(), b"test");
let resp: Response = "test".to_owned().into();
assert_eq!(resp.status(), StatusCode::OK);
@ -973,7 +980,7 @@ mod tests {
HeaderValue::from_static("text/plain; charset=utf-8")
);
assert_eq!(resp.status(), StatusCode::OK);
assert_eq!(resp.body().get_ref(), b"test");
assert_eq!(resp.get_body_ref(), b"test");
let resp: Response = (&"test".to_owned()).into();
assert_eq!(resp.status(), StatusCode::OK);
@ -982,7 +989,7 @@ mod tests {
HeaderValue::from_static("text/plain; charset=utf-8")
);
assert_eq!(resp.status(), StatusCode::OK);
assert_eq!(resp.body().get_ref(), b"test");
assert_eq!(resp.get_body_ref(), b"test");
let b = Bytes::from_static(b"test");
let resp: Response = b.into();
@ -992,7 +999,7 @@ mod tests {
HeaderValue::from_static("application/octet-stream")
);
assert_eq!(resp.status(), StatusCode::OK);
assert_eq!(resp.body().get_ref(), b"test");
assert_eq!(resp.get_body_ref(), b"test");
let b = Bytes::from_static(b"test");
let resp: Response = b.into();
@ -1002,7 +1009,7 @@ mod tests {
HeaderValue::from_static("application/octet-stream")
);
assert_eq!(resp.status(), StatusCode::OK);
assert_eq!(resp.body().get_ref(), b"test");
assert_eq!(resp.get_body_ref(), b"test");
let b = BytesMut::from("test");
let resp: Response = b.into();
@ -1013,7 +1020,7 @@ mod tests {
);
assert_eq!(resp.status(), StatusCode::OK);
assert_eq!(resp.body().get_ref(), b"test");
assert_eq!(resp.get_body_ref(), b"test");
let builder = Response::build_from(ResponseBuilder::new(StatusCode::OK))
.keep_alive()

22
ntex/src/web/responder.rs
View file

@ -371,7 +371,7 @@ pub(crate) mod tests {
let resp: HttpResponse = responder("test").respond_to(&req).await;
assert_eq!(resp.status(), StatusCode::OK);
assert_eq!(resp.body().get_ref(), b"test");
assert_eq!(resp.get_body_ref(), b"test");
assert_eq!(
resp.headers().get(CONTENT_TYPE).unwrap(),
HeaderValue::from_static("text/plain; charset=utf-8")
@ -379,7 +379,7 @@ pub(crate) mod tests {
let resp: HttpResponse = responder(&b"test"[..]).respond_to(&req).await;
assert_eq!(resp.status(), StatusCode::OK);
assert_eq!(resp.body().get_ref(), b"test");
assert_eq!(resp.get_body_ref(), b"test");
assert_eq!(
resp.headers().get(CONTENT_TYPE).unwrap(),
HeaderValue::from_static("application/octet-stream")
@ -387,7 +387,7 @@ pub(crate) mod tests {
let resp: HttpResponse = responder("test".to_string()).respond_to(&req).await;
assert_eq!(resp.status(), StatusCode::OK);
assert_eq!(resp.body().get_ref(), b"test");
assert_eq!(resp.get_body_ref(), b"test");
assert_eq!(
resp.headers().get(CONTENT_TYPE).unwrap(),
HeaderValue::from_static("text/plain; charset=utf-8")
@ -395,7 +395,7 @@ pub(crate) mod tests {
let resp: HttpResponse = responder(&"test".to_string()).respond_to(&req).await;
assert_eq!(resp.status(), StatusCode::OK);
assert_eq!(resp.body().get_ref(), b"test");
assert_eq!(resp.get_body_ref(), b"test");
assert_eq!(
resp.headers().get(CONTENT_TYPE).unwrap(),
HeaderValue::from_static("text/plain; charset=utf-8")
@ -405,7 +405,7 @@ pub(crate) mod tests {
.respond_to(&req)
.await;
assert_eq!(resp.status(), StatusCode::OK);
assert_eq!(resp.body().get_ref(), b"test");
assert_eq!(resp.get_body_ref(), b"test");
assert_eq!(
resp.headers().get(CONTENT_TYPE).unwrap(),
HeaderValue::from_static("application/octet-stream")
@ -415,7 +415,7 @@ pub(crate) mod tests {
.respond_to(&req)
.await;
assert_eq!(resp.status(), StatusCode::OK);
assert_eq!(resp.body().get_ref(), b"test");
assert_eq!(resp.get_body_ref(), b"test");
assert_eq!(
resp.headers().get(CONTENT_TYPE).unwrap(),
HeaderValue::from_static("application/octet-stream")
@ -440,7 +440,7 @@ pub(crate) mod tests {
)
.await;
assert_eq!(resp.status(), StatusCode::OK);
assert_eq!(resp.body().get_ref(), b"test");
assert_eq!(resp.get_body_ref(), b"test");
assert_eq!(
resp.headers().get(CONTENT_TYPE).unwrap(),
HeaderValue::from_static("text/plain; charset=utf-8")
@ -463,7 +463,7 @@ pub(crate) mod tests {
.respond_to(&req)
.await;
assert_eq!(res.status(), StatusCode::BAD_REQUEST);
assert_eq!(res.body().get_ref(), b"test");
assert_eq!(res.get_body_ref(), b"test");
let res = responder("test".to_string())
.with_header("content-type", "json")
@ -471,7 +471,7 @@ pub(crate) mod tests {
.await;
assert_eq!(res.status(), StatusCode::OK);
assert_eq!(res.body().get_ref(), b"test");
assert_eq!(res.get_body_ref(), b"test");
assert_eq!(
res.headers().get(CONTENT_TYPE).unwrap(),
HeaderValue::from_static("json")
@ -487,7 +487,7 @@ pub(crate) mod tests {
)
.await;
assert_eq!(res.status(), StatusCode::BAD_REQUEST);
assert_eq!(res.body().get_ref(), b"test");
assert_eq!(res.get_body_ref(), b"test");
let req = TestRequest::default().to_http_request();
let res =
@ -496,7 +496,7 @@ pub(crate) mod tests {
.respond_to(&req)
.await;
assert_eq!(res.status(), StatusCode::OK);
assert_eq!(res.body().get_ref(), b"test");
assert_eq!(res.get_body_ref(), b"test");
assert_eq!(
res.headers().get(CONTENT_TYPE).unwrap(),
HeaderValue::from_static("json")

2
ntex/src/web/types/form.rs
View file

@ -493,6 +493,6 @@ mod tests {
HeaderValue::from_static("application/x-www-form-urlencoded")
);
assert_eq!(resp.body().get_ref(), b"hello=world&counter=123");
assert_eq!(resp.get_body_ref(), b"hello=world&counter=123");
}
}

2
ntex/src/web/types/json.rs
View file

@ -434,7 +434,7 @@ mod tests {
header::HeaderValue::from_static("application/json")
);
assert_eq!(resp.body().get_ref(), b"{\"name\":\"test\"}");
assert_eq!(resp.get_body_ref(), b"{\"name\":\"test\"}");
}
#[crate::rt_test]