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
#[cfg(feature = "client")]
pub mod client;
#[cfg(feature = "server")]
pub mod server;

#[cfg(feature = "client")]
#[cfg(feature = "server")]
#[cfg(test)]
mod test {
    use super::{client, server};
    use myutil::{err::*, *};
    use std::{sync::Arc, thread, time::Duration};

    #[test]
    fn itworks() {
        const ADDR: &str = "127.0.0.1:9999";

        thread::spawn(|| {
            let cb = |recvd: &[u8],
                      hdr: Arc<server::Hdr>,
                      peer: server::PeerAddr|
             -> Result<()> {
                assert_eq!(b"ABC", recvd);
                hdr.sendto(b"DEF", &peer).map(|n| assert_eq!(n, 3)).c(d!())
            };
            pnk!(server::start_server(ADDR, None, cb, false));
        });

        // wait server to start
        thread::sleep(Duration::from_secs(1));

        let cli = pnk!(client::Hdr::new());
        assert_eq!(3, pnk!(cli.sendto_straddr(b"ABC", ADDR)));

        let buf = &mut [0; 8];
        let res = pnk!(cli.recvfrom(buf));
        assert_eq!(b"DEF", &buf[0..res.0]);
        assert_eq!(res.0, 3);
        assert_eq!(pnk!(res.1).to_string().as_str(), ADDR);
    }
}