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

wip

Browse source
This commit is contained in:
Nikolay Kim 2024-11-28 10:58:17 -08:00
parent 3d296530f5
commit d33eda04ab
2 changed files with 21 additions and 38 deletions
Download patch file Download diff file Expand all files Collapse all files

4
ntex-http/Cargo.toml
View file

@ -27,3 +27,7 @@ futures-core = { version = "0.3", default-features = false, features = ["alloc"]
[dev-dependencies] [dev-dependencies]
bincode = "1" bincode = "1"
serde_json = "1" serde_json = "1"
ntex = "2"
ntex-util = "2"
ntex-macros = "0.1.3"
futures-util = { version = "0.3", default-features = false, features = ["alloc"] }

55
ntex-http/src/body.rs
View file

@ -558,11 +558,12 @@ where
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use futures_util::stream;
use std::{future::poll_fn, io}; use std::{future::poll_fn, io};
use futures_util::stream;
use ntex_util::future::Ready;
use super::*; use super::*;
use crate::util::Ready;
impl Body { impl Body {
pub(crate) fn get_ref(&self) -> &[u8] { pub(crate) fn get_ref(&self) -> &[u8] {
@ -573,16 +574,7 @@ mod tests {
} }
} }
impl ResponseBody<Body> { #[ntex::test]
pub(crate) fn get_ref(&self) -> &[u8] {
match *self {
ResponseBody::Body(ref b) => b.get_ref(),
ResponseBody::Other(ref b) => b.get_ref(),
}
}
}
#[crate::rt_test]
async fn test_static_str() { async fn test_static_str() {
assert_eq!(Body::from("").size(), BodySize::Sized(0)); assert_eq!(Body::from("").size(), BodySize::Sized(0));
assert_eq!(Body::from("test").size(), BodySize::Sized(4)); assert_eq!(Body::from("test").size(), BodySize::Sized(4));
@ -600,7 +592,7 @@ mod tests {
assert!(poll_fn(|cx| "".poll_next_chunk(cx)).await.is_none()); assert!(poll_fn(|cx| "".poll_next_chunk(cx)).await.is_none());
} }
#[crate::rt_test] #[ntex::test]
async fn test_static_bytes() { async fn test_static_bytes() {
assert_eq!(Body::from(b"test".as_ref()).size(), BodySize::Sized(4)); assert_eq!(Body::from(b"test".as_ref()).size(), BodySize::Sized(4));
assert_eq!(Body::from(b"test".as_ref()).get_ref(), b"test"); assert_eq!(Body::from(b"test".as_ref()).get_ref(), b"test");
@ -622,7 +614,7 @@ mod tests {
assert!(poll_fn(|cx| (&b""[..]).poll_next_chunk(cx)).await.is_none()); assert!(poll_fn(|cx| (&b""[..]).poll_next_chunk(cx)).await.is_none());
} }
#[crate::rt_test] #[ntex::test]
async fn test_vec() { async fn test_vec() {
assert_eq!(Body::from(Vec::from("test")).size(), BodySize::Sized(4)); assert_eq!(Body::from(Vec::from("test")).size(), BodySize::Sized(4));
assert_eq!(Body::from(Vec::from("test")).get_ref(), b"test"); assert_eq!(Body::from(Vec::from("test")).get_ref(), b"test");
@ -647,7 +639,7 @@ mod tests {
.is_none()); .is_none());
} }
#[crate::rt_test] #[ntex::test]
async fn test_bytes() { async fn test_bytes() {
let mut b = Bytes::from("test"); let mut b = Bytes::from("test");
assert_eq!(Body::from(b.clone()).size(), BodySize::Sized(4)); assert_eq!(Body::from(b.clone()).size(), BodySize::Sized(4));
@ -661,7 +653,7 @@ mod tests {
assert!(poll_fn(|cx| b.poll_next_chunk(cx)).await.is_none(),); assert!(poll_fn(|cx| b.poll_next_chunk(cx)).await.is_none(),);
} }
#[crate::rt_test] #[ntex::test]
async fn test_bytes_mut() { async fn test_bytes_mut() {
let mut b = Body::from(BytesMut::from("test")); let mut b = Body::from(BytesMut::from("test"));
assert_eq!(b.size(), BodySize::Sized(4)); assert_eq!(b.size(), BodySize::Sized(4));
@ -682,7 +674,7 @@ mod tests {
assert!(poll_fn(|cx| b.poll_next_chunk(cx)).await.is_none(),); assert!(poll_fn(|cx| b.poll_next_chunk(cx)).await.is_none(),);
} }
#[crate::rt_test] #[ntex::test]
async fn test_string() { async fn test_string() {
let mut b = "test".to_owned(); let mut b = "test".to_owned();
assert_eq!(Body::from(b.clone()).size(), BodySize::Sized(4)); assert_eq!(Body::from(b.clone()).size(), BodySize::Sized(4));
@ -698,20 +690,20 @@ mod tests {
assert!(poll_fn(|cx| b.poll_next_chunk(cx)).await.is_none(),); assert!(poll_fn(|cx| b.poll_next_chunk(cx)).await.is_none(),);
} }
#[crate::rt_test] #[ntex::test]
async fn test_unit() { async fn test_unit() {
assert_eq!(().size(), BodySize::Empty); assert_eq!(().size(), BodySize::Empty);
assert!(poll_fn(|cx| ().poll_next_chunk(cx)).await.is_none()); assert!(poll_fn(|cx| ().poll_next_chunk(cx)).await.is_none());
} }
#[crate::rt_test] #[ntex::test]
async fn test_box() { async fn test_box() {
let mut val = Box::new(()); let mut val = Box::new(());
assert_eq!(val.size(), BodySize::Empty); assert_eq!(val.size(), BodySize::Empty);
assert!(poll_fn(|cx| val.poll_next_chunk(cx)).await.is_none()); assert!(poll_fn(|cx| val.poll_next_chunk(cx)).await.is_none());
} }
#[crate::rt_test] #[ntex::test]
#[allow(clippy::eq_op)] #[allow(clippy::eq_op)]
async fn test_body_eq() { async fn test_body_eq() {
assert!(Body::None == Body::None); assert!(Body::None == Body::None);
@ -724,27 +716,14 @@ mod tests {
assert!(Body::Bytes(Bytes::from_static(b"1")) != Body::None); assert!(Body::Bytes(Bytes::from_static(b"1")) != Body::None);
} }
#[crate::rt_test] #[ntex::test]
async fn test_body_debug() { async fn test_body_debug() {
assert!(format!("{:?}", Body::None).contains("Body::None")); assert!(format!("{:?}", Body::None).contains("Body::None"));
assert!(format!("{:?}", Body::Empty).contains("Body::Empty")); assert!(format!("{:?}", Body::Empty).contains("Body::Empty"));
assert!(format!("{:?}", Body::Bytes(Bytes::from_static(b"1"))).contains('1')); assert!(format!("{:?}", Body::Bytes(Bytes::from_static(b"1"))).contains('1'));
} }
#[crate::rt_test] #[ntex::test]
async fn test_serde_json() {
use serde_json::json;
assert_eq!(
Body::from(serde_json::Value::String("test".into())).size(),
BodySize::Sized(6)
);
assert_eq!(
Body::from(json!({"test-key":"test-value"})).size(),
BodySize::Sized(25)
);
}
#[crate::rt_test]
async fn body_stream() { async fn body_stream() {
let st = BodyStream::new(stream::once(Ready::<_, io::Error>::Ok(Bytes::from("1")))); let st = BodyStream::new(stream::once(Ready::<_, io::Error>::Ok(Bytes::from("1"))));
assert!(format!("{:?}", st).contains("BodyStream")); assert!(format!("{:?}", st).contains("BodyStream"));
@ -756,7 +735,7 @@ mod tests {
assert!(res.as_ref().is_some()); assert!(res.as_ref().is_some());
} }
#[crate::rt_test] #[ntex::test]
async fn boxed_body_stream() { async fn boxed_body_stream() {
let st = BoxedBodyStream::new(stream::once(Ready::<_, Box<dyn Error>>::Ok( let st = BoxedBodyStream::new(stream::once(Ready::<_, Box<dyn Error>>::Ok(
Bytes::from("1"), Bytes::from("1"),
@ -770,7 +749,7 @@ mod tests {
assert!(res.as_ref().is_some()); assert!(res.as_ref().is_some());
} }
#[crate::rt_test] #[ntex::test]
async fn body_skips_empty_chunks() { async fn body_skips_empty_chunks() {
let mut body = BodyStream::new(stream::iter( let mut body = BodyStream::new(stream::iter(
["1", "", "2"] ["1", "", "2"]
@ -787,7 +766,7 @@ mod tests {
); );
} }
#[crate::rt_test] #[ntex::test]
async fn sized_skips_empty_chunks() { async fn sized_skips_empty_chunks() {
let mut body = SizedStream::new( let mut body = SizedStream::new(
2, 2,