diff options
author | Carl Lerche <me@carllerche.com> | 2019-10-26 08:02:49 -0700 |
---|---|---|
committer | GitHub <noreply@github.com> | 2019-10-26 08:02:49 -0700 |
commit | 987ba7373cf95c570bf23768c6021f7a7508286e (patch) | |
tree | 616c62a65cb6b0c7e4c3e31f92f01c8d4749b0c3 /tokio/tests/io_read.rs | |
parent | 227533d456fe32e48ffcd3796f1e6c8f9318b230 (diff) |
io: move into `tokio` crate (#1691)
A step towards collapsing Tokio sub crates into a single `tokio`
crate (#1318).
The `io` implementation is now provided by the main `tokio` crate.
Functionality can be opted out of by using the various net related
feature flags.
Diffstat (limited to 'tokio/tests/io_read.rs')
-rw-r--r-- | tokio/tests/io_read.rs | 37 |
1 files changed, 37 insertions, 0 deletions
diff --git a/tokio/tests/io_read.rs b/tokio/tests/io_read.rs new file mode 100644 index 00000000..01cd2fd6 --- /dev/null +++ b/tokio/tests/io_read.rs @@ -0,0 +1,37 @@ +#![warn(rust_2018_idioms)] + +use tokio::io::{AsyncRead, AsyncReadExt}; +use tokio_test::assert_ok; + +use std::io; +use std::pin::Pin; +use std::task::{Context, Poll}; + +#[tokio::test] +async fn read() { + #[derive(Default)] + struct Rd { + poll_cnt: usize, + } + + impl AsyncRead for Rd { + fn poll_read( + mut self: Pin<&mut Self>, + _cx: &mut Context<'_>, + buf: &mut [u8], + ) -> Poll<io::Result<usize>> { + assert_eq!(0, self.poll_cnt); + self.poll_cnt += 1; + + buf[0..11].copy_from_slice(b"hello world"); + Poll::Ready(Ok(11)) + } + } + + let mut buf = Box::new([0; 11]); + let mut rd = Rd::default(); + + let n = assert_ok!(rd.read(&mut buf[..]).await); + assert_eq!(n, 11); + assert_eq!(buf[..], b"hello world"[..]); +} |