integration/tests/bench/http.rs

use super::run_bench_and_wait_for_finish;
use iggy::utils::byte_size::IggyByteSize;
use integration::test_server::{IpAddrKind, TestServer, Transport};
use serial_test::parallel;
use std::str::FromStr;

#[test]
#[parallel]
fn http_ipv4_bench() {
    let mut test_server = TestServer::default();
    test_server.start();
    let server_addr = test_server.get_http_api_addr().unwrap();
    run_bench_and_wait_for_finish(
        &server_addr,
        Transport::Http,
        "pinned-producer",
        IggyByteSize::from_str("8MB").unwrap(),
    );
    run_bench_and_wait_for_finish(
        &server_addr,
        Transport::Http,
        "pinned-consumer",
        IggyByteSize::from_str("8MB").unwrap(),
    );
}

#[cfg_attr(feature = "ci-qemu", ignore)]
#[test]
#[parallel]
fn http_ipv6_bench() {
    let mut test_server = TestServer::new(None, true, None, IpAddrKind::V6);
    test_server.start();
    let server_addr = test_server.get_http_api_addr().unwrap();
    run_bench_and_wait_for_finish(
        &server_addr,
        Transport::Http,
        "pinned-producer",
        IggyByteSize::from_str("8MB").unwrap(),
    );
    run_bench_and_wait_for_finish(
        &server_addr,
        Transport::Http,
        "pinned-consumer",
        IggyByteSize::from_str("8MB").unwrap(),
    );
}