From 8c27b1138169e87da6eee2b59ec4b4a65d47671f Mon Sep 17 00:00:00 2001
From: Ming5024 <774517491@qq.com>
Date: Thu, 4 May 2023 18:00:29 +0800
Subject: [PATCH] feat: support testsuite in testsuite

---
 testbase/loader.py | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/testbase/loader.py b/testbase/loader.py
index 3a05c1a..5da4191 100644
--- 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)]