Move to unittest.mock.patch

Author: cmaloney

Lib/test/test_netrc.py

@@ -2,7 +2,7 @@
 from pathlib import Path
 from test import support
 from test.support import os_helper
-from unittest.mock import Mock
+from unittest.mock import patch
 
 
 temp_filename = os_helper.TESTFN
@@ -318,22 +318,19 @@ def test_security(self):
     def test_security_only_once(self):
         # Make sure security check is only run once per parse when multiple
         # entries are found.
-        check_called = netrc.netrc._security_check = Mock(return_value=True)
-
-        # Parse a default netrc with more than one password line.
-        with os_helper.temp_dir() as tmp_dir:
-            netrc_path = Path(tmp_dir) / '.netrc'
-            netrc_path.write_text("""\
-            machine foo.domain.com login bar password pass
-            machine bar.domain.com login foo password pass
-            """)
-            netrc_path.chmod(0o600)
-            with os_helper.EnvironmentVarGuard() as environ:
-                environ.set('HOME', tmp_dir)
-                netrc.netrc()
+        with patch.object(netrc.netrc, "_security_check") as mock:
+            with os_helper.temp_dir() as tmp_dir:
+                netrc_path = Path(tmp_dir) / '.netrc'
+                netrc_path.write_text("""\
+                machine foo.domain.com login bar password pass
+                machine bar.domain.com login foo password pass
+                """)
+                netrc_path.chmod(0o600)
+                with os_helper.EnvironmentVarGuard() as environ:
+                    environ.set('HOME', tmp_dir)
+                    netrc.netrc()
 
-        check_called.assert_called_once()
-        del check_called
+            mock.assert_called_once()
 
 
 if __name__ == "__main__":