scheduling_test.rs

use declarative_dataflow::scheduling::{AsScheduler, RealtimeScheduler, SchedulingEvent};
use std::time::Duration;

#[test]
fn test_schedule_now() {
    let mut scheduler = RealtimeScheduler::new();

    assert!(!scheduler.has_pending());
    assert!(scheduler.until_next().is_none());

    scheduler.event_after(Duration::from_secs(0), SchedulingEvent::Tick);

    assert!(scheduler.has_pending());
    assert_eq!(
        scheduler.next().unwrap().schedule(),
        Some(SchedulingEvent::Tick)
    );
}

#[test]
fn test_schedule_after() {
    let mut scheduler = RealtimeScheduler::new();

    scheduler.event_after(Duration::from_secs(2), SchedulingEvent::Tick);

    assert!(!scheduler.has_pending());
    assert!(scheduler.next().is_none());
    assert!(scheduler.until_next().is_some());

    std::thread::sleep(scheduler.until_next().unwrap());

    assert!(scheduler.has_pending());
    assert_eq!(
        scheduler.next().unwrap().schedule(),
        Some(SchedulingEvent::Tick)
    );
    assert!(scheduler.until_next().is_none());
}