From 452da98cbede99a7f621c7cd3fbb0fe08d0d6c6f Mon Sep 17 00:00:00 2001
From: Aravinth Manivannan <realaravinth@batsense.net>
Date: Sun, 14 Jul 2024 22:11:42 +0530
Subject: [PATCH] fix: reset env var post testing

---
 src/db/sqlx_postgres.rs | 1 -
 src/settings/mod.rs     | 4 ++++
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/db/sqlx_postgres.rs b/src/db/sqlx_postgres.rs
index 342de17..d39eb15 100644
--- a/src/db/sqlx_postgres.rs
+++ b/src/db/sqlx_postgres.rs
@@ -39,7 +39,6 @@ pub struct PostgresDatabase;
 #[async_trait::async_trait]
 impl super::create_database::CreateDatabase for PostgresDatabase {
     async fn create_database(&self, url: &url::Url) {
-        println!("Trying to connect to: {}", url.as_str());
         sqlx::Postgres::create_database(url.as_str()).await.unwrap();
     }
 }
diff --git a/src/settings/mod.rs b/src/settings/mod.rs
index 1ae7248..c4d45c1 100644
--- a/src/settings/mod.rs
+++ b/src/settings/mod.rs
@@ -130,6 +130,7 @@ pub mod tests {
     macro_rules! env_helper {
         ($init_settings:ident, $env:expr, $val:expr, $val_typed:expr, $($param:ident).+) => {
             println!("Setting env var {} to {} for test", $env, $val);
+            let current = env::var($env);
             env::set_var($env, $val);
             {
                 let new_settings = $crate::settings::Settings::new().unwrap();
@@ -137,6 +138,9 @@ pub mod tests {
                 assert_ne!(new_settings.$($param).+, $init_settings.$($param).+);
             }
             env::remove_var($env);
+            if let Ok(current) = current {
+                env::set_var($env, current);
+            }
         };