48 lines
1.4 KiB
Rust
48 lines
1.4 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());
|
||
|
|
}
|
||
|
|
}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(())
|
||
|
|
}
|