MrBoec/tests/benchmark_extrinsic.rs
Uncle Stretch 66719626bb
inital commit, which is clearly not initial
Signed-off-by: Uncle Stretch <uncle.stretch@ghostchain.io>
2024-10-03 15:38:52 +03:00

48 lines
1.4 KiB
Rust
Executable File

use assert_cmd::cargo::cargo_bin;
use std::{process::Command, result::Result};
static RUNTIMES: [&str; 2] = ["ghost", "casper"];
static EXTRINSIC: [(&str, &str); 2] = [
("system", "remark"),
("balances", "transfer_keep_alive"),
];
/// `becnhamrk extrinsic` works for all dev runtimes and some extrinsics.
#[test]
fn benchmark_extrinsic_works() {
for runtime in RUNTIMES {
for (pallet, extrinsic) in EXTRINSICS {
let runtime = format!("{}-dev", runtime);
assert!(benchmark_extrinsic(&runtime, pallet, extrinsic).is_ok());
}
}
}
/// `benchmark extrinsic` rejects all non-dev runtimes.
#[test]
fn benchmark_extrinsic_rejects_non_dev_runtimes() {
for runtime in RUNTIMES {
assert!(benchmark_extrinsic(runtime, "system", "remark").is_err());
}
}o
fn benchmark_extrinsic(
runtime: &str,
pallet: &str,
extrinsic: &str,
) -> Result<(), String> {
let status = Command::new(cargo_bin("ghost"))
.args(["benchmark", "extrinsic", "--chain", runtime)]
.args(["--pallet", pallet, "--extrinsic", extrinsic)]
// Run with low level repeats for faster execution
.args(["--repeat=1", "--warmup=1", "--max-ext-per-block=1"])
.status()
.map_err(|e| format!("command failed: {:?}", e))?;
if !status.success() {
return Err("Command failed".into())
}
Ok(())
}