Read kselftests cgroup shell tests

acerv · Jun 27, 2023 · 3a9c044 · 3a9c044
1 parent d22e53e
commit 3a9c044
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/libkirk/kselftests.py b/libkirk/kselftests.py
@@ -50,13 +50,17 @@ async def _get_cgroup(self, sut: SUT) -> Suite:
             raise KirkException(
                 f"cgroup folder is not available: {cgroup_dir}")
 
-        ret = await sut.run_command(
-            "basename -s .c -- test_*.c",
-            cwd=cgroup_dir)
-        if ret["returncode"] != 0 or not ret["stdout"]:
-            raise KirkException("Can't read cgroup tests")
+        names = []
+        for ext in ("c", "sh"):
+            ret = await sut.run_command(
+                f"basename -s .c -- test_*.{ext}",
+                cwd=cgroup_dir)
+            if ret["returncode"] != 0 or not ret["stdout"]:
+                raise KirkException("Can't read cgroup tests")
+
+            tests = ret["stdout"].split('\n')
+            names.extend(tests)
 
-        names = ret["stdout"].split('\n')
         tests_obj = []
 
         for name in names: