typos/benches/checks.rs

mod data;

use assert_fs::prelude::*;
use criterion::{criterion_group, criterion_main, BenchmarkId, Criterion};
use typos_cli::file::FileChecker;

fn bench_checks(c: &mut Criterion) {
    let dict = typos_cli::dict::BuiltIn::new(Default::default());
    let tokenizer = typos::tokens::Tokenizer::new();
    let policy = typos_cli::policy::Policy::new()
        .dict(&dict)
        .tokenizer(&tokenizer);

    let mut group = c.benchmark_group("checks");
    for (name, sample) in data::DATA {
        let len = sample.len();
        group.bench_with_input(BenchmarkId::new("files", name), &len, |b, _| {
            let temp = assert_fs::TempDir::new().unwrap();
            let sample_path = temp.child("sample");
            sample_path.write_str(sample).unwrap();

            b.iter(|| {
                typos_cli::file::FoundFiles.check_file(
                    sample_path.path(),
                    true,
                    &policy,
                    &typos_cli::report::PrintSilent,
                )
            });

            temp.close().unwrap();
        });
        group.bench_with_input(BenchmarkId::new("identifiers", name), &len, |b, _| {
            let temp = assert_fs::TempDir::new().unwrap();
            let sample_path = temp.child("sample");
            sample_path.write_str(sample).unwrap();

            b.iter(|| {
                typos_cli::file::Identifiers.check_file(
                    sample_path.path(),
                    true,
                    &policy,
                    &typos_cli::report::PrintSilent,
                )
            });

            temp.close().unwrap();
        });
        group.bench_with_input(BenchmarkId::new("words", name), &len, |b, _| {
            let temp = assert_fs::TempDir::new().unwrap();
            let sample_path = temp.child("sample");
            sample_path.write_str(sample).unwrap();

            b.iter(|| {
                typos_cli::file::Words.check_file(
                    sample_path.path(),
                    true,
                    &policy,
                    &typos_cli::report::PrintSilent,
                )
            });

            temp.close().unwrap();
        });
        group.bench_with_input(BenchmarkId::new("typos", name), &len, |b, _| {
            let temp = assert_fs::TempDir::new().unwrap();
            let sample_path = temp.child("sample");
            sample_path.write_str(sample).unwrap();

            b.iter(|| {
                typos_cli::file::Typos.check_file(
                    sample_path.path(),
                    true,
                    &policy,
                    &typos_cli::report::PrintSilent,
                )
            });

            temp.close().unwrap();
        });
    }
    group.finish();
}

criterion_group!(benches, bench_checks,);
criterion_main!(benches);