diff --git a/examples/postman_echo/request_methods/demo_testsuite_yml/request_with_functions_test.py b/examples/postman_echo/request_methods/demo_testsuite_yml/request_with_functions_test.py index 7319ebaa..2c4505e4 100644 --- a/examples/postman_echo/request_methods/demo_testsuite_yml/request_with_functions_test.py +++ b/examples/postman_echo/request_methods/demo_testsuite_yml/request_with_functions_test.py @@ -1,5 +1,6 @@ # NOTICE: Generated By HttpRunner. DO'NOT EDIT! # FROM: examples/postman_echo/request_methods/request_with_functions.yml + from httprunner import HttpRunner, TConfig, TStep diff --git a/examples/postman_echo/request_methods/demo_testsuite_yml/request_with_testcase_reference_test.py b/examples/postman_echo/request_methods/demo_testsuite_yml/request_with_testcase_reference_test.py index a7647e8a..c3755910 100644 --- a/examples/postman_echo/request_methods/demo_testsuite_yml/request_with_testcase_reference_test.py +++ b/examples/postman_echo/request_methods/demo_testsuite_yml/request_with_testcase_reference_test.py @@ -1,5 +1,11 @@ # NOTICE: Generated By HttpRunner. DO'NOT EDIT! # FROM: examples/postman_echo/request_methods/request_with_testcase_reference.yml + +import os +import sys + +sys.path.insert(0, os.getcwd()) + from httprunner import HttpRunner, TConfig, TStep from examples.postman_echo.request_methods.request_with_functions_test import ( diff --git a/examples/postman_echo/request_methods/hardcode_test.py b/examples/postman_echo/request_methods/hardcode_test.py index 934c0d66..b6308880 100644 --- a/examples/postman_echo/request_methods/hardcode_test.py +++ b/examples/postman_echo/request_methods/hardcode_test.py @@ -1,5 +1,6 @@ # NOTICE: Generated By HttpRunner. DO'NOT EDIT! # FROM: examples/postman_echo/request_methods/hardcode.yml + from httprunner import HttpRunner, TConfig, TStep diff --git a/examples/postman_echo/request_methods/request_with_functions_test.py b/examples/postman_echo/request_methods/request_with_functions_test.py index 1933c03a..6b8f5d46 100644 --- a/examples/postman_echo/request_methods/request_with_functions_test.py +++ b/examples/postman_echo/request_methods/request_with_functions_test.py @@ -1,5 +1,6 @@ # NOTICE: Generated By HttpRunner. DO'NOT EDIT! # FROM: examples/postman_echo/request_methods/request_with_functions.yml + from httprunner import HttpRunner, TConfig, TStep diff --git a/examples/postman_echo/request_methods/request_with_testcase_reference_test.py b/examples/postman_echo/request_methods/request_with_testcase_reference_test.py index ea3921c1..70f3139c 100644 --- a/examples/postman_echo/request_methods/request_with_testcase_reference_test.py +++ b/examples/postman_echo/request_methods/request_with_testcase_reference_test.py @@ -1,5 +1,11 @@ # NOTICE: Generated By HttpRunner. DO'NOT EDIT! # FROM: examples/postman_echo/request_methods/request_with_testcase_reference.yml + +import os +import sys + +sys.path.insert(0, os.getcwd()) + from httprunner import HttpRunner, TConfig, TStep from examples.postman_echo.request_methods.request_with_functions_test import ( diff --git a/examples/postman_echo/request_methods/request_with_variables_test.py b/examples/postman_echo/request_methods/request_with_variables_test.py index 12dade62..37ec3582 100644 --- a/examples/postman_echo/request_methods/request_with_variables_test.py +++ b/examples/postman_echo/request_methods/request_with_variables_test.py @@ -1,5 +1,6 @@ # NOTICE: Generated By HttpRunner. DO'NOT EDIT! # FROM: examples/postman_echo/request_methods/request_with_variables.yml + from httprunner import HttpRunner, TConfig, TStep diff --git a/examples/postman_echo/request_methods/validate_with_functions_test.py b/examples/postman_echo/request_methods/validate_with_functions_test.py index a4561d96..8e48cf02 100644 --- a/examples/postman_echo/request_methods/validate_with_functions_test.py +++ b/examples/postman_echo/request_methods/validate_with_functions_test.py @@ -1,5 +1,6 @@ # NOTICE: Generated By HttpRunner. DO'NOT EDIT! # FROM: examples/postman_echo/request_methods/validate_with_functions.yml + from httprunner import HttpRunner, TConfig, TStep diff --git a/examples/postman_echo/request_methods/validate_with_variables_test.py b/examples/postman_echo/request_methods/validate_with_variables_test.py index 5fe266b9..4ad2c814 100644 --- a/examples/postman_echo/request_methods/validate_with_variables_test.py +++ b/examples/postman_echo/request_methods/validate_with_variables_test.py @@ -1,5 +1,6 @@ # NOTICE: Generated By HttpRunner. DO'NOT EDIT! # FROM: examples/postman_echo/request_methods/validate_with_variables.yml + from httprunner import HttpRunner, TConfig, TStep diff --git a/httprunner/ext/make/__init__.py b/httprunner/ext/make/__init__.py index 8af9b58f..11133489 100644 --- a/httprunner/ext/make/__init__.py +++ b/httprunner/ext/make/__init__.py @@ -22,12 +22,17 @@ make_files_cache_set: Set = set() __TEMPLATE__ = jinja2.Template( """# NOTICE: Generated By HttpRunner. DO'NOT EDIT! # FROM: {{ testcase_path }} +{% if imports_list %} +import os +import sys + +sys.path.insert(0, os.getcwd()) +{% endif %} from httprunner import HttpRunner, TConfig, TStep {% for import_str in imports_list %} {{ import_str }} {% endfor %} - class {{ class_name }}(HttpRunner): config = TConfig(**{{ config }}) diff --git a/httprunner/ext/make/make_test.py b/httprunner/ext/make/make_test.py index ade746dd..04c2adf4 100644 --- a/httprunner/ext/make/make_test.py +++ b/httprunner/ext/make/make_test.py @@ -89,7 +89,7 @@ from examples.postman_echo.request_methods.request_with_functions_test import ( path = ["examples/postman_echo/request_methods/demo_testsuite.yml"] make_files_cache_set.clear() testcase_python_list = main_make(path) - # FIXME: self.assertEqual(len(testcase_python_list), 2) + self.assertEqual(len(testcase_python_list), 3) self.assertIn( "examples/postman_echo/request_methods/demo_testsuite_yml/request_with_functions_test.py", testcase_python_list, @@ -98,3 +98,7 @@ from examples.postman_echo.request_methods.request_with_functions_test import ( "examples/postman_echo/request_methods/demo_testsuite_yml/request_with_testcase_reference_test.py", testcase_python_list, ) + self.assertIn( + "examples/postman_echo/request_methods/request_with_functions_test.py", + testcase_python_list, + )