Merge pull request #80 from indrat/add-mock-backport

Add backport for mock

Author: netromdk

tests/backports.py

@@ -14,6 +14,7 @@ def test_modules(self):
       "faulthandler",
       "importlib",
       "ipaddress",
+      "mock",
       "statistics",
       "typing",
     ), Backports.modules())
@@ -32,5 +33,6 @@ def test_str(self):
    faulthandler - https://pypi.org/project/faulthandler/
    importlib    - https://pypi.org/project/importlib/
    ipaddress    - https://pypi.org/project/ipaddress/
+   mock         - https://pypi.org/project/mock/
    statistics   - https://pypi.org/project/statistics/
    typing       - https://pypi.org/project/typing/""", Backports.str(3))

tests/class.py

@@ -102,6 +102,8 @@ def test_TextTestResult_of_unittest(self):
 
   def test_AsyncMock_of_unittest_mock(self):
     self.assertOnlyIn((3, 8), self.detect("from unittest.mock import AsyncMock"))
+    self.assertTrue(self.config.add_backport("mock"))
+    self.assertOnlyIn(((3, 6)), self.detect("from unittest.mock import AsyncMock"))
 
   def test_DefragResult_of_urllib_parse(self):
     self.assertOnlyIn((3, 2), self.detect("from urllib.parse import DefragResult"))

tests/constants.py

@@ -3152,9 +3152,13 @@ def test_TYPE_IGNORE_of_token(self):
 
   def test_args_of_unittest_mock_Mock_call_args(self):
     self.assertOnlyIn((3, 8), self.detect("from unittest.mock.Mock.call_args import args"))
+    self.assertTrue(self.config.add_backport("mock"))
+    self.assertOnlyIn((3, 6), self.detect("from unittest.mock.Mock.call_args import args"))
 
   def test_kwargs_of_unittest_mock_Mock_call_args(self):
     self.assertOnlyIn((3, 8), self.detect("from unittest.mock.Mock.call_args import kwargs"))
+    self.assertTrue(self.config.add_backport("mock"))
+    self.assertOnlyIn((3, 6), self.detect("from unittest.mock.Mock.call_args import kwargs"))
 
   def test_status_of_urllib_response_addinfourl(self):
     self.assertOnlyIn((3, 9), self.detect("from urllib.response.addinfourl import status"))

tests/function.py

@@ -1815,6 +1815,8 @@ def test_import_module_from_test_support(self):
 
   def test_seal_from_unittest_mock(self):
     self.assertOnlyIn((3, 7), self.detect("import unittest.mock\nunittest.mock.seal()"))
+    self.assertTrue(self.config.add_backport("mock"))
+    self.assertOnlyIn(((3, 6)), self.detect("import unittest.mock\nunittest.mock.seal()"))
 
   def test_quoteattr_from_xml_sax_saxutils(self):
     self.assertOnlyIn(((2, 2), (3, 0)), self.detect(

vermin/backports.py

@@ -10,6 +10,7 @@
   ("faulthandler", ["https://pypi.org/project/faulthandler/"]),
   ("importlib", ["https://pypi.org/project/importlib/"]),
   ("ipaddress", ["https://pypi.org/project/ipaddress/"]),
+  ("mock", ["https://pypi.org/project/mock/"]),
   ("statistics", ["https://pypi.org/project/statistics/"]),
   ("typing", ["https://pypi.org/project/typing/"]),
 )

vermin/rules.py

@@ -348,7 +348,7 @@ def MOD_MEM_REQS(config):
     "typing.TypedDict": (None, (3, 8)) if not bp("typing", config) else ((2, 7), (3, 8)),
     "unittest.IsolatedAsyncioTestCase": (None, (3, 8)),
     "unittest.TextTestResult": ((2, 7), (3, 2)),
-    "unittest.mock.AsyncMock": (None, (3, 8)),
+    "unittest.mock.AsyncMock": (None, (3, 8)) if not bp("mock", config) else (None, (3, 6)),
     "urllib.parse.DefragResult": (None, (3, 2)),
     "urllib.parse.DefragResultBytes": (None, (3, 2)),
     "urllib.parse.ParseResultBytes": (None, (3, 2)),
@@ -1597,7 +1597,7 @@ def MOD_MEM_REQS(config):
     "unittest.mock.Mock.assert_called": (None, (3, 6)),
     "unittest.mock.Mock.assert_called_once": (None, (3, 6)),
     "unittest.mock.Mock.assert_not_called": (None, (3, 5)),
-    "unittest.mock.seal": (None, (3, 7)),
+    "unittest.mock.seal": (None, (3, 7)) if not bp("mock", config) else (None, (3, 6)),
     "unittest.registerResult": ((2, 7), (3, 2)),
     "unittest.removeHandler": ((2, 7), (3, 2)),
     "unittest.removeResult": ((2, 7), (3, 2)),
@@ -2478,8 +2478,10 @@ def MOD_MEM_REQS(config):
     "unittest.TestResult.failfast": ((2, 7), (3, 2)),
     "unittest.TestResult.skipped": ((2, 7), (3, 1)),
     "unittest.TestResult.tb_locals": (None, (3, 5)),
-    "unittest.mock.Mock.call_args.args": (None, (3, 8)),
-    "unittest.mock.Mock.call_args.kwargs": (None, (3, 8)),
+    "unittest.mock.Mock.call_args.args": (None, (3, 8))
+    if not bp("mock", config) else (None, (3, 6)),
+    "unittest.mock.Mock.call_args.kwargs": (None, (3, 8))
+    if not bp("mock", config) else (None, (3, 6)),
     "urllib.error.HTTPError.headers": (None, (3, 4)),
     "urllib.parse._UNSAFE_URL_BYTES_TO_REMOVE": (None, (3, 10)),
     "urllib.request.Request.method": (None, (3, 3)),