// This file is part of Bizinikiwi.

#![allow(deprecated)]
// Copyright (C) Parity Technologies (UK) Ltd.
// SPDX-License-Identifier: GPL-3.0-or-later WITH Classpath-exception-2.0

// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.

// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License
// along with this program. If not, see <https://www.gnu.org/licenses/>.

#![cfg(feature = "runtime-benchmarks")]

use assert_cmd::cargo::cargo_bin;
use std::process::Command;

/// `benchmark pezpallet` works for the different combinations of `steps` and `repeat`.
#[test]
fn benchmark_pezpallet_works() {
	// Some invalid combinations:
	benchmark_pezpallet(0, 10, false);
	benchmark_pezpallet(1, 10, false);
	// ... and some valid:
	benchmark_pezpallet(2, 1, true);
	benchmark_pezpallet(50, 20, true);
	benchmark_pezpallet(20, 50, true);
}

#[test]
fn benchmark_pezpallet_args_work() {
	benchmark_pezpallet_args(&["--list", "--pezpallet=pezpallet_balances"], true);
	benchmark_pezpallet_args(&["--list", "--pezpallet=pezpallet_balances"], true);
	benchmark_pezpallet_args(
		&["--list", "--pezpallet=pezpallet_balances", "--genesis-builder=spec-genesis"],
		true,
	);
	benchmark_pezpallet_args(
		&[
			"--list",
			"--pezpallet=pezpallet_balances",
			"--chain=dev",
			"--genesis-builder=spec-genesis",
		],
		true,
	);
	benchmark_pezpallet_args(
		&[
			"--list",
			"--pezpallet=pezpallet_balances",
			"--chain=dev",
			"--genesis-builder=spec-runtime",
		],
		true,
	);
	// Error because no runtime is provided:
	benchmark_pezpallet_args(
		&["--list", "--pezpallet=pezpallet_balances", "--chain=dev", "--genesis-builder=runtime"],
		false,
	);
}

fn benchmark_pezpallet(steps: u32, repeat: u32, should_work: bool) {
	let status = Command::new(cargo_bin("bizinikiwi-node"))
		.args(["benchmark", "pezpallet", "--dev"])
		// Use the `addition` benchmark since is the fastest.
		.args(["--pezpallet", "pezframe-benchmarking", "--extrinsic", "addition"])
		.args(["--steps", &format!("{}", steps), "--repeat", &format!("{}", repeat)])
		.args([
			"--wasm-execution=compiled",
			"--no-storage-info",
			"--no-median-slopes",
			"--no-min-squares",
			"--heap-pages=4096",
		])
		.status()
		.unwrap();

	assert_eq!(status.success(), should_work);
}

fn benchmark_pezpallet_args(args: &[&str], should_work: bool) {
	let status = Command::new(cargo_bin("bizinikiwi-node"))
		.args(["benchmark", "pezpallet"])
		.args(args)
		.status()
		.unwrap();

	assert_eq!(status.success(), should_work);
}