diff options
Diffstat (limited to 'tokio/tests/io_write.rs')
-rw-r--r-- | tokio/tests/io_write.rs | 47 |
1 files changed, 0 insertions, 47 deletions
diff --git a/tokio/tests/io_write.rs b/tokio/tests/io_write.rs deleted file mode 100644 index 482f21ba..00000000 --- a/tokio/tests/io_write.rs +++ /dev/null @@ -1,47 +0,0 @@ -#![deny(warnings, rust_2018_idioms)] -#![feature(async_await)] - -use tokio::io::{AsyncWrite, AsyncWriteExt}; -use tokio_test::assert_ok; - -use bytes::BytesMut; -use std::io; -use std::pin::Pin; -use std::task::{Context, Poll}; - -#[tokio::test] -async fn write() { - struct Wr { - buf: BytesMut, - cnt: usize, - } - - impl AsyncWrite for Wr { - fn poll_write( - mut self: Pin<&mut Self>, - _cx: &mut Context<'_>, - buf: &[u8], - ) -> Poll<io::Result<usize>> { - assert_eq!(self.cnt, 0); - self.buf.extend(&buf[0..4]); - Ok(4).into() - } - - fn poll_flush(self: Pin<&mut Self>, _cx: &mut Context<'_>) -> Poll<io::Result<()>> { - Ok(()).into() - } - - fn poll_shutdown(self: Pin<&mut Self>, _cx: &mut Context<'_>) -> Poll<io::Result<()>> { - Ok(()).into() - } - } - - let mut wr = Wr { - buf: BytesMut::with_capacity(64), - cnt: 0, - }; - - let n = assert_ok!(wr.write(b"hello world").await); - assert_eq!(n, 4); - assert_eq!(wr.buf, b"hell"[..]); -} |