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

Complete Unit Testing

This commit is contained in:
yinho999 2023-04-19 22:08:00 +01:00
parent 7038c4fe45
commit ba58221310
1 changed files with 3 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
actix-http/src/body/channel.rs
View File

@ -16,7 +16,7 @@ use super::{BodySize, MessageBody};
/// ``` /// ```
/// use actix_web::{HttpResponse, web}; /// use actix_web::{HttpResponse, web};
/// use std::convert::Infallible; /// use std::convert::Infallible;
/// use actix_web::body::channel; /// use actix_http::body::channel;
/// ///
/// #[actix_rt::main] /// #[actix_rt::main]
/// async fn main() { /// async fn main() {
@ -186,14 +186,12 @@ mod tests {
#[actix_rt::test] #[actix_rt::test]
async fn test_body_channel_error() { async fn test_body_channel_error() {
let (mut tx, rx) = channel::<io::Error>(); let (mut tx, rx) = channel::<io::Error>();
let mut tx_cloned = tx.clone();
let rx = rx.boxed(); let rx = rx.boxed();
pin!(rx); pin!(rx);
assert_eq!(rx.size(), BodySize::Stream); assert_eq!(rx.size(), BodySize::Stream);
tx.send(Bytes::from_static(b"test")).unwrap(); tx.send(Bytes::from_static(b"test")).unwrap();
tx_cloned.send(Bytes::from_static(b"test2")).unwrap();
let err = io::Error::new(io::ErrorKind::Other, "test"); let err = io::Error::new(io::ErrorKind::Other, "test");
@ -214,20 +212,15 @@ mod tests {
let (tx, rx) = channel::<io::Error>(); let (tx, rx) = channel::<io::Error>();
let rx = rx.boxed(); let rx = rx.boxed();
pin!(rx); pin!(rx);
drop(tx); drop(tx);
let received = poll_fn(|cx| rx.as_mut().poll_next(cx)).await;
let err = poll_fn(|cx| rx.as_mut().poll_next(cx)).await.unwrap().err(); assert!(received.is_none());
assert!(err.is_some());
assert_eq!(err.unwrap().to_string(), "channel closed");
} }
#[actix_rt::test] #[actix_rt::test]
async fn test_dropped_receiver() { async fn test_dropped_receiver() {
let (mut tx, rx) = channel::<io::Error>(); let (mut tx, rx) = channel::<io::Error>();
let rx = rx.boxed(); let rx = rx.boxed();
pin!(rx);
drop(rx); drop(rx);
let err = tx.send(Bytes::from_static(b"test")).unwrap_err(); let err = tx.send(Bytes::from_static(b"test")).unwrap_err();