GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-07-09 06:23:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

get bind functions from upward searched debugtalk.py

Browse Source
This commit is contained in:
debugtalk
2017-08-31 12:46:38 +08:00
parent f9aaffce0e
commit 47e06b9c27
2 changed files with 29 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
ate/testcase.py
View File

@@ -1,8 +1,8 @@
import ast import ast
import os
import re import re
from ate.utils import long_type from ate import exception, utils
from ate.exception import ParamsError
variable_regexp = r"\$([\w_]+)" variable_regexp = r"\$([\w_]+)"
function_regexp = r"\$\{[\w_]+\([\$\w_ =,]*\)\}" function_regexp = r"\$\{[\w_]+\([\$\w_ =,]*\)\}"
@@ -105,7 +105,7 @@ def eval_content_variables(content, variable_mapping):
variables_list = extract_variables(content) variables_list = extract_variables(content)
for variable_name in variables_list: for variable_name in variables_list:
if variable_name not in variable_mapping: if variable_name not in variable_mapping:
raise ParamsError( raise exception.ParamsError(
"%s is not defined in bind variables!" % variable_name) "%s is not defined in bind variables!" % variable_name)
variable_value = variable_mapping.get(variable_name) variable_value = variable_mapping.get(variable_name)
@@ -124,9 +124,10 @@ def eval_content_variables(content, variable_mapping):
class TestcaseParser(object): class TestcaseParser(object):
def __init__(self, variables_binds={}, functions_binds={}): def __init__(self, variables_binds={}, functions_binds={}, file_path=None):
self.bind_variables(variables_binds) self.bind_variables(variables_binds)
self.bind_functions(functions_binds) self.bind_functions(functions_binds)
self.file_path = file_path
def bind_variables(self, variables_binds): def bind_variables(self, variables_binds):
""" bind variables to current testcase parser """ bind variables to current testcase parser
@@ -149,16 +150,25 @@ class TestcaseParser(object):
""" """
self.functions_binds = functions_binds self.functions_binds = functions_binds
def get_bind_fuctions(self, func_name):
func = self.functions_binds.get(func_name)
if func:
return func
try:
assert self.file_path is not None
return utils.search_conf_function(self.file_path, func_name)
except (AssertionError, exception.FunctionNotFound):
raise exception.ParamsError(
"%s is not defined in bind functions!" % func_name)
def eval_content_functions(self, content): def eval_content_functions(self, content):
functions_list = extract_functions(content) functions_list = extract_functions(content)
for func_content in functions_list: for func_content in functions_list:
function_meta = parse_function(func_content) function_meta = parse_function(func_content)
func_name = function_meta['func_name'] func_name = function_meta['func_name']
func = self.functions_binds.get(func_name) func = self.get_bind_fuctions(func_name)
if func is None:
raise ParamsError(
"%s is not defined in bind functions!" % func_name)
args = function_meta.get('args', []) args = function_meta.get('args', [])
kwargs = function_meta.get('kwargs', {}) kwargs = function_meta.get('kwargs', {})
@@ -222,7 +232,7 @@ class TestcaseParser(object):
return evaluated_data return evaluated_data
if isinstance(content, (int, long_type, float, complex)): if isinstance(content, (int, utils.long_type, float, complex)):
return content return content
# content is in string format here # content is in string format here

10
tests/test_testcase.py
View File

@@ -266,6 +266,16 @@ class TestcaseParserUnittest(unittest.TestCase):
"/api/3" "/api/3"
) )
def test_eval_content_functions_search_upward(self):
testcase_parser = testcase.TestcaseParser()
with self.assertRaises(ParamsError):
testcase_parser.eval_content_functions("/api/${gen_md5(abc)}")
testcase_parser.file_path = "tests/data/demo_testset_hardcode.yml"
content = testcase_parser.eval_content_functions("/api/${gen_md5(abc)}")
self.assertEqual(content, "/api/900150983cd24fb0d6963f7d28e17f72")
def test_parse_content_with_bindings_testcase(self): def test_parse_content_with_bindings_testcase(self):
variables_binds = { variables_binds = {
"uid": "1000", "uid": "1000",