From 82b8975691873409700642504b7278904f71e2e5 Mon Sep 17 00:00:00 2001 From: debugtalk Date: Wed, 3 Jun 2020 21:36:05 +0800 Subject: [PATCH] feat: get raw testcase model from pytest file --- httprunner/make.py | 2 +- httprunner/runner.py | 7 +++++++ 2 files changed, 8 insertions(+), 1 deletion(-) diff --git a/httprunner/make.py b/httprunner/make.py index c656bad9..b5fe6fd6 100644 --- a/httprunner/make.py +++ b/httprunner/make.py @@ -24,7 +24,7 @@ make_files_cache_set: Set = set() pytest_files_set: Set = set() __TEMPLATE__ = jinja2.Template( - """# NOTICE: Generated By HttpRunner. DO NOT EDIT! + """# NOTICE: Generated By HttpRunner. # FROM: {{ testcase_path }} {% if imports_list %} import os diff --git a/httprunner/runner.py b/httprunner/runner.py index b42aef3a..9b94cf5e 100644 --- a/httprunner/runner.py +++ b/httprunner/runner.py @@ -58,6 +58,13 @@ class HttpRunner(object): for step in self.teststeps: self.__teststeps.append(step.perform()) + @property + def raw_testcase(self) -> TestCase: + if not hasattr(self, "__config"): + self.__init_tests__() + + return TestCase(config=self.__config, teststeps=self.__teststeps) + def with_project_meta(self, project_meta: ProjectMeta) -> "HttpRunner": self.__project_meta = project_meta return self