Recognize pytest (#63)

playbooks/test-integration.yml

@@ -15,6 +15,8 @@
         url: https://github.com/cookiecutter/cookiecutter.git
       - name: typeshed
         url: https://github.com/python/typeshed.git
+      - name: flask-babel
+        url: https://github.com/python-babel/flask-babel.git
   tasks:
 
     - name: Clone repo

samples/integration/ansible-lint.txt

@@ -19,6 +19,7 @@ py38-devel
 py39-ansible29
 py39-core
 py39-devel
+test
 test-requirements
 test-setup
 uninstall

samples/integration/flask-babel.txt

@@ -0,0 +1,5 @@
+build
+install
+test
+uninstall
+up

setup.cfg

@@ -66,6 +66,7 @@ mk_tools =
     make=mk.tools.make:MakeTool
     npm=mk.tools.npm:NpmTool
     pypackage=mk.tools.py_package:PyPackageTool
+    pytest=mk.tools.pytest:PyTestTool
     pre-commit=mk.tools.pre_commit:PreCommitTool
     shell=mk.tools.shell:ShellTool
     tox=mk.tools.tox:ToxTool

src/mk/tools/git.py

@@ -77,8 +77,9 @@ def up(self):
             result = run_or_fail(["gh", "pr", "list", "-S", f"head:{repo.active_branch}"])
             if result.returncode == 0:
                 pr_list = []
-                for line in result.stdout.splitlines():
-                    pr_list.append(line.split("\t")[0])
+                if result.stdout:
+                    for line in result.stdout.splitlines():
+                        pr_list.append(line.split("\t")[0])
                 if len(pr_list) == 0:
                     logging.debug("Existing PR not found, creating one.")
                     commit = repo.head.commit

src/mk/tools/pytest.py

@@ -0,0 +1,40 @@
+import os
+import sys
+from typing import List, Optional
+
+from mk.exec import run_or_fail
+from mk.tools import Action, Tool
+
+
+class PyTestTool(Tool):
+    """Expose test command if pytest config is detected."""
+
+    name = "pytest"
+
+    def __init__(self) -> None:
+        super().__init__(self)
+
+    def run(self, action: Optional[Action] = None) -> None:
+        if not action:
+            return
+        if action.name == "test":
+            cmd = [sys.executable, "-m", "pytest"]
+            run_or_fail(cmd, tee=True)
+
+    def is_present(self, path: str) -> bool:
+        for name in ("pytest.ini",):
+            if os.path.isfile(name):
+                return True
+        return False
+
+    def actions(self) -> List[Action]:
+        actions = []
+        actions.append(
+            Action(
+                name="test",
+                tool=self,
+                description="Run pytest",
+            )
+        )
+
+        return actions