cargo-rbmt 0.2.0

Maintainer tools for rust-bitcoin projects
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39

// SPDX-License-Identifier: MIT AND Apache-2.0

//! Benchmark testing tasks.

use xshell::Shell;

use crate::environment::{get_workspace_packages, OutputMode, ProgressGuard};
use crate::lock::LockFile;
use crate::toolchain::{prepare_toolchain, Toolchain};

/// Run benchmark tests for all crates in the workspace.
pub fn run(
    sh: &Shell,
    lockfile: LockFile,
    packages: &[String],
) -> Result<(), Box<dyn std::error::Error>> {
    let packages = get_workspace_packages(sh, packages)?;
    let _lockfile_guard = lockfile.activate(sh)?;
    let _progress = ProgressGuard::new();
    prepare_toolchain(sh, Toolchain::Nightly)?;
    rbmt_eprintln!("Running bench tests for {} crates", packages.len());

    for package in packages {
        rbmt_eprintln!("Running bench tests in: {}", package.dir.display());

        // Use pushd pattern to change and restore directory.
        let _dir = sh.push_dir(&package.dir);

        // Capture output and show in stdout for verbose mode.
        let output =
            rbmt_cmd!(sh, "cargo --locked bench").env("RUSTFLAGS", "--cfg=bench").read()?;
        if matches!(OutputMode::from_env(), OutputMode::Verbose) {
            println!("{}", output);
        }
    }

    rbmt_eprintln!("Benches complete.");
    Ok(())
}