GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-09-05 15:37:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor parameterize with csv

Browse Source
This commit is contained in:
debugtalk
2018-08-09 08:46:25 +08:00
parent 6c218a1c4f
commit 85cdbb2665
1 changed files with 14 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
httprunner/parser.py
+14 -7
View File
@@ -386,12 +386,19 @@ class TestcaseParser(object):
def get_bind_variable(self, variable_name): def get_bind_variable(self, variable_name):
return self._get_bind_item("variables", variable_name) return self._get_bind_item("variables", variable_name)
def parameterize(self, csv_file_name, fetch_method="Sequential"): def load_csv_list(self, csv_file_name, fetch_method="Sequential"):
parameter_file_path = os.path.join( """ locate csv file and load csv content.
os.path.dirname(self.file_path),
"{}".format(csv_file_name) Args:
) csv_file_name (str): csv file name
csv_content_list = loader.load_file(parameter_file_path) fetch_method (str): fetch data method, defaults to Sequential.
If set to "random", csv data list will be reordered in random.
Returns:
list: csv data list
"""
csv_file_path = loader.locate_file(self.file_path, csv_file_name)
csv_content_list = loader.load_file(csv_file_path)
if fetch_method.lower() == "random": if fetch_method.lower() == "random":
random.shuffle(csv_content_list) random.shuffle(csv_content_list)
@@ -410,7 +417,7 @@ class TestcaseParser(object):
kwargs = self.eval_content_with_bindings(kwargs) kwargs = self.eval_content_with_bindings(kwargs)
if func_name in ["parameterize", "P"]: if func_name in ["parameterize", "P"]:
eval_value = self.parameterize(*args, **kwargs) eval_value = self.load_csv_list(*args, **kwargs)
else: else:
func = self.get_bind_function(func_name) func = self.get_bind_function(func_name)
eval_value = func(*args, **kwargs) eval_value = func(*args, **kwargs)