third_party/rust_crates/vendor/pulse/tests/select.rs

//   Copyright 2015 Colin Sherratt
//
//   Licensed under the Apache License, Version 2.0 (the "License");
//   you may not use this file except in compliance with the License.
//   You may obtain a copy of the License at
//
//       http://www.apache.org/licenses/LICENSE-2.0
//
//   Unless required by applicable law or agreed to in writing, software
//   distributed under the License is distributed on an "AS IS" BASIS,
//   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
//   See the License for the specific language governing permissions and
//   limitations under the License.


extern crate pulse;

use std::thread;
use pulse::*;

#[test]
fn select_one() {
    let (p, t) = Signal::new();
    let mut select = Select::new();
    let id = select.add(p);
    t.pulse();
    let p = select.next().unwrap();
    assert_eq!(id, p.id());
}

#[test]
fn select_three() {
    let (p0, t0) = Signal::new();
    let (p1, t1) = Signal::new();
    let (p2, t2) = Signal::new();

    let mut select = Select::new();
    let id0 = select.add(p0);
    let id1 = select.add(p1);
    let id2 = select.add(p2);

    t0.pulse();
    let p = select.next().unwrap();
    assert_eq!(id0, p.id());

    t1.pulse();
    let p = select.next().unwrap();
    assert_eq!(id1, p.id());

    t2.pulse();
    let p = select.next().unwrap();
    assert_eq!(id2, p.id());

    let p = select.next();
    assert!(p.is_none());
}

#[test]
fn select_thread() {
    let (p0, t0) = Signal::new();
    let (p1, t1) = Signal::new();
    let (p2, t2) = Signal::new();

    let mut select = Select::new();
    let id0 = select.add(p0);
    let id1 = select.add(p1);
    let id2 = select.add(p2);

    thread::spawn(move || {
        thread::sleep_ms(10);
        t0.pulse();
        thread::sleep_ms(10);
        t1.pulse();
        thread::sleep_ms(10);
        t2.pulse();
    });

    let p = select.next().unwrap();
    assert_eq!(id0, p.id());
    let p = select.next().unwrap();
    assert_eq!(id1, p.id());
    let p = select.next().unwrap();
    assert_eq!(id2, p.id());
    let p = select.next();
    assert!(p.is_none());
}

#[test]
fn select_barrier() {
    let (p0, t0) = Signal::new();
    let (p1, t1) = Signal::new();
    let (p2, t2) = Signal::new();

    let mut select = Select::new();
    let _ = select.add(p0);
    let _ = select.add(p1);
    let _ = select.add(p2);

    thread::spawn(move || {
        thread::sleep_ms(10);
        t0.pulse();
        thread::sleep_ms(10);
        t1.pulse();
        thread::sleep_ms(10);
        t2.pulse();
    });

    select.into_barrier().signal().wait().unwrap();
}

#[test]
fn select_already_pulsed() {
    let (p0, t0) = Signal::new();
    t0.pulse();

    let mut select = Select::new();
    let id0 = select.add(p0);

    let p = select.next().unwrap();
    assert_eq!(id0, p.id());
    let p = select.next();
    assert!(p.is_none());
}

#[test]
fn select_remove() {
    let (p0, t0) = Signal::new();
    let (p1, t1) = Signal::new();

    let mut select = Select::new();
    let id0 = select.add(p0);
    let id1 = select.add(p1);

    thread::spawn(move || {
        // This should only matter if the bug still exists
        // this thread will otherwise unblock the select
        thread::sleep_ms(100);
        t1.pulse();
    });

    t0.pulse();
    select.remove(id0).unwrap();
    assert_eq!(id1, select.next().unwrap().id());
}