tokio/tokio-test/tests/clock.rs

#![cfg(feature = "broken")]
#![warn(rust_2018_idioms)]

use futures::Future;
use std::time::{Duration, Instant};
use tokio_test::clock::MockClock;
use tokio_test::task::MockTask;
use tokio_test::{assert_not_ready, assert_ready};
use tokio_timer::delay;

#[test]
fn clock() {
    let mut mock = MockClock::new();

    mock.enter(|handle| {
        let deadline = Instant::now() + Duration::from_secs(1);
        let mut delay = delay(deadline);

        assert_not_ready!(delay.poll());

        handle.advance(Duration::from_secs(2));

        assert_ready!(delay.poll());
    });
}

#[test]
fn notify() {
    let deadline = Instant::now() + Duration::from_secs(1);
    let mut mock = MockClock::new();
    let mut task = MockTask::new();

    mock.enter(|handle| {
        let mut delay = delay(deadline);

        task.enter(|| assert_not_ready!(delay.poll()));

        handle.advance(Duration::from_secs(1));

        assert!(task.is_notified());
        assert_ready!(delay.poll());
    });
}