From 34e942ee5cae69975989bb5deeed822f4ed8ba0e Mon Sep 17 00:00:00 2001 From: debugtalk Date: Mon, 17 Sep 2018 21:40:25 +0800 Subject: [PATCH] initialize TestLoader in HttpRunner initialization --- httprunner/api.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/httprunner/api.py b/httprunner/api.py index 35ff3a9a..d0ce9c1c 100644 --- a/httprunner/api.py +++ b/httprunner/api.py @@ -41,6 +41,7 @@ class HttpRunner(object): self.http_client_session = kwargs.pop("http_client_session", None) kwargs.setdefault("resultclass", report.HtmlTestResult) self.unittest_runner = unittest.TextTestRunner(**kwargs) + self.test_loader = unittest.TestLoader() def load_tests(self, path_or_testcases): """ load testcases, extend and merge with api/testcase definitions. @@ -216,9 +217,7 @@ class HttpRunner(object): self.exception_stage = "add tests to test suite" - testcases_list = [] - loader = unittest.TestLoader() - loaded_testcases = [] + test_suite = unittest.TestSuite() for testcase in testcases: config = testcase.get("config", {}) test_runner = runner.Runner(config, self.http_client_session) @@ -233,13 +232,12 @@ class HttpRunner(object): test_method = __add_teststep(test_runner, config, teststep_dict) setattr(TestSequense, test_method_name, test_method) - loaded_testcase = loader.loadTestsFromTestCase(TestSequense) + loaded_testcase = self.test_loader.loadTestsFromTestCase(TestSequense) setattr(loaded_testcase, "config", config) setattr(loaded_testcase, "teststeps", testcase.get("teststeps", [])) setattr(loaded_testcase, "runner", test_runner) - loaded_testcases.append(loaded_testcase) + test_suite.addTest(loaded_testcase) - test_suite = unittest.TestSuite(loaded_testcases) return test_suite def run_tests(self, test_suite):