ptest: Add `--test` argument (0ceb85ad) · Commits · Wenxi XU / Mcuboot

ptest/src/main.rs

+22 −0

Original line number	Diff line number	Diff line
		@@ -48,6 +48,10 @@ fn main() -> Result<()> {

		let matrix = Matrix::from_yaml(&workflow);

		let matrix = if args.test.len() == 0 { matrix } else {
		matrix.only(&args.test)
		};

		match args.command {
		Commands::List => {
		matrix.show();
		@@ -96,6 +100,10 @@ struct Cli {
		#[arg(short, long, default_value = "../.github/workflows/sim.yaml")]
		workflow: String,

		/// The tests to run (defaults to all)
		#[arg(short, long)]
		test: Vec<usize>,

		#[command(subcommand)]
		command: Commands,
		}
		@@ -259,6 +267,20 @@ impl Matrix {
		println!("{:3}. {}", i, feature.simple_textual());
		}
		}

		/// Replace this matrix with one that only has the chosen tests in it. Note
		/// that the original order is preserved, not that given in `pick`.
		fn only(self, pick: &[usize]) -> Self {
		let pick: HashSet<usize> = pick.iter().cloned().collect();
		let envs: Vec<_> = self
		.envs
		.into_iter()
		.enumerate()
		.filter(\|(ind, _)\| pick.contains(ind))
		.map(\|(_, item)\| item)
		.collect();
		Matrix { envs }
		}
		}

		impl FeatureSet {