refactor: parameters

httprunner/make.py

@@ -37,23 +37,31 @@ pytest_files_run_set: Set = set()
 __TEMPLATE__ = jinja2.Template(
     """# NOTE: Generated By HttpRunner v{{ version }}
 # FROM: {{ testcase_path }}
+
 {% if imports_list and diff_levels > 0 %}
 import sys
 from pathlib import Path
-
-sys.path.insert(0, str(Path(__file__)
-{% for _ in range(diff_levels) %}
-.parent
-{% endfor %}
-))
+sys.path.insert(0, str(Path(__file__){% for _ in range(diff_levels) %}.parent{% endfor %}))
 {% endif %}
+
+{% if parameters %}
+import pytest
+from httprunner import parse_parameters
+{% endif %}
+
 from httprunner import HttpRunner, Config, Step, RunRequest, RunTestCase
 {% for import_str in imports_list %}
 {{ import_str }}
 {% endfor %}
 
 class {{ class_name }}(HttpRunner):
-    {{ customization_test_start }}
+
+    {% if parameters %}
+    @pytest.mark.parametrize('parameters', parse_parameters({{parameters}}))
+    def test_start(self, parameters):
+        super().test_start(parameters)
+    {% endif %}
+
     config = {{ config_chain_style }}
 
     teststeps = [
@@ -411,7 +419,7 @@ def make_testcase(testcase: Dict, dir_path: Text = None) -> Text:
         "class_name": f"TestCase{testcase_cls_name}",
         "imports_list": imports_list,
         "config_chain_style": make_config_chain_style(config),
-        "customization_test_start": make_test_start_style(config),
+        "parameters": config.get("parameters"),
         "teststeps_chain_style": [
             make_teststep_chain_style(step) for step in teststeps
         ],
@@ -611,22 +619,3 @@ def init_make_parser(subparsers):
     )
 
     return parser
-
-
-def make_test_start_style(config: Dict) -> Text:
-    test_start_style = ""
-    if "parameters" in config.keys():
-        params = config["parameters"]
-        test_start_style = f"""
-    import pytest
-    from httprunner.parser import parse_parameters
-    
-    param = [{params}]
-
-    @pytest.mark.parametrize('parametrize', parse_parameters(param))
-    def test_start(self, parametrize):
-        super().test_start(parametrize)
-        """
-    else:
-        pass
-    return test_start_style