use std::time::Duration;

use rand::Rng;

use bench_rs::{bench, Stats};
use bench_rs::Bencher;

#[test]
fn test_bencher() {
    let data = rand::thread_rng().sample_iter(rand::distributions::Alphanumeric).take(1000).collect::<String>();

    let mut bencher = Bencher::new("test_bencher", 1000, data.len());
    bencher.iter(|| {
        let _ = rcnb_rs::encode(&data);
    });

    // Custom formatting
    bencher.format_fn = |stats: &Stats, b: &Bencher| {
        println!("{}: custom formatting: {:?}\n", &b.name, stats)
    };

    bencher.finish();
}

#[bench(count = 100)]
fn test_async_with_tokio(b: &mut Bencher) {
    let mut rt = tokio::runtime::Runtime::new().unwrap();
    let fut = b.async_iter(|| async {
        futures_timer::Delay::new(Duration::from_nanos(20_000_000)).await
    });
    rt.block_on(fut);
}

#[bench(count = 100)]
fn test_async_with_smol(b: &mut Bencher) {
    let fut = b.async_iter(|| async {
        futures_timer::Delay::new(Duration::from_nanos(20_000_000)).await
    });
    smol::block_on(fut);
}

#[bench(count = 100)]
fn test_async_with_async_std(b: &mut Bencher) {
    let fut = b.async_iter(|| async {
        futures_timer::Delay::new(Duration::from_nanos(20_000_000)).await
    });
    async_std::task::block_on(fut);
}

#[bench(count = 100)]
fn test_async_with_futures(b: &mut Bencher) {
    let fut = b.async_iter(|| async {
        futures_timer::Delay::new(Duration::from_nanos(20_000_000)).await
    });
    futures::executor::block_on(fut);
}

#[bench(name = "test_rcnb_encoding")]
fn test_rcnb(b: &mut Bencher) {
    let data = rand::thread_rng().sample_iter(rand::distributions::Alphanumeric).take(1000).collect::<String>();
    b.iter(|| {
        let _ = rcnb_rs::encode(&data);
    });
    b.bytes = data.len()
}

#[bench]
fn test_base64(b: &mut Bencher) {
    let data = rand::thread_rng().sample_iter(rand::distributions::Alphanumeric).take(1000).collect::<String>();
    b.iter(|| {
        let _ = base64::encode(&data);
    });
    b.bytes = data.len()
}

#[bench(no_test)]
fn test_no_run(_: &mut Bencher) {
    println!("no #[test]");
}