Add tests for mypy support

Author: moshemoshe137

sync_pre_commit_deps.py

@@ -56,8 +56,10 @@ def main(argv: Sequence[str] | None = None) -> int:
     for repo in loaded['repos']:
         for hook in repo['hooks']:
             if (hid := hook['id']) in SUPPORTED:
-                if hook_rev := re.match(r'[vV]?(?P<rev>[.\d]+)$', repo['rev']):
-                    versions[hid] = hook_rev.group('rev')
+                hook_rev = re.match(r'[vV]?(?P<rev>.+)$', repo['rev'])
+                # assert for mypy
+                assert hook_rev is not None, f'Invalid rev {repo["rev"]!r}'
+                versions[hid] = hook_rev.group('rev')
 
     updated = []
     for repo in loaded['repos']:

tests/sync_pre_commit_deps_test.py

@@ -25,7 +25,7 @@ def test_main_noop(tmpdir):
     assert cfg.read() == s
 
 
-def test_main_writes_both(tmpdir):
+def test_main_writes_all(tmpdir):
     cfg = tmpdir.join('.pre-commit-config.yaml')
     cfg.write(
         'repos:\n'
@@ -44,7 +44,12 @@ def test_main_writes_both(tmpdir):
         '    rev: 6.0.0\n'
         '    hooks:\n'
         '    -   id: flake8\n'
-        # all 3 below should be rewritten
+        # gives the `mypy` version
+        '-   repo: https://github.com/pre-commit/mirrors-mypy\n'
+        '    rev: v1.13.0\n'
+        '    hooks:\n'
+        '    -   id: mypy\n'
+        # all repos below should have their additional_dependencies rewritten
         '-   repo: https://github.com/asottile/yesqa\n'
         '    rev: v1.5.0\n'
         '    hooks:\n'
@@ -57,13 +62,20 @@ def test_main_writes_both(tmpdir):
         '    -   id: blacken-docs\n'
         '        additional_dependencies:\n'
         '        -   black==22.12.0\n'
+        '-   repo: https://github.com/nbQA-dev/nbQA\n'
+        '    rev: 1.9.1\n'
+        '    hooks:\n'
+        '    -   id: nbqa-mypy\n'
+        '        additional_dependencies:\n'
+        '        -   mypy==0.910\n'
         '-   repo: https://github.com/example/example\n'
         '    rev: v1.0.0\n'
         '    hooks:\n'
         '    -   id: example\n'
         '        additional_dependencies:\n'
         '        -   black==22.12.0\n'
-        '        -   flake8==5.0.0\n',
+        '        -   flake8==5.0.0\n'
+        '        -   mypy==0.123\n',
     )
 
     assert main((str(cfg),))
@@ -82,6 +94,10 @@ def test_main_writes_both(tmpdir):
         '    rev: 6.0.0\n'
         '    hooks:\n'
         '    -   id: flake8\n'
+        '-   repo: https://github.com/pre-commit/mirrors-mypy\n'
+        '    rev: v1.13.0\n'
+        '    hooks:\n'
+        '    -   id: mypy\n'
         '-   repo: https://github.com/asottile/yesqa\n'
         '    rev: v1.5.0\n'
         '    hooks:\n'
@@ -94,13 +110,20 @@ def test_main_writes_both(tmpdir):
         '    -   id: blacken-docs\n'
         '        additional_dependencies:\n'
         '        -   black==23.3.0\n'
+        '-   repo: https://github.com/nbQA-dev/nbQA\n'
+        '    rev: 1.9.1\n'
+        '    hooks:\n'
+        '    -   id: nbqa-mypy\n'
+        '        additional_dependencies:\n'
+        '        -   mypy==1.13.0\n'
         '-   repo: https://github.com/example/example\n'
         '    rev: v1.0.0\n'
         '    hooks:\n'
         '    -   id: example\n'
         '        additional_dependencies:\n'
         '        -   black==23.3.0\n'
         '        -   flake8==6.0.0\n'
+        '        -   mypy==1.13.0\n'
     )
 
 
@@ -117,15 +140,23 @@ def test_main_no_dep_on_one_and_writes_other(tmpdir):
         '    hooks:\n'
         '    -   id: yesqa\n'
         '        additional_dependencies:\n'
-        # should not be rewritten because target version can't be found
+        # should not be rewritten because target versions can't be found
         '        -   flake8==5.0.0\n'
+        '        -   mypy==0.910\n'
         '-   repo: https://github.com/adamchainz/blacken-docs\n'
         '    rev: 1.15.0\n'
         '    hooks:\n'
         '    -   id: blacken-docs\n'
         '        additional_dependencies:\n'
         # should be rewritten
-        '        -   black==22.12.0\n',
+        '        -   black==22.12.0\n'
+        '-   repo: https://github.com/nbQA-dev/nbQA\n'
+        '    rev: 1.9.1\n'
+        '    hooks:\n'
+        '    -   id: nbqa-mypy\n'
+        '        additional_dependencies:\n'
+        # should not be rewritten because target version can't be found
+        '        -   mypy==0.123',
     )
 
     assert main((str(cfg),))
@@ -142,10 +173,17 @@ def test_main_no_dep_on_one_and_writes_other(tmpdir):
         '    -   id: yesqa\n'
         '        additional_dependencies:\n'
         '        -   flake8==5.0.0\n'
+        '        -   mypy==0.910\n'
         '-   repo: https://github.com/adamchainz/blacken-docs\n'
         '    rev: 1.15.0\n'
         '    hooks:\n'
         '    -   id: blacken-docs\n'
         '        additional_dependencies:\n'
         '        -   black==23.3.0\n'
+        '-   repo: https://github.com/nbQA-dev/nbQA\n'
+        '    rev: 1.9.1\n'
+        '    hooks:\n'
+        '    -   id: nbqa-mypy\n'
+        '        additional_dependencies:\n'
+        '        -   mypy==0.123\n'
     )