From 2aa4a82499d4becd2284cdb482213d541b8804dd Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Sun, 28 Apr 2024 16:29:10 +0200 Subject: Adding upstream version 86.0.1. Signed-off-by: Daniel Baumann --- third_party/rust/tokio/tests/stream_reader.rs | 35 +++++++++++++++++++++++++++ 1 file changed, 35 insertions(+) create mode 100644 third_party/rust/tokio/tests/stream_reader.rs (limited to 'third_party/rust/tokio/tests/stream_reader.rs') diff --git a/third_party/rust/tokio/tests/stream_reader.rs b/third_party/rust/tokio/tests/stream_reader.rs new file mode 100644 index 0000000000..8370df4dac --- /dev/null +++ b/third_party/rust/tokio/tests/stream_reader.rs @@ -0,0 +1,35 @@ +#![warn(rust_2018_idioms)] +#![cfg(feature = "full")] + +use bytes::Bytes; +use tokio::io::{stream_reader, AsyncReadExt}; +use tokio::stream::iter; + +#[tokio::test] +async fn test_stream_reader() -> std::io::Result<()> { + let stream = iter(vec![ + Ok(Bytes::from_static(&[])), + Ok(Bytes::from_static(&[0, 1, 2, 3])), + Ok(Bytes::from_static(&[])), + Ok(Bytes::from_static(&[4, 5, 6, 7])), + Ok(Bytes::from_static(&[])), + Ok(Bytes::from_static(&[8, 9, 10, 11])), + Ok(Bytes::from_static(&[])), + ]); + + let mut read = stream_reader(stream); + + let mut buf = [0; 5]; + read.read_exact(&mut buf).await?; + assert_eq!(buf, [0, 1, 2, 3, 4]); + + assert_eq!(read.read(&mut buf).await?, 3); + assert_eq!(&buf[..3], [5, 6, 7]); + + assert_eq!(read.read(&mut buf).await?, 4); + assert_eq!(&buf[..4], [8, 9, 10, 11]); + + assert_eq!(read.read(&mut buf).await?, 0); + + Ok(()) +} -- cgit v1.2.3