add tests around pylint exit kwargs

Signed-off-by: Matthew Peveler <[email protected]>

pylint_runner/main.py

@@ -7,14 +7,14 @@
 from argparse import ArgumentParser
 import configparser
 import os
-import re
 import sys
 
 import colorama
 import pylint
 import pylint.lint
 
-from pylint_runner import __version__
+from . import __version__
+from .utils import get_pylint_kwargs
 
 PYTHON_VERSION = ".".join([str(x) for x in sys.version_info[0:3]])
 
@@ -188,14 +188,7 @@ def run(self, output=None, error=None):
             # positional arguments
             self.args.insert(0, f"--rcfile={self.rcfile}")
 
-        pylint_version = pylint.__version__.split('.')[:3]
-        pylint_version[2] = re.sub(r'[a-z-].*', '', pylint_version[2])
-        pylint_version = [int(x) for x in pylint_version]
-
-        if pylint_version < [2, 5, 1]:
-            exit_kwarg = {"do_exit": False}
-        else:
-            exit_kwarg = {"exit": False}
+        exit_kwargs = get_pylint_kwargs(pylint.__version__)
 
         run = pylint.lint.Run(self.args + pylint_files, **exit_kwarg)
         sys.stdout = savedout

pylint_runner/utils.py

@@ -0,0 +1,11 @@
+import re
+
+
+def get_pylint_kwargs(pylint_version):
+    """ Get the correct keyword arguments for the pylint runner. """
+    pylint_version = pylint_version.split('.')[:3]
+    pylint_version[2] = re.sub(r'[a-z-].*', '', pylint_version[2])
+    pylint_version = [int(x) for x in pylint_version]
+    if pylint_version < [2, 5, 1]:
+        return {"do_exit": False}
+    return {"exit": False}