packages/sysmaster
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
sysmaster/backport-test-devmaster-add-UT-for-execute-unit.patch

276 lines
11 KiB
Diff
Raw Normal View History

sync patches from upstream
2023-12-07 00:19:38 +08:00
From ac6f7187a7e696e1a90210cbcfb6c649385616ac Mon Sep 17 00:00:00 2001
From: chenjiayi <chenjiayi22@huawei.com>
Date: Thu, 23 Nov 2023 20:17:49 +0800
Subject: [PATCH 084/103] test(devmaster): add UT for execute unit
---
exts/devmaster/src/lib/rules/exec_unit.rs | 255 ++++++++++++++++++++++
1 file changed, 255 insertions(+)
diff --git a/exts/devmaster/src/lib/rules/exec_unit.rs b/exts/devmaster/src/lib/rules/exec_unit.rs
index 828fe1af..93684659 100644
--- a/exts/devmaster/src/lib/rules/exec_unit.rs
+++ b/exts/devmaster/src/lib/rules/exec_unit.rs
@@ -987,3 +987,258 @@ fn get_subst_type(
*idx = idx_b;
Ok(Some((subst, attr)))
}
+
+#[cfg(test)]
+mod test {
+ use super::*;
+ use crate::rules::rules_load::tests::create_tmp_file;
+ use device::utils::*;
+ use nix::sys::stat::{major, minor};
+ use std::cell::RefCell;
+ use std::fs::remove_dir_all;
+ use std::path::Path;
+ use std::rc::Rc;
+
+ #[test]
+ fn test_update_devnode() {
+ if let Err(e) = LoopDev::inner_process(
+ "/tmp/test_update_devnode_tmpfile",
+ 1024 * 1024 * 10,
+ |dev| {
+ let dev = Rc::new(RefCell::new(dev.shallow_clone().unwrap()));
+ let id = dev.borrow().get_device_id().unwrap();
+ let devnum = dev.borrow().get_devnum().unwrap();
+ let major_minor = format!("{}:{}", major(devnum), minor(devnum));
+
+ create_tmp_file("/tmp/test_update_devnode/data", &id, "", true);
+
+ dev.borrow().sealed.replace(true);
+ dev.borrow().set_base_path("/tmp/test_update_devnode");
+ dev.borrow().set_devgid("1").unwrap();
+ dev.borrow().set_devuid("1").unwrap();
+ dev.borrow().set_devmode("666").unwrap();
+ dev.borrow().add_devlink("test_update_devnode/bbb").unwrap();
+
+ let mut unit = ExecuteUnitData::new(dev.clone());
+ unit.clone_device_db().unwrap();
+
+ unit.update_devnode(&HashMap::new()).unwrap();
+ /* Record the devlink in db */
+ dev.borrow().update_db().unwrap();
+
+ let p = Path::new("/dev/test_update_devnode/bbb");
+ let p_block_s = format!("/dev/block/{}", major_minor);
+ let prior_p_s = format!("/run/devmaster/links/test_update_devnode\\x2fbbb/{}", id);
+ let prior_p = Path::new(&prior_p_s);
+ assert!(p.exists());
+ assert!(Path::new(&p_block_s).exists());
+ assert!(p
+ .canonicalize()
+ .unwrap()
+ .ends_with(&dev.borrow().get_sysname().unwrap()));
+ let _ = prior_p.symlink_metadata().unwrap(); // Test symlink exists.
+
+ let new_dev = Rc::new(RefCell::new(Device::from_device_id(&id).unwrap()));
+ new_dev.borrow().sealed.replace(true);
+ new_dev.borrow().set_base_path("/tmp/test_update_devnode");
+ new_dev.borrow().set_devgid("0").unwrap();
+ new_dev.borrow().set_devuid("0").unwrap();
+ new_dev.borrow().set_devmode("600").unwrap();
+
+ let mut unit = ExecuteUnitData::new(new_dev);
+ /* See the devlink in db, but it is absent in the current device object.
+ *
+ * Then update_devnode method will remove the devlink.
+ */
+ unit.clone_device_db().unwrap();
+ unit.update_devnode(&HashMap::new()).unwrap();
+
+ assert!(!Path::new("/dev/test_update_devnode/bbb").exists());
+ assert!(Path::new(&p_block_s).exists());
+ let _ = prior_p.symlink_metadata().unwrap_err(); // Test symlink does not exists.
+
+ remove_dir_all("/tmp/test_update_devnode").unwrap();
+
+ /* Non-block devices do not have device nodes, thus update_devnode method will do nothing. */
+ let lo = Rc::new(RefCell::new(
+ Device::from_subsystem_sysname("net", "lo").unwrap(),
+ ));
+
+ let mut unit = ExecuteUnitData::new(lo);
+ unit.update_devnode(&HashMap::new()).unwrap();
+
+ /* Cover error paths when uid, gid or mode is not set. */
+ let dev = Rc::new(RefCell::new(Device::from_device_id(&id).unwrap()));
+ dev.borrow().sealed.replace(true);
+ dev.borrow().add_devlink("test_update_devnode/xxx").unwrap();
+
+ let mut unit = ExecuteUnitData::new(dev.clone());
+ unit.clone_device_db().unwrap();
+ unit.update_devnode(&HashMap::new()).unwrap();
+
+ let p = Path::new("/dev/test_update_devnode/xxx");
+ let prior_p_s = format!("/run/devmaster/links/test_update_devnode\\x2fxxx/{}", id);
+ assert!(p.exists());
+ assert!(Path::new(&p_block_s).exists());
+ assert!(p
+ .canonicalize()
+ .unwrap()
+ .ends_with(&dev.borrow().get_sysname().unwrap()));
+ let _ = Path::new(&prior_p_s).symlink_metadata().unwrap(); // Test symlink exists.
+
+ cleanup_node(dev).unwrap();
+
+ assert!(!p.exists());
+ assert!(!Path::new(&p_block_s).exists());
+ let _ = Path::new(&prior_p_s).symlink_metadata().unwrap_err(); // Test symlink exists.
+
+ Ok(())
+ },
+ ) {
+ assert!(e.is_errno(nix::Error::EACCES) || e.is_errno(nix::Error::EBUSY));
+ }
+ }
+
+ #[test]
+ fn test_subst_format() {
+ if let Err(e) =
+ LoopDev::inner_process("/tmp/test_subst_format_tmpfile", 1024 * 1024 * 10, |dev| {
+ let dev = Rc::new(RefCell::new(dev.shallow_clone().unwrap()));
+ let mut unit = ExecuteUnitData::new(dev.clone());
+ let devnum = dev.borrow().get_devnum().unwrap();
+ let major_minor = format!("{}:{}", major(devnum), minor(devnum));
+ let sysname = dev.borrow().get_sysname().unwrap();
+
+ assert_eq!(
+ unit.subst_format(FormatSubstitutionType::Attr, Some("dev".to_string()))
+ .unwrap(),
+ major_minor
+ );
+
+ assert_eq!(
+ unit.subst_format(
+ FormatSubstitutionType::Attr,
+ Some(format!("[block/{}]dev", sysname))
+ )
+ .unwrap(),
+ major_minor
+ );
+
+ unit.set_parent(Some(Rc::new(RefCell::new(
+ Device::from_subsystem_sysname("net", "lo").unwrap(),
+ ))));
+
+ /* Get the sysattr of parent device set in unit. */
+ assert_eq!(
+ unit.subst_format(FormatSubstitutionType::Attr, Some("ifindex".to_string()))
+ .unwrap(),
+ "1".to_string()
+ );
+
+ /* Invalid sysattr will be replaced with empty string. */
+ assert!(unit
+ .subst_format(
+ FormatSubstitutionType::Attr,
+ Some("asdfasdfads".to_string())
+ )
+ .unwrap()
+ .is_empty());
+
+ dev.borrow().add_property("hello", "world").unwrap();
+
+ assert_eq!(
+ unit.subst_format(FormatSubstitutionType::Env, Some("hello".to_string()))
+ .unwrap(),
+ "world".to_string()
+ );
+
+ assert!(unit
+ .subst_format(FormatSubstitutionType::Env, Some("asdfgasd".to_string()))
+ .unwrap()
+ .is_empty());
+
+ assert!(unit
+ .subst_format(FormatSubstitutionType::Driver, None)
+ .unwrap()
+ .is_empty());
+
+ assert_eq!(
+ unit.subst_format(FormatSubstitutionType::Id, None).unwrap(),
+ "lo".to_string()
+ );
+
+ let major = unit
+ .subst_format(FormatSubstitutionType::Major, None)
+ .unwrap();
+ let minor = unit
+ .subst_format(FormatSubstitutionType::Minor, None)
+ .unwrap();
+ assert_eq!(format!("{}:{}", major, minor), major_minor);
+
+ unit.program_result = "hello world test".to_string();
+ assert_eq!(
+ unit.subst_format(FormatSubstitutionType::Result, None)
+ .unwrap(),
+ "hello world test".to_string()
+ );
+ assert_eq!(
+ unit.subst_format(FormatSubstitutionType::Result, Some("0".to_string()))
+ .unwrap(),
+ "hello".to_string()
+ );
+ assert_eq!(
+ unit.subst_format(FormatSubstitutionType::Result, Some("1+".to_string()))
+ .unwrap(),
+ "world test".to_string()
+ );
+ unit.subst_format(FormatSubstitutionType::Result, Some("x".to_string()))
+ .unwrap_err();
+ unit.subst_format(FormatSubstitutionType::Result, Some("x+".to_string()))
+ .unwrap_err();
+
+ assert!(unit
+ .subst_format(FormatSubstitutionType::Result, Some("3+".to_string()))
+ .unwrap()
+ .is_empty());
+
+ assert_eq!(
+ unit.subst_format(FormatSubstitutionType::Name, None)
+ .unwrap(),
+ sysname
+ );
+
+ unit.name = "test".to_string();
+ assert_eq!(
+ unit.subst_format(FormatSubstitutionType::Name, None)
+ .unwrap(),
+ "test".to_string(),
+ );
+
+ dev.borrow().add_devlink("test").unwrap();
+ dev.borrow().sealed.replace(true);
+ assert_eq!(
+ unit.subst_format(FormatSubstitutionType::Links, None)
+ .unwrap(),
+ "test".to_string(),
+ );
+
+ unit.subst_format(FormatSubstitutionType::Invalid, None)
+ .unwrap_err();
+
+ Ok(())
+ })
+ {
+ assert!(e.is_errno(nix::Error::EACCES) || e.is_errno(nix::Error::EBUSY));
+ }
+
+ let dev = Rc::new(RefCell::new(
+ Device::from_subsystem_sysname("net", "lo").unwrap(),
+ ));
+ let unit = ExecuteUnitData::new(dev);
+ assert_eq!(
+ unit.subst_format(FormatSubstitutionType::Name, None)
+ .unwrap(),
+ "lo".to_string()
+ );
+ }
+}
--
2.33.0
Reference in New Issue Copy Permalink