Update src/pytest_plone/fixtures/env.py

Co-authored-by: David Glick <david@glicksoftware.com>
collective · Dec 5, 2023 · d6de0ce · d6de0ce
1 parent f7007e3
commit d6de0ce
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/src/pytest_plone/fixtures/env.py b/src/pytest_plone/fixtures/env.py
@@ -13,9 +13,11 @@ def generate_mo():
     key = "zope_i18n_compile_mo_files"
     current_value = os.getenv(key, None)
     os.environ[key] = "1"
-    yield
-    # Revert to previous state
-    if current_value is None:
-        os.environ.pop(key)
-    else:
-        os.environ[key] = current_value
+    try:
+        yield
+    finally:
+        # Revert to previous state
+        if current_value is None:
+            os.environ.pop(key)
+        else:
+            os.environ[key] = current_value