Create a instance if the test func is a test object

Signed-off-by: Luyao Huang <[email protected]>
iccaszhulili · May 25, 2020 · 85fdf21 · 85fdf21
1 parent c235582
commit 85fdf21
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/depend_test_framework/runner_handlers.py b/depend_test_framework/runner_handlers.py
@@ -5,7 +5,7 @@
 import itertools
 import contextlib
 
-from .test_object import MistClearException, TestEndException, StaticMist, MistDeadEndException
+from .test_object import MistClearException, TestEndException, StaticMist, MistDeadEndException, is_TestObject
 from .case import Case
 from .log import get_logger
 
@@ -91,6 +91,8 @@ def handle_func(self, func, doc_func,
         else:
             self.desc_logger(test_func, only_doc)
 
+            if is_TestObject(test_func):
+                test_func = test_func()
             new_mist = test_func(params, tgt_env)
 
         if new_mist and mists is not None: