Adjust tests to with statement

ctrlaltcoop · Jun 20, 2024 · 0cef292 · 0cef292
1 parent 8a4e674
commit 0cef292
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/certbot_dns_hetzner/dns_hetzner_test.py b/certbot_dns_hetzner/dns_hetzner_test.py
@@ -43,9 +43,16 @@ def setUp(self):
         self.auth = Authenticator(self.config, "hetzner")
 
         self.mock_client = mock.MagicMock()
+
+        mock_client_wrapper = mock.MagicMock()
+        mock_client_wrapper.__enter__ = mock.MagicMock(
+            return_value=self.mock_client
+        )
+
         # _get_ispconfig_client | pylint: disable=protected-access
         self.auth._get_hetzner_client = mock.MagicMock(
-            return_value=self.mock_client)
+            return_value=mock_client_wrapper
+        )
 
     @patch_display_util()
     def test_perform(self, _unused_mock_get_utility):