68 lines
2.0 KiB
Rust
68 lines
2.0 KiB
Rust
use anyhow::{Context, Error};
|
|
use wasi_common::{
|
|
file::{FdFlags, OFlags},
|
|
sched::{Poll, RwEventFlags, SubscriptionResult, Userdata},
|
|
WasiDir,
|
|
};
|
|
use wasi_tokio::{sched::poll_oneoff, Dir};
|
|
|
|
#[tokio::test(flavor = "multi_thread")]
|
|
async fn empty_file_readable() -> Result<(), Error> {
|
|
let workspace = unsafe { cap_tempfile::tempdir().expect("create tempdir") };
|
|
workspace.create_dir("d").context("create dir")?;
|
|
let d = workspace.open_dir("d").context("open dir")?;
|
|
let d = Dir::from_cap_std(d);
|
|
|
|
let mut readable_f = d
|
|
.open_file(false, "f", OFlags::CREATE, true, false, FdFlags::empty())
|
|
.await
|
|
.context("create readable file")?;
|
|
|
|
let mut poll = Poll::new();
|
|
poll.subscribe_read(&mut *readable_f, Userdata::from(123));
|
|
poll_oneoff(&mut poll).await?;
|
|
|
|
let events = poll.results();
|
|
|
|
assert_eq!(events.len(), 1);
|
|
match events[0] {
|
|
(SubscriptionResult::Read(Ok((0, flags))), ud) => {
|
|
assert_eq!(flags, RwEventFlags::empty());
|
|
assert_eq!(ud, Userdata::from(123));
|
|
}
|
|
_ => panic!(""),
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[tokio::test(flavor = "multi_thread")]
|
|
async fn empty_file_writable() -> Result<(), Error> {
|
|
let workspace = unsafe { cap_tempfile::tempdir().expect("create tempdir") };
|
|
workspace.create_dir("d").context("create dir")?;
|
|
let d = workspace.open_dir("d").context("open dir")?;
|
|
let d = Dir::from_cap_std(d);
|
|
|
|
let mut writable_f = d
|
|
.open_file(false, "f", OFlags::CREATE, true, true, FdFlags::empty())
|
|
.await
|
|
.context("create writable file")?;
|
|
|
|
let mut poll = Poll::new();
|
|
poll.subscribe_write(&mut *writable_f, Userdata::from(123));
|
|
poll_oneoff(&mut poll).await?;
|
|
|
|
let events = poll.results();
|
|
|
|
assert_eq!(events.len(), 1);
|
|
match events[0] {
|
|
(SubscriptionResult::Write(Ok((0, flags))), ud) => {
|
|
assert_eq!(flags, RwEventFlags::empty());
|
|
assert_eq!(ud, Userdata::from(123));
|
|
}
|
|
_ => panic!(""),
|
|
}
|
|
|
|
Ok(())
|
|
}
|