Merge pull request #2196 from mgorny/called-once

Fix assertions in `test_git_hook`
PyCQA · Dec 4, 2023 · 96e2998 · 96e2998
2 parents 0118e90 + abfb91f
commit 96e2998
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/unit/test_hooks.py b/tests/unit/test_hooks.py
@@ -11,7 +11,7 @@ def test_git_hook(src_dir):
     # Ensure correct subprocess command is called
     with patch("subprocess.run", MagicMock()) as run_mock:
         hooks.git_hook()
-        assert run_mock.called_once()
+        run_mock.assert_called_once()
         assert run_mock.call_args[0][0] == [
             "git",
             "diff-index",
@@ -21,8 +21,9 @@ def test_git_hook(src_dir):
             "HEAD",
         ]
 
+    with patch("subprocess.run", MagicMock()) as run_mock:
         hooks.git_hook(lazy=True)
-        assert run_mock.called_once()
+        run_mock.assert_called_once()
         assert run_mock.call_args[0][0] == [
             "git",
             "diff-index",