#![feature(test)]
extern crate test;
use test::Bencher;

use crate::helper::test_messages::TcpMsg;

mod helper;

#[bench]
fn single_tcp_big(b: &mut Bencher) {
    let (client, mut server) = helper::create_client_server_pair();

    let string: String = vec!['A'; 504].into_iter().collect();
    let s_msg = TcpMsg::new(string);

    b.iter(|| {
        server.clear_msgs();
        client.send(&s_msg).unwrap();
        while server.recv_msgs() == 0 {}
        let msg = server.recv::<TcpMsg>().next().unwrap();
        assert_eq!(msg.m, &s_msg);
    })
}

#[bench]
fn single_tcp_small(b: &mut Bencher) {
    let (client, mut server) = helper::create_client_server_pair();

    let string: String = vec!['A'; 10].into_iter().collect();
    let s_msg = TcpMsg::new(string);

    b.iter(|| {
        server.clear_msgs();
        client.send(&s_msg).unwrap();
        while server.recv_msgs() == 0 {}
        let msg = server.recv::<TcpMsg>().next().unwrap();
        assert_eq!(msg.m, &s_msg);
    })
}

#[bench]
fn many_tcp_big(b: &mut Bencher) {
    let (client, mut server) = helper::create_client_server_pair();

    let string: String = vec!['A'; 504].into_iter().collect();
    let s_msg = TcpMsg::new(string);

    b.iter(|| {
        server.clear_msgs();
        for _ in 0..100 {
            client.send(&s_msg).unwrap();
        }
        let mut n = 0;
        while n < 100 {
            n += server.recv_msgs();
        }
        assert_eq!(server.recv::<TcpMsg>().count(), 100);
    })
}

#[bench]
fn many_tcp_small(b: &mut Bencher) {
    let (client, mut server) = helper::create_client_server_pair();

    let string: String = vec!['A'; 10].into_iter().collect();
    let s_msg = TcpMsg::new(string);

    b.iter(|| {
        server.clear_msgs();
        for _ in 0..100 {
            client.send(&s_msg).unwrap();
        }
        let mut n = 0;
        while n < 100 {
            n += server.recv_msgs();
        }
        assert_eq!(server.recv::<TcpMsg>().count(), 100);
    })
}