feat: support testsuite in testsuite

Tencent · May 4, 2023 · 8c27b11 · 8c27b11
1 parent 77a9795
commit 8c27b11
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/testbase/loader.py b/testbase/loader.py
@@ -317,9 +317,15 @@ def _load_from_testsuite(
                         ignore_testsuite=True,
                     )
             else:
-                tests += self._load_from_class(
-                    test, data_key, exclude_data_key=exclude_data_key, attrs=attrs
-                )
+                if issubclass(test, TestSuite):
+                    testcases = self._load_from_testsuite(
+                        test, data_key, exclude_data_key=exclude_data_key, attrs=attrs
+                    )
+                    tests += [test(testcases)]
+                else:
+                    tests += self._load_from_class(
+                        test, data_key, exclude_data_key=exclude_data_key, attrs=attrs
+                    )
 
         return [it for it in tests if not cls.filter(it)]