Add cleanup to _fail_for_invalid_template_variable fixture

pytest-dev · Oct 26, 2023 · c9c29aa · c9c29aa
1 parent 787ceef
commit c9c29aa
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/pytest_django/plugin.py b/pytest_django/plugin.py
@@ -628,14 +628,17 @@ def __mod__(self, var: str) -> str:
             else:
                 return msg
 
-    if (
-        os.environ.get(INVALID_TEMPLATE_VARS_ENV, "false") == "true"
-        and django_settings_is_configured()
-    ):
-        from django.conf import settings as dj_settings
+    with pytest.MonkeyPatch.context() as mp:
+        if (
+            os.environ.get(INVALID_TEMPLATE_VARS_ENV, "false") == "true"
+            and django_settings_is_configured()
+        ):
+            from django.conf import settings as dj_settings
+
+            if dj_settings.TEMPLATES:
+                mp.setitem(dj_settings.TEMPLATES[0]["OPTIONS"], "string_if_invalid", InvalidVarException())
+        yield
 
-        if dj_settings.TEMPLATES:
-            dj_settings.TEMPLATES[0]["OPTIONS"]["string_if_invalid"] = InvalidVarException()
 
 
 @pytest.fixture(autouse=True)