mirror of
https://github.com/tokio-rs/tokio.git
synced 2026-08-27 00:00:12 +02:00
Carl Lerche
7b4c999341
Changes the set of `default` feature flags to `[]`. By default, only core traits are included without specifying feature flags. This makes it easier for users to pick the components they need. For convenience, a `full` feature flag is included that includes all components. Tests are configured to require the `full` feature. Testing individual feature flags will need to be moved to a separate crate. Closes #1791
111 lines
2.4 KiB
Rust
111 lines
2.4 KiB
Rust
#![warn(rust_2018_idioms)]
|
|
#![cfg(feature = "full")]
|
|
|
|
use tokio::sync::oneshot;
|
|
use tokio::time::{self, timeout, timeout_at, Instant};
|
|
use tokio_test::*;
|
|
|
|
use futures::future::pending;
|
|
use std::time::Duration;
|
|
|
|
#[tokio::test]
|
|
async fn simultaneous_deadline_future_completion() {
|
|
// Create a future that is immediately ready
|
|
let mut fut = task::spawn(timeout_at(Instant::now(), async {}));
|
|
|
|
// Ready!
|
|
assert_ready_ok!(fut.poll());
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn completed_future_past_deadline() {
|
|
// Wrap it with a deadline
|
|
let mut fut = task::spawn(timeout_at(Instant::now() - ms(1000), async {}));
|
|
|
|
// Ready!
|
|
assert_ready_ok!(fut.poll());
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn future_and_deadline_in_future() {
|
|
time::pause();
|
|
|
|
// Not yet complete
|
|
let (tx, rx) = oneshot::channel();
|
|
|
|
// Wrap it with a deadline
|
|
let mut fut = task::spawn(timeout_at(Instant::now() + ms(100), rx));
|
|
|
|
assert_pending!(fut.poll());
|
|
|
|
// Turn the timer, it runs for the elapsed time
|
|
time::advance(ms(90)).await;
|
|
|
|
assert_pending!(fut.poll());
|
|
|
|
// Complete the future
|
|
tx.send(()).unwrap();
|
|
assert!(fut.is_woken());
|
|
|
|
assert_ready_ok!(fut.poll()).unwrap();
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn future_and_timeout_in_future() {
|
|
time::pause();
|
|
|
|
// Not yet complete
|
|
let (tx, rx) = oneshot::channel();
|
|
|
|
// Wrap it with a deadline
|
|
let mut fut = task::spawn(timeout(ms(100), rx));
|
|
|
|
// Ready!
|
|
assert_pending!(fut.poll());
|
|
|
|
// Turn the timer, it runs for the elapsed time
|
|
time::advance(ms(90)).await;
|
|
|
|
assert_pending!(fut.poll());
|
|
|
|
// Complete the future
|
|
tx.send(()).unwrap();
|
|
|
|
assert_ready_ok!(fut.poll()).unwrap();
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn deadline_now_elapses() {
|
|
use futures::future::pending;
|
|
|
|
time::pause();
|
|
|
|
// Wrap it with a deadline
|
|
let mut fut = task::spawn(timeout_at(Instant::now(), pending::<()>()));
|
|
|
|
// Factor in jitter
|
|
// TODO: don't require this
|
|
time::advance(ms(1)).await;
|
|
|
|
assert_ready_err!(fut.poll());
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn deadline_future_elapses() {
|
|
time::pause();
|
|
|
|
// Wrap it with a deadline
|
|
let mut fut = task::spawn(timeout_at(Instant::now() + ms(300), pending::<()>()));
|
|
|
|
assert_pending!(fut.poll());
|
|
|
|
time::advance(ms(301)).await;
|
|
|
|
assert!(fut.is_woken());
|
|
assert_ready_err!(fut.poll());
|
|
}
|
|
|
|
fn ms(n: u64) -> Duration {
|
|
Duration::from_millis(n)
|
|
}
|