Imported Upstream version 0.7

[rustc.git] / src / test / run-pass / pipe-select.rs

// xfail-fast

// Copyright 2012 The Rust Project Developers. See the COPYRIGHT
// file at the top-level directory of this distribution and at
// http://rust-lang.org/COPYRIGHT.
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.

// xfail-pretty
// xfail-win32

extern mod extra;
use extra::timer::sleep;
use extra::uv;

use std::cell::Cell;
use std::pipes::*;
use std::pipes;
use std::task;

proto! oneshot (
    waiting:send {
        signal -> !
    }
)

proto! stream (
    Stream:send<T:Send> {
        send(T) -> Stream<T>
    }
)

pub fn spawn_service<T:Send,Tb:Send>(
            init: extern fn() -> (RecvPacketBuffered<T, Tb>,
                                  SendPacketBuffered<T, Tb>),
            service: ~fn(v: RecvPacketBuffered<T, Tb>))
        -> SendPacketBuffered<T, Tb> {
    let (server, client) = init();

    // This is some nasty gymnastics required to safely move the pipe
    // into a new task.
    let server = Cell::new(server);
    do task::spawn {
        service(server.take());
    }

    client
}

pub fn main() {
    use oneshot::client::*;
    use stream::client::*;

    let iotask = &uv::global_loop::get();

    let c = spawn_service(stream::init, |p| {
        error!("waiting for pipes");
        let stream::send(x, p) = recv(p);
        error!("got pipes");
        let (left, right) : (oneshot::server::waiting,
                             oneshot::server::waiting)
            = x;
        error!("selecting");
        let (i, _, _) = select(~[left, right]);
        error!("selected");
        assert_eq!(i, 0);

        error!("waiting for pipes");
        let stream::send(x, _) = recv(p);
        error!("got pipes");
        let (left, right) : (oneshot::server::waiting,
                             oneshot::server::waiting)
            = x;
        error!("selecting");
        let (i, m, _) = select(~[left, right]);
        error!("selected %?", i);
        if m.is_some() {
            assert_eq!(i, 1);
        }
    });

    let (p1, c1) = oneshot::init();
    let (p2, _c2) = oneshot::init();

    let c = send(c, (p1, p2));

    sleep(iotask, 100);

    signal(c1);

    let (p1, _c1) = oneshot::init();
    let (p2, c2) = oneshot::init();

    send(c, (p1, p2));

    sleep(iotask, 100);

    signal(c2);

    test_select2();
}

fn test_select2() {
    let (ap, ac) = stream::init();
    let (bp, bc) = stream::init();

    stream::client::send(ac, 42);

    match pipes::select2(ap, bp) {
      Left(*) => { }
      Right(*) => { fail!() }
    }

    stream::client::send(bc, ~"abc");

    error!("done with first select2");

    let (ap, ac) = stream::init();
    let (bp, bc) = stream::init();

    stream::client::send(bc, ~"abc");

    match pipes::select2(ap, bp) {
      Left(*) => { fail!() }
      Right(*) => { }
    }

    stream::client::send(ac, 42);
}