Add handle_n_inputs function in process test

1 files changed, 8 insertions, 0 deletions

diff --git a/src/process/testutil.rs b/src/process/testutil.rs
index 3f9c87f..d1fdbc5 100644
--- a/src/process/testutil.rs
+++ b/src/process/testutil.rs
@@ -47,6 +47,14 @@ impl<'t> TestContext<'t> {
 		module.handle_input(self.input_handler, self.rebase_todo_file, self.view)
 	}
 
+	pub fn handle_n_inputs(&mut self, module: &'_ mut dyn ProcessModule, n: usize) -> Vec<ProcessResult> {
+		let mut results = vec![];
+		for _ in 0..n {
+			results.push(module.handle_input(self.input_handler, self.rebase_todo_file, self.view));
+		}
+		results
+	}
+
 	pub fn handle_all_inputs(&mut self, module: &'_ mut dyn ProcessModule) -> Vec<ProcessResult> {
 		let mut results = vec![];
 		for _ in 0..self.num_inputs {