mirrors/envision
1
0
Fork 0
mirror of https://gitlab.com/gabmus/envision.git synced 2025-04-21 20:14:50 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: forbid reusing existing Runner

Browse source
This commit is contained in:
Gabriele Musco 2023-06-01 19:29:22 +02:00
parent 9880b8e4e7
commit 466eb70078
1 changed files with 23 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
src/runner.rs
View file

@ -32,15 +32,20 @@ impl Runner {
}
pub fn start(&mut self) {
self.process = Some(
Command::new(&self.command)
.args(&self.args)
.envs(self.environment.clone())
.stdout(Stdio::piped())
.stderr(Stdio::piped())
.spawn()
.expect("Failed to execute runner"),
);
match self.process {
Some(_) => panic!("Cannot reuse existing Runner"),
None => {
self.process = Some(
Command::new(&self.command)
.args(&self.args)
.envs(self.environment.clone())
.stdout(Stdio::piped())
.stderr(Stdio::piped())
.spawn()
.expect("Failed to execute runner"),
);
}
};
}
pub fn terminate(&mut self) {
@ -51,7 +56,6 @@ impl Runner {
Some(proc) => {
proc.kill().expect("Failed to kill process");
proc.wait().expect("Failed to wait for process");
self.process = None
}
}
}
@ -109,7 +113,11 @@ mod tests {
fn can_run_command_and_read_env() {
let mut env = HashMap::new();
env.insert("REX2TEST".to_string(), "Lorem ipsum dolor".to_string());
let mut runner = Runner::new(env, "bash".into(), vec!["-c".into(), "echo \"REX2TEST: $REX2TEST\"".into()]);
let mut runner = Runner::new(
env,
"bash".into(),
vec!["-c".into(), "echo \"REX2TEST: $REX2TEST\"".into()],
);
runner.start();
while runner.status() == RunnerStatus::Running {
sleep(time::Duration::from_millis(10));
@ -117,6 +125,9 @@ mod tests {
runner.flush_out();
assert_eq!(runner.status(), RunnerStatus::Stopped);
assert_eq!(runner.stdout.len(), 2);
assert_eq!(runner.stdout.get(0).unwrap(), "REX2TEST: Lorem ipsum dolor\n");
assert_eq!(
runner.stdout.get(0).unwrap(),
"REX2TEST: Lorem ipsum dolor\n"
);
}
}