src/vendor/net2/tests/all.rs

   1 extern crate net2;
   2
   3 use std::net::{TcpStream, IpAddr, Ipv4Addr, Ipv6Addr};
   4 use std::io::prelude::*;
   5 use std::thread;
   6
   7 use net2::TcpBuilder;
   8
   9 macro_rules! t {
  10     ($e:expr) => (match $e {
  11         Ok(e) => e,
  12         Err(e) => panic!("{} failed with: {}", stringify!($e), e),
  13     })
  14 }
  15
  16 #[test]
  17 fn smoke_build_listener() {
  18     let b = t!(TcpBuilder::new_v4());
  19     t!(b.bind("127.0.0.1:0"));
  20
  21     let addr = t!(b.local_addr());
  22     assert_eq!(addr.ip(), IpAddr::V4(Ipv4Addr::new(127, 0, 0, 1)));
  23
  24     let listener = t!(b.listen(200));
  25
  26     let t = thread::spawn(move || {
  27         let mut s = t!(listener.accept()).0;
  28         let mut b = [0; 4];
  29         t!(s.read(&mut b));
  30         assert_eq!(b, [1, 2, 3, 0]);
  31     });
  32
  33     let mut stream = t!(TcpStream::connect(&addr));
  34     t!(stream.write(&[1,2,3]));
  35     t.join().unwrap();
  36 }
  37
  38 #[test]
  39 fn smoke_build_listener_v6() {
  40     let b = t!(TcpBuilder::new_v6());
  41     t!(b.bind("::1:0"));
  42
  43     let addr = t!(b.local_addr());
  44     assert_eq!(addr.ip(), IpAddr::V6(Ipv6Addr::new(0, 0, 0, 0, 0, 0, 0, 1)));
  45 }