iotests/297: split test into sub-cases

Take iotest 297's main() test function and split it into two sub-cases
that can be skipped individually. We can also drop custom environment
setup from the pylint test as it isn't needed.

Signed-off-by: John Snow <jsnow@redhat.com>
Reviewed-by: Hanna Reitz <hreitz@redhat.com>
Message-id: 20211019144918.3159078-11-jsnow@redhat.com
Signed-off-by: John Snow <jsnow@redhat.com>

tests/qemu-iotests/297

@@ -82,36 +82,51 @@ def run_linter(
     )
 
 
-def main() -> None:
+def check_linter(linter: str) -> bool:
-    for linter in ('pylint', 'mypy'):
+    try:
-        try:
+        run_linter(linter, ['--version'], suppress_output=True)
-            run_linter(linter, ['--version'], suppress_output=True)
+    except subprocess.CalledProcessError:
-        except subprocess.CalledProcessError:
+        iotests.case_notrun(f"'{linter}' not found")
-            iotests.notrun(f"'{linter}' not found")
+        return False
+    return True
 
+
+def test_pylint(files: List[str]) -> None:
+    print('=== pylint ===')
+    sys.stdout.flush()
+
+    if not check_linter('pylint'):
+        return
+
+    run_linter('pylint', files)
+
+
+def test_mypy(files: List[str]) -> None:
+    print('=== mypy ===')
+    sys.stdout.flush()
+
+    if not check_linter('mypy'):
+        return
+
+    env = os.environ.copy()
+    env['MYPYPATH'] = env['PYTHONPATH']
+
+    run_linter('mypy', files, env=env, suppress_output=True)
+
+
+def main() -> None:
     files = get_test_files()
 
     iotests.logger.debug('Files to be checked:')
     iotests.logger.debug(', '.join(sorted(files)))
 
-    env = os.environ.copy()
+    for test in (test_pylint, test_mypy):
-    env['MYPYPATH'] = env['PYTHONPATH']
+        try:
-
+            test(files)
-    print('=== pylint ===')
+        except subprocess.CalledProcessError as exc:
-    sys.stdout.flush()
+            # Linter failure will be caught by diffing the IO.
-    try:
+            if exc.output:
-        run_linter('pylint', files, env=env)
+                print(exc.output)
-    except subprocess.CalledProcessError:
-        # pylint failure will be caught by diffing the IO.
-        pass
-
-    print('=== mypy ===')
-    sys.stdout.flush()
-    try:
-        run_linter('mypy', files, env=env, suppress_output=True)
-    except subprocess.CalledProcessError as exc:
-        if exc.output:
-            print(exc.output)
 
 
 iotests.script_main(main)