Add a separate driver for Python

Author: setton

testsuite/drivers/basic.py

@@ -1,8 +1,6 @@
 import glob
 import os
-import sys
 
-from e3.os.process import Run, PIPE
 from e3.testsuite.result import TestStatus
 
 from drivers import ALSTestDriver

testsuite/drivers/python_driver.py

@@ -0,0 +1,55 @@
+import os
+
+from e3.testsuite.result import TestStatus
+
+from drivers import ALSTestDriver
+
+import inspect
+import importlib.util
+
+from drivers.lsp_python_driver import run_simple_test
+
+
+class PythonTestDriver(ALSTestDriver):
+    """Each test should have:
+    - a test.yaml containing
+          title: '<test name>'
+
+    - a number of test drivers, in .json files.
+    """
+
+    def run(self, previous_values, slot):
+        # Check whether the test should be skipped
+        if self.should_skip():
+            return False
+
+        # The working directory
+        wd = self.test_env["working_dir"]
+
+        status = TestStatus.PASS
+
+        # If there is a "test.py", evaluate it
+        if os.path.exists(os.path.join(wd, "test.py")):
+            # Load test.py as a module
+            python_file = os.path.join(wd, "test.py")
+            spec = importlib.util.spec_from_file_location("module.name", python_file)
+            module = importlib.util.module_from_spec(spec)
+            spec.loader.exec_module(module)
+
+            # Look for functions with the decorator @simple_test and run them
+            errors = [
+                f"no function with @simple_test or @complex_test found in {python_file}"
+            ]
+
+            for _, obj in inspect.getmembers(module):
+                if inspect.isfunction(obj) and (
+                    hasattr(obj, "simple_test") or hasattr(obj, "complex_test")
+                ):
+                    errors = obj(wd)
+
+            if len(errors) > 0:
+                self.result.log = "\n".join(errors)
+                status = TestStatus.FAIL
+
+            self.result.set_status(status)
+            self.push_result()

testsuite/testsuite.py

@@ -9,6 +9,7 @@
 from drivers.basic import JsonTestDriver
 from drivers.gnatcov import GNATcov
 from drivers.shell import ShellTestDriver
+from drivers.python_driver import PythonTestDriver
 from e3.testsuite import Testsuite
 
 VALGRIND_OPTIONS = [
@@ -126,7 +127,7 @@ def tear_down(self):
 
     @property
     def test_driver_map(self):
-        return {"ada_lsp": JsonTestDriver, "shell": ShellTestDriver}
+        return {"ada_lsp": JsonTestDriver, "shell": ShellTestDriver, "python": PythonTestDriver}
 
     @property
     def default_driver(self):