test(logging): isolate tests by setting env temporarily

aljoshare · Aug 31, 2024 · 5379766 · 5379766
1 parent e68bc9d
commit 5379766
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 4 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -34,6 +34,7 @@ time = "0.3.36"
 assert_cmd = "2.0.14"
 criterion = { version = "0.5.1", features = ["html_reports"] }
 rstest = "0.22.0"
+temp-env = "0.3.6"
 
 [[bench]]
 name = "main"

diff --git a/src/logging.rs b/src/logging.rs
@@ -33,13 +33,16 @@ mod tests {
     #[case("DEBUG")]
     #[case("TRACE")]
     fn get_valid_subscriber(#[case] log_level: String) {
-        env::set_var("RUST_LOG", &log_level);
-        assert!(get_subscriber(false).is_ok());
+        temp_env::with_var("RUST_LOG", Some(&log_level), || {
+            assert!(get_subscriber(false).is_ok());
+            env::remove_var("RUST_LOG");
+        });
     }
 
     #[rstest]
     fn err_if_level_not_set() {
-        env::remove_var("RUST_LOG");
-        assert!(get_subscriber(false).is_err());
+        temp_env::with_var("RUST_LOG", None::<&str>, || {
+            assert!(get_subscriber(false).is_err());
+        });
     }
 }