From 108cc7dfdfee028974f2d07f188d3bb56015945a Mon Sep 17 00:00:00 2001
From: WoodenMaiden <yann.pomie@laposte.net>
Date: Mon, 30 Oct 2023 14:09:40 +0100
Subject: [PATCH] test: add more tests on process spawn

Signed-off-by: WoodenMaiden <yann.pomie@laposte.net>
---
 agent/lib/src/runner_engine/service.rs | 40 ++++++++++++++++++++++----
 1 file changed, 34 insertions(+), 6 deletions(-)

diff --git a/agent/lib/src/runner_engine/service.rs b/agent/lib/src/runner_engine/service.rs
index 64eaf55..b4ec073 100644
--- a/agent/lib/src/runner_engine/service.rs
+++ b/agent/lib/src/runner_engine/service.rs
@@ -223,16 +223,38 @@ mod tests {
         string
     }
 
+    #[test]
+    fn run_one_works_with_ouputs_and_code() {
+        let res = RunnerEngine::new(ExecuteRequest { 
+            id: "".to_string(),
+            files: vec![], 
+            steps: vec![] 
+        }).run_one("echo -n 'This is stdout' && echo -n 'This is stderr' >&2 && exit 1");   
+
+        assert!(res.is_ok());
+
+        let code_return = res.unwrap();
+
+        assert_eq!(code_return.stdout, "This is stdout");
+        assert_eq!(code_return.stderr, "This is stderr");
+        assert_eq!(code_return.exit_code, 1);
+    }
+
     /// Test the creation of a file
     #[test]
     fn workload_runs_correctly() {
         let files: Vec<FileModel> = Vec::new();
-        let mut steps: Vec<ExecuteRequestStep> = Vec::new();
-        let step = ExecuteRequestStep {
-            command: "echo 'This is stdout' && echo 'This is stderr' >&2".to_string(),
-            enable_output: true,
-        };
-        steps.push(step);
+        let steps: Vec<ExecuteRequestStep> = vec![
+            ExecuteRequestStep {
+                command: "echo 'This is stdout' && echo 'This is stderr' >&2".to_string(),
+                enable_output: true,
+            },
+            ExecuteRequestStep {
+                command: "echo 'This is stdout' && echo 'This is stderr' >&2 && exit 1".to_string(),
+                enable_output: false,
+            },
+        ];
+
         let request_data = ExecuteRequest {
             id: "4bf68974-c315-4c41-aee2-3dc2920e76e9".to_string(),
             files,
@@ -246,6 +268,12 @@ mod tests {
         assert_eq!(res.steps[0].exit_code, 0);
         assert_eq!(res.steps[0].stderr, "This is stderr\n");
         assert_eq!(res.steps[0].stdout, "This is stdout\n");
+
+        println!("{:?}", res.steps[1]);
+        assert_eq!(res.steps[1].exit_code, 1);
+        assert_eq!(res.steps[1].stderr, "This is stderr\n");
+        assert!(res.steps[1].stdout.is_empty());
+        
         assert_eq!(res.id, "4bf68974-c315-4c41-aee2-3dc2920e76e9");
     }