GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-05-24 17:59:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

LazyFunction: wrap argument

Browse Source
This commit is contained in:
debugtalk
2019-04-10 20:23:50 +08:00
parent b5d4b686df
commit 8cd5ac4b25
2 changed files with 15 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
httprunner/parser.py
View File

@@ -348,6 +348,7 @@ class LazyFunction(object):
function_meta["func_name"],
self.functions_mapping
)
self.func_name = self._func.__name__
self._args = prepare_lazy_data(
function_meta.get("args", []),
self.functions_mapping,
@@ -359,15 +360,21 @@ class LazyFunction(object):
self.check_variables_set
)
if self._func.__name__ == "load_csv_file":
if self.func_name == "load_csv_file":
if len(self._args) != 1 or self._kwargs:
raise exceptions.ParamsError("P() should only pass in one argument!")
self._args = [self._args[0]]
elif self._func.__name__ == "get_os_environ":
elif self.func_name == "get_os_environ":
if len(self._args) != 1 or self._kwargs:
raise exceptions.ParamsError("ENV() should only pass in one argument!")
self._args = [self._args[0]]
def get_args(self):
return self._args
def update_args(self, args):
self._args = args
def __repr__(self):
args_string = ""
@@ -383,10 +390,10 @@ class LazyFunction(object):
]
args_string += ", ".join(str_kwargs)
return "LazyFunction({}({}))".format(self._func.__name__, args_string)
return "LazyFunction({}({}))".format(self.func_name, args_string)
def __prepare_cache_key(self, args, kwargs):
return (self._func.__name__, repr(args), repr(kwargs))
return (self.func_name, repr(args), repr(kwargs))
def to_value(self, variables_mapping=None):
""" parse lazy data with evaluated variables mapping.