mirror of
https://github.com/tokio-rs/tokio.git
synced 2026-08-24 00:00:11 +02:00
* 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]>
62 lines
1.8 KiB
Rust
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)
|
|
));
|
|
}
|