//! Benches the raw [`TcpCon`] and [`UdpCon`] abstractions.

#![feature(test)]
extern crate test;

use crate::helper::create_tcp_con_pair;
use test::Bencher;

mod helper;

#[bench]
fn single_raw_tcp_big(b: &mut Bencher) {
    let (s1, mut s2) = create_tcp_con_pair();
    s1.set_nonblocking(true).unwrap();

    let msg = ['A' as u8; 504];

    b.iter(|| {
        s1.send(0, &msg[..]).unwrap();
        let recv = s2.recv().unwrap();
        assert_eq!(recv, (0, &msg[..]));
    })
}

#[bench]
fn single_raw_tcp_small(b: &mut Bencher) {
    let (s1, mut s2) = create_tcp_con_pair();
    s1.set_nonblocking(true).unwrap();

    let msg = ['A' as u8; 10];

    b.iter(|| {
        s1.send(0, &msg[..]).unwrap();
        let recv = s2.recv().unwrap();
        assert_eq!(recv, (0, &msg[..]));
    })
}

#[bench]
fn many_raw_tcp_big(b: &mut Bencher) {
    let (s1, mut s2) = create_tcp_con_pair();
    s1.set_nonblocking(true).unwrap();

    let msg = ['A' as u8; 504];

    b.iter(|| {
        for _ in 0..100 {
            s1.send(0, &msg).unwrap();
        }
        for _ in 0..100 {
            let (_mid, _msg) = s2.recv().unwrap();
        }
        // Data validation is done in different tests.
    })
}

#[bench]
fn many_raw_tcp_small(b: &mut Bencher) {
    let (s1, mut s2) = create_tcp_con_pair();
    s1.set_nonblocking(true).unwrap();

    let msg = ['A' as u8; 10];

    b.iter(|| {
        for _ in 0..100 {
            s1.send(0, &msg).unwrap();
        }
        for _ in 0..100 {
            let (_mid, _msg) = s2.recv().unwrap();
        }
        // Data validation is done in different tests.
    })
}