test: add test for root.rs

Young-TW · Nov 5, 2024 · 93b3aa5 · 93b3aa5
1 parent e4ab036
commit 93b3aa5
Showing 1 changed file with 40 additions and 0 deletions.
diff --git a/src/root.rs b/src/root.rs
@@ -10,3 +10,43 @@ pub fn get_sudo(manager: String) -> std::process::Command {
 
     command
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_get_sudo() {
+        let manager = "apt".to_string();
+        let cmd = get_sudo(manager.clone());
+        assert_eq!(cmd.get_program(), "sudo");
+
+        let manager = "dnf".to_string();
+        let cmd = get_sudo(manager.clone());
+        assert_eq!(cmd.get_program(), "sudo");
+
+        let manager = "yum".to_string();
+        let cmd = get_sudo(manager.clone());
+        assert_eq!(cmd.get_program(), "sudo");
+
+        let manager = "pacman".to_string();
+        let cmd = get_sudo(manager.clone());
+        assert_eq!(cmd.get_program(), "sudo");
+
+        let manager = "zypper".to_string();
+        let cmd = get_sudo(manager.clone());
+        assert_eq!(cmd.get_program(), "sudo");
+
+        let manager = "apk".to_string();
+        let cmd = get_sudo(manager.clone());
+        assert_eq!(cmd.get_program(), "sudo");
+
+        let manager = "portage".to_string();
+        let cmd = get_sudo(manager.clone());
+        assert_eq!(cmd.get_program(), "sudo");
+
+        let manager = "brew".to_string();
+        let cmd = get_sudo(manager.clone());
+        assert_eq!(cmd.get_program(), "brew");
+    }
+}