GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-07-21 12:33:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

add args and kwargs validation for ENV() and P()

Browse Source
This commit is contained in:
httprunner
2018-11-16 12:45:06 +08:00
parent c3b52bbc79
commit fc9f828ed3
2 changed files with 31 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
httprunner/parser.py
View File

@@ -414,10 +414,14 @@ def parse_string_functions(content, variables_mapping, functions_mapping):
kwargs = parse_data(kwargs, variables_mapping, functions_mapping)
if func_name in ["parameterize", "P"]:
if len(args) != 1 or kwargs:
raise exceptions.ParamsError("P() should only pass in one argument!")
from httprunner import loader
eval_value = loader.load_csv_file(*args)
eval_value = loader.load_csv_file(args[0])
elif func_name in ["environ", "ENV"]:
eval_value = utils.get_os_environ(*args)
if len(args) != 1 or kwargs:
raise exceptions.ParamsError("ENV() should only pass in one argument!")
eval_value = utils.get_os_environ(args[0])
else:
func = get_mapping_function(func_name, functions_mapping)
eval_value = func(*args, **kwargs)