Files
tokio/tokio/tests/io_util_empty.rs
T
Michael MaciasandWeijia Jiang dff4ecd0e7 io: implement AsyncSeek for Empty (#6663)
* io: implement `AsyncSeek` for `Empty`

* io: add more tests for seeking `Empty`

This adds more tests seeking with other kinds of `SeekFrom`. It follows
the same structure as used in
rust-lang/rust@f1cd17961c.

See <https://github.com/tokio-rs/tokio/pull/6663#discussion_r1659491016>.

* io: add inline attribute to `AsyncSeek` implementations

This follows the style of the other trait implementations of `Empty`.

See <https://github.com/tokio-rs/tokio/pull/6663#discussion_r1658835942>.

---------

Co-authored-by: Weijia Jiang <[email protected]>
2024-07-01 09:50:49 +08:00

62 lines
1.8 KiB
Rust

#![cfg(feature = "full")]
use tokio::io::{AsyncBufReadExt, AsyncReadExt, AsyncSeekExt};
#[tokio::test]
async fn empty_read_is_cooperative() {
tokio::select! {
biased;
_ = async {
loop {
let mut buf = [0u8; 4096];
let _ = tokio::io::empty().read(&mut buf).await;
}
} => {},
_ = tokio::task::yield_now() => {}
}
}
#[tokio::test]
async fn empty_buf_reads_are_cooperative() {
tokio::select! {
biased;
_ = async {
loop {
let mut buf = String::new();
let _ = tokio::io::empty().read_line(&mut buf).await;
}
} => {},
_ = tokio::task::yield_now() => {}
}
}
#[tokio::test]
async fn empty_seek() {
use std::io::SeekFrom;
let mut empty = tokio::io::empty();
assert!(matches!(empty.seek(SeekFrom::Start(0)).await, Ok(0)));
assert!(matches!(empty.seek(SeekFrom::Start(1)).await, Ok(0)));
assert!(matches!(empty.seek(SeekFrom::Start(u64::MAX)).await, Ok(0)));
assert!(matches!(empty.seek(SeekFrom::End(i64::MIN)).await, Ok(0)));
assert!(matches!(empty.seek(SeekFrom::End(-1)).await, Ok(0)));
assert!(matches!(empty.seek(SeekFrom::End(0)).await, Ok(0)));
assert!(matches!(empty.seek(SeekFrom::End(1)).await, Ok(0)));
assert!(matches!(empty.seek(SeekFrom::End(i64::MAX)).await, Ok(0)));
assert!(matches!(
empty.seek(SeekFrom::Current(i64::MIN)).await,
Ok(0)
));
assert!(matches!(empty.seek(SeekFrom::Current(-1)).await, Ok(0)));
assert!(matches!(empty.seek(SeekFrom::Current(0)).await, Ok(0)));
assert!(matches!(empty.seek(SeekFrom::Current(1)).await, Ok(0)));
assert!(matches!(
empty.seek(SeekFrom::Current(i64::MAX)).await,
Ok(0)
));
}