mirror of
https://chromium.googlesource.com/crosvm/crosvm
synced 2024-11-24 20:48:55 +00:00
ee82228475
Change-Id: Ib235ac810bb70a0027b9ff431b76b1c5fe14ada2 Reviewed-on: https://chromium-review.googlesource.com/c/crosvm/crosvm/+/4567121 Commit-Queue: Frederick Mayle <fmayle@google.com> Reviewed-by: Richard Zhang <rizhang@google.com> Reviewed-by: Daniel Verkamp <dverkamp@chromium.org>
52 lines
1.5 KiB
Rust
52 lines
1.5 KiB
Rust
// Copyright 2023 The ChromiumOS Authors
|
|
// Use of this source code is governed by a BSD-style license that can be
|
|
// found in the LICENSE file.
|
|
|
|
use cros_async::Executor;
|
|
use cros_async::ExecutorKind;
|
|
|
|
#[cfg(unix)]
|
|
fn all_kinds() -> Vec<ExecutorKind> {
|
|
let mut kinds = vec![ExecutorKind::Fd];
|
|
if cros_async::is_uring_stable() {
|
|
kinds.push(ExecutorKind::Uring);
|
|
}
|
|
kinds
|
|
}
|
|
#[cfg(windows)]
|
|
fn all_kinds() -> Vec<ExecutorKind> {
|
|
vec![ExecutorKind::Handle]
|
|
}
|
|
|
|
#[test]
|
|
fn cancel_pending_task() {
|
|
for kind in all_kinds() {
|
|
let ex = Executor::with_executor_kind(kind).unwrap();
|
|
let task = ex.spawn(std::future::pending::<()>());
|
|
assert_eq!(ex.run_until(task.cancel()).unwrap(), None);
|
|
}
|
|
}
|
|
|
|
// Testing a completed task without relying on implementation details is tricky. We create a future
|
|
// that signals a channel when it is polled so that we can delay the `task.cancel()` call until we
|
|
// know the task has been executed.
|
|
#[test]
|
|
fn cancel_ready_task() {
|
|
for kind in all_kinds() {
|
|
let ex = Executor::with_executor_kind(kind).unwrap();
|
|
let (s, r) = futures::channel::oneshot::channel();
|
|
let mut s = Some(s);
|
|
let task = ex.spawn(futures::future::poll_fn(move |_| {
|
|
s.take().unwrap().send(()).unwrap();
|
|
std::task::Poll::Ready(5)
|
|
}));
|
|
assert_eq!(
|
|
ex.run_until(async {
|
|
r.await.unwrap();
|
|
task.cancel().await
|
|
})
|
|
.unwrap(),
|
|
Some(5)
|
|
);
|
|
}
|
|
}
|