From b499944a2374283a949ec78daa0198bd9b5daac0 Mon Sep 17 00:00:00 2001
From: dasschnee <46967552+dasschnee@users.noreply.github.com>
Date: Fri, 16 Aug 2024 13:01:26 +0200
Subject: [PATCH] Improve test-coverage tests/helpers.py (#9624)

---
 tests/test_helpers.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/tests/test_helpers.py b/tests/test_helpers.py
index 7db5580b460..42c56dfef32 100644
--- a/tests/test_helpers.py
+++ b/tests/test_helpers.py
@@ -1,6 +1,9 @@
 from __future__ import annotations
 
+import os
+
 from tests.helpers import flatten_dict
+from tests.helpers import isolated_environment
 
 
 def test_flatten_dict() -> None:
@@ -21,3 +24,23 @@ def test_flatten_dict() -> None:
     }
 
     assert flattened_dict == flatten_dict(orig_dict, delimiter=":")
+
+
+def test_isolated_environment_restores_original_environ() -> None:
+    original_environ = dict(os.environ)
+    with isolated_environment():
+        os.environ["TEST_VAR"] = "test"
+    assert os.environ == original_environ
+
+
+def test_isolated_environment_clears_environ() -> None:
+    os.environ["TEST_VAR"] = "test"
+    with isolated_environment(clear=True):
+        assert "TEST_VAR" not in os.environ
+    assert "TEST_VAR" in os.environ
+
+
+def test_isolated_environment_updates_environ() -> None:
+    with isolated_environment(environ={"NEW_VAR": "new_value"}):
+        assert os.environ["NEW_VAR"] == "new_value"
+    assert "NEW_VAR" not in os.environ