1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
extern crate zmtp;

#[macro_use]
mod core;
mod req;
mod rep;

pub use core::*;
pub use req::*;
pub use rep::*;

#[cfg(test)]
mod tests {
    use rep::*;
    use req::*;
    use std::io::{Result};
    use std::sync::mpsc::channel;
    use std::thread;

    const ADDRESS: &'static str = "localhost:50000";

    const BASIC_REQUEST:  &'static str = "My Little Request";
    const BASIC_RESPONSE: &'static str = "Server is not amused";

    #[allow(unused_assignments)]
    fn basic_client(mut sender: ReqSend) -> Result<String> {
        let receiver = sender.send_request(BASIC_REQUEST);
        let mut response = Ok(String::new());
        sender = receiver.receive_response(&mut response);
        response
    }

    #[allow(unused_assignments)]
    pub fn basic_server(mut receiver: RepReceive) -> Result<String> {
        let mut request = Ok(String::new());
        let sender = receiver.receive_request(&mut request);
        receiver = sender.send_response(BASIC_RESPONSE);
        request
    }

    #[test]
    fn basic_rep_req() {
        let (tx_client, rx_client) = channel();
        thread::spawn(move || {
            let req = req(ADDRESS).unwrap();
            let response = basic_client(req).unwrap();
            tx_client.send(response).unwrap();
        });

        let (tx_server, rx_server) = channel();
        thread::spawn(move || {
            let rep = rep(ADDRESS).unwrap();
            let request = basic_server(rep).unwrap();
            tx_server.send(request).unwrap();
        });

        assert_eq!(rx_client.recv().unwrap(), BASIC_RESPONSE);
        assert_eq!(rx_server.recv().unwrap(), BASIC_REQUEST);
    }
}

//  LocalWords:  Req RustyQlient TcpStream