use criterion::{criterion_group, criterion_main, BenchmarkId, Criterion};
use std::hint::black_box;
use std::process::Command;
use std::time::Duration;
#[path = "bench_utils.rs"]
mod bench_utils;
use bench_utils::crush_binary;
fn bench_version_flag(c: &mut Criterion) {
let binary = crush_binary();
let mut group = c.benchmark_group("cli_startup");
group.measurement_time(Duration::from_secs(10));
group.sample_size(100);
group.bench_function("version_flag", |b| {
b.iter(|| {
let output = Command::new(&binary)
.arg("--version")
.output()
.expect("Failed to run crush --version");
black_box(output)
})
});
group.finish();
}
fn bench_invalid_command(c: &mut Criterion) {
let binary = crush_binary();
let mut group = c.benchmark_group("cli_startup");
group.measurement_time(Duration::from_secs(10));
group.sample_size(100);
group.bench_function("invalid_command", |b| {
b.iter(|| {
let output = Command::new(&binary)
.arg("nonexistent-command")
.output()
.expect("Failed to run crush with invalid command");
black_box(output)
})
});
group.finish();
}
fn bench_no_args(c: &mut Criterion) {
let binary = crush_binary();
let mut group = c.benchmark_group("cli_startup");
group.measurement_time(Duration::from_secs(10));
group.sample_size(100);
group.bench_function("no_args", |b| {
b.iter(|| {
let output = Command::new(&binary)
.output()
.expect("Failed to run crush with no args");
black_box(output)
})
});
group.finish();
}
fn bench_verbose_flags(c: &mut Criterion) {
let binary = crush_binary();
let mut group = c.benchmark_group("cli_startup");
group.measurement_time(Duration::from_secs(10));
group.sample_size(100);
for verbosity in ["-v", "-vv", "-vvv"].iter() {
group.bench_with_input(
BenchmarkId::new("verbose", verbosity),
verbosity,
|b, &v| {
b.iter(|| {
let output = Command::new(&binary)
.arg(v)
.arg("--version")
.output()
.expect("Failed to run crush with verbose flags");
black_box(output)
})
},
);
}
group.finish();
}
criterion_group!(
benches,
bench_version_flag,
bench_invalid_command,
bench_no_args,
bench_verbose_flags
);
criterion_main!(benches);