e2e tests: Add process test (#2968)

* Add process test Signed-off-by: sat0ken <[email protected]> * rm unused imports Signed-off-by: sat0ken <[email protected]> * fix err Signed-off-by: sat0ken <[email protected]> * fix set env Signed-off-by: sat0ken <[email protected]> * fix format err Signed-off-by: sat0ken <[email protected]> * add mkdir for test Signed-off-by: sat0ken <[email protected]> * set permision to /test Signed-off-by: sat0ken <[email protected]> * fix my miss of delete Signed-off-by: sat0ken <[email protected]> * set process args to run test Signed-off-by: sat0ken <[email protected]> * fix check process env Signed-off-by: sat0ken <[email protected]> * fix create cwd dir Signed-off-by: sat0ken <[email protected]> * fix err to set env value Signed-off-by: sat0ken <[email protected]> * fix err to append env value to default Signed-off-by: sat0ken <[email protected]> * delete unnecessary code Signed-off-by: sat0ken <[email protected]> * fix to compare variables Signed-off-by: sat0ken <[email protected]> * remove unused import Signed-off-by: sat0ken <[email protected]> * fix error message actual/expected val Signed-off-by: Yashodhan Joshi <[email protected]> --------- Signed-off-by: sat0ken <[email protected]> Signed-off-by: Yashodhan Joshi <[email protected]> Co-authored-by: Yashodhan Joshi <[email protected]>
youki-dev · Nov 25, 2024 · 3b78801 · 3b78801
1 parent f8a6148
commit 3b78801
Show file tree

Hide file tree

Showing 6 changed files with 90 additions and 0 deletions.
diff --git a/tests/contest/contest/src/main.rs b/tests/contest/contest/src/main.rs
@@ -21,6 +21,7 @@ use crate::tests::linux_ns_itype::get_ns_itype_tests;
 use crate::tests::mounts_recursive::get_mounts_recursive_test;
 use crate::tests::no_pivot::get_no_pivot_test;
 use crate::tests::pidfile::get_pidfile_test;
+use crate::tests::process::get_process_test;
 use crate::tests::process_oom_score_adj::get_process_oom_score_adj_test;
 use crate::tests::process_rlimits::get_process_rlimits_test;
 use crate::tests::process_user::get_process_user_test;
@@ -117,6 +118,7 @@ fn main() -> Result<()> {
     let scheduler = get_scheduler_test();
     let io_priority_test = get_io_priority_test();
     let devices = get_devices_test();
+    let process = get_process_test();
     let process_user = get_process_user_test();
     let process_rlimtis = get_process_rlimits_test();
     let no_pivot = get_no_pivot_test();
@@ -144,6 +146,7 @@ fn main() -> Result<()> {
     tm.add_test_group(Box::new(sysctl));
     tm.add_test_group(Box::new(scheduler));
     tm.add_test_group(Box::new(devices));
+    tm.add_test_group(Box::new(process));
     tm.add_test_group(Box::new(process_user));
     tm.add_test_group(Box::new(process_rlimtis));
     tm.add_test_group(Box::new(no_pivot));

diff --git a/tests/contest/contest/src/tests/mod.rs b/tests/contest/contest/src/tests/mod.rs
@@ -11,6 +11,7 @@ pub mod linux_ns_itype;
 pub mod mounts_recursive;
 pub mod no_pivot;
 pub mod pidfile;
+pub mod process;
 pub mod process_oom_score_adj;
 pub mod process_rlimits;
 pub mod process_user;

diff --git a/tests/contest/contest/src/tests/process/mod.rs b/tests/contest/contest/src/tests/process/mod.rs
@@ -0,0 +1,2 @@
+mod process_test;
+pub use process_test::get_process_test;
diff --git a/tests/contest/contest/src/tests/process/process_test.rs b/tests/contest/contest/src/tests/process/process_test.rs
@@ -0,0 +1,50 @@
+use std::fs;
+
+use anyhow::{bail, Context, Ok, Result};
+use oci_spec::runtime::{ProcessBuilder, Spec, SpecBuilder};
+use test_framework::{test_result, Test, TestGroup, TestResult};
+
+use crate::utils::test_inside_container;
+
+fn create_spec() -> Result<Spec> {
+    let mut process = ProcessBuilder::default()
+        .args(vec!["runtimetest".to_string(), "process".to_string()])
+        .cwd("/test")
+        .build()
+        .expect("error in creating process config");
+    let mut env = process.env().clone().unwrap();
+    env.push("testa=valuea".to_string());
+    env.push("testb=123".to_string());
+    process.set_env(Some(env));
+
+    let spec = SpecBuilder::default()
+        .process(process)
+        .build()
+        .context("failed to build spec")?;
+
+    Ok(spec)
+}
+
+fn process_test() -> TestResult {
+    let spec = test_result!(create_spec());
+
+    test_inside_container(spec, &|bundle| {
+        match fs::create_dir(bundle.join("test")) {
+            Result::Ok(_) => { /*This is expected*/ }
+            Err(e) => {
+                bail!(e)
+            }
+        }
+
+        Ok(())
+    })
+}
+
+pub fn get_process_test() -> TestGroup {
+    let mut process_test_group = TestGroup::new("process");
+
+    let test = Test::new("process_test", Box::new(process_test));
+    process_test_group.add(vec![Box::new(test)]);
+
+    process_test_group
+}
diff --git a/tests/contest/runtimetest/src/main.rs b/tests/contest/runtimetest/src/main.rs
@@ -44,6 +44,7 @@ fn main() {
         "io_priority_class_be" => tests::test_io_priority_class(&spec, IoprioClassBe),
         "io_priority_class_idle" => tests::test_io_priority_class(&spec, IoprioClassIdle),
         "devices" => tests::validate_devices(&spec),
+        "process" => tests::validate_process(&spec),
         "process_user" => tests::validate_process_user(&spec),
         "process_rlimits" => tests::validate_process_rlimits(&spec),
         "no_pivot" => tests::validate_rootfs(),

diff --git a/tests/contest/runtimetest/src/tests.rs b/tests/contest/runtimetest/src/tests.rs
@@ -1,3 +1,4 @@
+use std::env;
 use std::fs::{self, read_dir};
 use std::os::linux::fs::MetadataExt;
 use std::os::unix::fs::{FileTypeExt, PermissionsExt};
@@ -550,6 +551,38 @@ pub fn test_io_priority_class(spec: &Spec, io_priority_class: IOPriorityClass) {
     }
 }
 
+pub fn validate_process(spec: &Spec) {
+    let process = spec.process().as_ref().unwrap();
+    let expected_cwd = process.cwd();
+    let cwd = &getcwd().unwrap();
+
+    if expected_cwd != cwd {
+        eprintln!(
+            "error due to spec cwd want {:?}, got {:?}",
+            expected_cwd, cwd
+        )
+    }
+
+    for env_str in process.env().as_ref().unwrap().iter() {
+        match env_str.split_once("=") {
+            Some((env_key, expected_val)) => {
+                let actual_val = env::var(env_key).unwrap();
+                if actual_val != expected_val {
+                    eprintln!(
+                        "error due to spec environment value of {:?} want {:?}, got {:?}",
+                        env_key, expected_val, actual_val
+                    )
+                }
+            }
+            None => {
+                eprintln!(
+                    "spec env value is not correct : expected key=value format, got {env_str}"
+                )
+            }
+        }
+    }
+}
+
 pub fn validate_process_user(spec: &Spec) {
     let process = spec.process().as_ref().unwrap();
     let expected_uid = Uid::from(process.user().uid());
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,2 @@
		mod process_test;
		pub use process_test::get_process_test;