address @mgrimesix

truenas · Sep 5, 2024 · 6591196 · 6591196
1 parent f1bb577
commit 6591196
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/tests/api2/test_mail.py b/tests/api2/test_mail.py
@@ -2,6 +2,7 @@
 
 
 def test_config_settings():
+    orig_config = call("mail.config")
     payload = {
         "fromemail": "[email protected]",
         "outgoingserver": "mail.ixsystems.com",
@@ -11,7 +12,11 @@ def test_config_settings():
         "smtp": True,
         "user": "[email protected]"
     }
-    call("mail.update", payload)
-    config = call("mail.config")
-    # test that payload is a subset of config
-    assert payload.items() <= config.items()
+    try:
+        call("mail.update", payload)
+        config = call("mail.config")
+        # test that payload is a subset of config
+        assert payload.items() <= config.items()
+    finally:
+        call("mail.update", {key: orig_config[key] for key in payload})
+        assert call("mail.config") == orig_config