extern crate solana_ws as ws;

use std::net::Ipv4Addr;

struct Handler;
impl ws::Handler for Handler {}

#[test]
fn bind_port_zero() {
    let ws = ws::WebSocket::new(|_sender| Handler).unwrap();
    let ws = ws.bind("127.0.0.1:0").unwrap();

    let local_addr = ws.local_addr().unwrap();
    println!("Listening on {}", local_addr);

    assert_eq!(Ipv4Addr::new(127, 0, 0, 1), local_addr.ip());
    assert_ne!(0, local_addr.port());
}

#[test]
fn bind_try_multiple_addrs() {
    let invalid_addr = "99.99.99.99:0".parse().unwrap();
    let valid_addr = "127.0.0.1:9876".parse().unwrap();
    let addrs = vec![invalid_addr, valid_addr, invalid_addr];

    let ws = ws::WebSocket::new(|_sender| Handler).unwrap();
    let ws = ws.bind(&addrs[..]).unwrap();

    let local_addr = ws.local_addr().unwrap();
    assert_eq!(valid_addr, local_addr);
}