T0b1/wasmtime
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

poll_oneoff test: outline assertions

Browse Source 
this makes the panic message have the line number, which
is handy
This commit is contained in:
Pat Hickey
2021-01-14 17:40:04 -08:00
parent 9a1ce1a272
commit 7d7acc4614
1 changed files with 12 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
crates/test-programs/wasi-tests/src/bin/poll_oneoff.rs
View File

@@ -4,19 +4,14 @@ use wasi_tests::{open_scratch_directory, STDERR_FD, STDIN_FD, STDOUT_FD};
const CLOCK_ID: wasi::Userdata = 0x0123_45678; const CLOCK_ID: wasi::Userdata = 0x0123_45678;
unsafe fn poll_oneoff_impl(r#in: &[wasi::Subscription], nexpected: usize) -> Vec<wasi::Event> { unsafe fn poll_oneoff_impl(r#in: &[wasi::Subscription]) -> Result<Vec<wasi::Event>, wasi::Error> {
let mut out: Vec<wasi::Event> = Vec::new(); let mut out: Vec<wasi::Event> = Vec::new();
out.resize_with(r#in.len(), || { out.resize_with(r#in.len(), || {
MaybeUninit::<wasi::Event>::zeroed().assume_init() MaybeUninit::<wasi::Event>::zeroed().assume_init()
}); });
let size = wasi::poll_oneoff(r#in.as_ptr(), out.as_mut_ptr(), r#in.len()) let size = wasi::poll_oneoff(r#in.as_ptr(), out.as_mut_ptr(), r#in.len())?;
.expect("poll_oneoff should succeed"); out.truncate(size);
assert_eq!( Ok(out)
size, nexpected,
"poll_oneoff should return {} events",
nexpected
);
out
} }
unsafe fn test_empty_poll() { unsafe fn test_empty_poll() {
@@ -45,7 +40,8 @@ unsafe fn test_timeout() {
u: wasi::SubscriptionUU { clock }, u: wasi::SubscriptionUU { clock },
}, },
}]; }];
let out = poll_oneoff_impl(&r#in, 1); let out = poll_oneoff_impl(&r#in).unwrap();
assert_eq!(out.len(), 1, "should return 1 event");
let event = &out[0]; let event = &out[0];
assert_eq!( assert_eq!(
event.error, event.error,
@@ -92,7 +88,8 @@ unsafe fn test_stdin_read() {
}, },
}, },
]; ];
let out = poll_oneoff_impl(&r#in, 1); let out = poll_oneoff_impl(&r#in).unwrap();
assert_eq!(out.len(), 1, "should return 1 event");
let event = &out[0]; let event = &out[0];
assert_eq!( assert_eq!(
event.error, event.error,
@@ -137,7 +134,8 @@ unsafe fn test_stdout_stderr_write() {
}, },
}, },
]; ];
let out = poll_oneoff_impl(&r#in, 2); let out = poll_oneoff_impl(&r#in).unwrap();
assert_eq!(out.len(), 2, "should return 2 events");
assert_eq!( assert_eq!(
out[0].userdata, 1, out[0].userdata, 1,
"the event.userdata should contain fd userdata specified by the user" "the event.userdata should contain fd userdata specified by the user"
@@ -192,7 +190,8 @@ unsafe fn test_fd_readwrite(fd: wasi::Fd, error_code: wasi::Errno) {
}, },
}, },
]; ];
let out = poll_oneoff_impl(&r#in, 2); let out = poll_oneoff_impl(&r#in).unwrap();
assert_eq!(out.len(), 2, "should return 2 events");
assert_eq!( assert_eq!(
out[0].userdata, 1, out[0].userdata, 1,
"the event.userdata should contain fd userdata specified by the user" "the event.userdata should contain fd userdata specified by the user"