Use mock.patch as context manager rather than as decorators

diff --git a/tests/test_command.py b/tests/test_command.py
index f04b190245e0f2c706aee6789ba2eba871f024cf..933d7c8d88ecc94c931d1489d6b508e6ef98a0f3 100644 (file)
--- a/tests/test_command.py
+++ b/tests/test_command.py
@@ -798,15 +798,16 @@ class CommandLineTest(TestBase):
                             # not too long
                             assert len(help_text) < 80
         assert not commands, "Commands without help text: %s" % commands
-    
-    @mock.patch("alembic.command.os.listdir", return_value = ["file1", "file2"])
-    @mock.patch("alembic.command.os.access", return_value = True)
-    def test_init_file_exists_and_is_not_empty(self, mocked_listdir, mocked_access):
-        directory = "alembic"
-        assert_raises_message(
-            util.CommandError,
-            "Directory %s already exists and is not empty" % directory,
-            command.init,
-            self.cfg,
-            directory = directory
-        )
+
+    def test_init_file_exists_and_is_not_empty(self):
+        with mock.patch(
+            "alembic.command.os.listdir", return_value=["file1", "file2"]
+        ), mock.patch("alembic.command.os.access", return_value=True):
+            directory = "alembic"
+            assert_raises_message(
+                util.CommandError,
+                "Directory %s already exists and is not empty" % directory,
+                command.init,
+                self.cfg,
+                directory=directory,
+            )