use std::time::{Duration, Instant};

use bytes::Bytes;
use futures::{join, prelude::*};
use srt_tokio::SrtSocket;
use tokio::time::sleep;

#[tokio::test]
async fn rendezvous() {
    let _ = pretty_env_logger::try_init();

    let a = SrtSocket::builder()
        .local_port(5001)
        .rendezvous("127.0.0.1:5000");

    let b = SrtSocket::builder()
        .local_port(5000)
        .rendezvous("127.0.0.1:5001");

    join!(
        async move {
            let mut a = a.await.unwrap();

            a.send((Instant::now(), Bytes::from_static(b"hi")))
                .await
                .unwrap();

            sleep(Duration::from_millis(10)).await;

            a.close().await.unwrap();
        },
        async move {
            let mut b = b.await.unwrap();

            assert_eq!(&*b.try_next().await.unwrap().unwrap().1, b"hi");
            assert_eq!(b.try_next().await.unwrap(), None);

            b.close().await.unwrap();
        }
    );
}