test(logging): test if set_logger works

aljoshare · Aug 26, 2024 · 78ef664 · 78ef664
1 parent f795b18
commit 78ef664
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/src/logging.rs b/src/logging.rs
@@ -21,10 +21,17 @@ pub fn set_logger() -> Result<LevelFilter, CantAttachLogger> {
 
 #[cfg(test)]
 mod tests {
+    use std::env;
+
     use crate::logging::set_logger;
+    use rstest::*;
 
-    #[test]
-    fn is_logger_set() {
-        assert!(set_logger());
+    #[rstest]
+    #[case("info")]
+    fn set_valid_logger(#[case] log_level: String) {
+        env::set_var("RUST_LOG", &log_level);
+        let result = set_logger();
+        assert!(result.is_ok());
+        assert_eq!(result.unwrap().to_string(), log_level)
     }
 }