42 lines
1.3 KiB
Rust
42 lines
1.3 KiB
Rust
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());
|
|
}
|
|
}
|
|
|
|
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(())
|
|
}
|