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

refactor names: simplify functions

Browse Source
This commit is contained in:
httprunner
2018-02-16 01:41:16 +08:00
parent 6e4c48979a
commit 151acd19be
6 changed files with 65 additions and 72 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
httprunner/context.py
View File

@@ -110,12 +110,12 @@ class Context(object):
variables = utils.convert_to_order_dict(variables) variables = utils.convert_to_order_dict(variables)
for variable_name, value in variables.items(): for variable_name, value in variables.items():
variable_evale_value = self.testcase_parser.parse_content_with_bindings(value) variable_eval_value = self.eval_content(value)
if level == "testset": if level == "testset":
self.testset_shared_variables_mapping[variable_name] = variable_evale_value self.testset_shared_variables_mapping[variable_name] = variable_eval_value
self.testcase_variables_mapping[variable_name] = variable_evale_value self.testcase_variables_mapping[variable_name] = variable_eval_value
self.testcase_parser.update_binded_variables(self.testcase_variables_mapping) self.testcase_parser.update_binded_variables(self.testcase_variables_mapping)
def bind_extracted_variables(self, variables): def bind_extracted_variables(self, variables):
@@ -140,13 +140,19 @@ class Context(object):
self.testcase_functions_config.update(config_mapping) self.testcase_functions_config.update(config_mapping)
self.testcase_parser.bind_functions(self.testcase_functions_config) self.testcase_parser.bind_functions(self.testcase_functions_config)
def eval_content(self, content):
""" evaluate content recursively, take effect on each variable and function in content.
content may be in any data structure, include dict, list, tuple, number, string, etc.
"""
return self.testcase_parser.eval_content_with_bindings(content)
def get_parsed_request(self, request_dict, level="testcase"): def get_parsed_request(self, request_dict, level="testcase"):
""" get parsed request with bind variables and functions. """ get parsed request with bind variables and functions.
@param request_dict: request config mapping @param request_dict: request config mapping
@param level: testset or testcase @param level: testset or testcase
""" """
if level == "testset": if level == "testset":
request_dict = self.testcase_parser.parse_content_with_bindings( request_dict = self.eval_content(
request_dict request_dict
) )
self.testset_request_config.update(request_dict) self.testset_request_config.update(request_dict)
@@ -155,20 +161,12 @@ class Context(object):
copy.deepcopy(self.testset_request_config), copy.deepcopy(self.testset_request_config),
request_dict request_dict
) )
parsed_request = self.testcase_parser.parse_content_with_bindings( parsed_request = self.eval_content(
testcase_request_config testcase_request_config
) )
return parsed_request return parsed_request
def get_testcase_variables_mapping(self):
return self.testcase_variables_mapping
def exec_content_functions(self, content):
""" execute functions in content.
"""
return self.testcase_parser.eval_content_functions(content)
def eval_check_item(self, validator, resp_obj): def eval_check_item(self, validator, resp_obj):
""" evaluate check item in validator """ evaluate check item in validator
@param (dict) validator @param (dict) validator
@@ -190,7 +188,7 @@ class Context(object):
# 3, regex string, e.g. "LB[\d]*(.*)RB[\d]*" # 3, regex string, e.g. "LB[\d]*(.*)RB[\d]*"
if testcase.extract_variables(check_item): if testcase.extract_variables(check_item):
# type 1 # type 1
check_value = self.testcase_parser.eval_content_variables(check_item) check_value = self.eval_content(check_item)
else: else:
try: try:
# type 2 or type 3 # type 2 or type 3
@@ -203,7 +201,7 @@ class Context(object):
# expect_value should only be in 2 types: # expect_value should only be in 2 types:
# 1, variable reference, e.g. $expect_status_code # 1, variable reference, e.g. $expect_status_code
# 2, actual value, e.g. 200 # 2, actual value, e.g. 200
expect_value = self.testcase_parser.eval_content_variables(validator["expect"]) expect_value = self.eval_content(validator["expect"])
validator["expect"] = expect_value validator["expect"] = expect_value
return validator return validator

13
httprunner/runner.py
View File

@@ -82,12 +82,12 @@ class Runner(object):
elif "skipIf" in testcase_dict: elif "skipIf" in testcase_dict:
skip_if_condition = testcase_dict["skipIf"] skip_if_condition = testcase_dict["skipIf"]
if self.context.exec_content_functions(skip_if_condition): if self.context.eval_content(skip_if_condition):
skip_reason = "{} evaluate to True".format(skip_if_condition) skip_reason = "{} evaluate to True".format(skip_if_condition)
elif "skipUnless" in testcase_dict: elif "skipUnless" in testcase_dict:
skip_unless_condition = testcase_dict["skipUnless"] skip_unless_condition = testcase_dict["skipUnless"]
if not self.context.exec_content_functions(skip_unless_condition): if not self.context.eval_content(skip_unless_condition):
skip_reason = "{} evaluate to False".format(skip_unless_condition) skip_reason = "{} evaluate to False".format(skip_unless_condition)
if skip_reason: if skip_reason:
@@ -138,7 +138,7 @@ class Runner(object):
def setup_teardown(actions): def setup_teardown(actions):
for action in actions: for action in actions:
self.context.exec_content_functions(action) self.context.eval_content(action)
setup_teardown(setup_actions) setup_teardown(setup_actions)
@@ -168,15 +168,10 @@ class Runner(object):
return True return True
def get_eval_content(self, content):
""" evaluate content with variable bindings
"""
return self.context.testcase_parser.parse_content_with_bindings(content)
def extract_output(self, output_variables_list): def extract_output(self, output_variables_list):
""" extract output variables """ extract output variables
""" """
variables_mapping = self.context.get_testcase_variables_mapping() variables_mapping = self.context.testcase_variables_mapping
output = {} output = {}
for variable in output_variables_list: for variable in output_variables_list:

2
httprunner/task.py
View File

@@ -72,7 +72,7 @@ class ApiTestSuite(unittest.TestSuite):
def _add_tests_to_suite(self, testcases): def _add_tests_to_suite(self, testcases):
for testcase_dict in testcases: for testcase_dict in testcases:
testcase_name = self.test_runner.get_eval_content(testcase_dict["name"]) testcase_name = self.test_runner.context.eval_content(testcase_dict["name"])
if utils.PYTHON_VERSION == 3: if utils.PYTHON_VERSION == 3:
ApiTestCase.runTest.__doc__ = testcase_name ApiTestCase.runTest.__doc__ = testcase_name
else: else:

22
httprunner/testcase.py
View File

@@ -717,7 +717,7 @@ class TestcaseParser(object):
raise exception.ParamsError( raise exception.ParamsError(
"{} is not defined in bind {}s!".format(item_name, item_type)) "{} is not defined in bind {}s!".format(item_name, item_type))
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)
@@ -727,8 +727,8 @@ class TestcaseParser(object):
args = function_meta.get('args', []) args = function_meta.get('args', [])
kwargs = function_meta.get('kwargs', {}) kwargs = function_meta.get('kwargs', {})
args = self.parse_content_with_bindings(args) args = self.eval_content_with_bindings(args)
kwargs = self.parse_content_with_bindings(kwargs) kwargs = self.eval_content_with_bindings(kwargs)
eval_value = func(*args, **kwargs) eval_value = func(*args, **kwargs)
func_content = "${" + func_content + "}" func_content = "${" + func_content + "}"
@@ -744,7 +744,7 @@ class TestcaseParser(object):
return content return content
def eval_content_variables(self, content): def _eval_content_variables(self, content):
""" replace all variables of string content with mapping value. """ replace all variables of string content with mapping value.
@param (str) content @param (str) content
@return (str) parsed content @return (str) parsed content
@@ -775,7 +775,7 @@ class TestcaseParser(object):
return content return content
def parse_content_with_bindings(self, content): def eval_content_with_bindings(self, content):
""" parse content recursively, each variable and function in content will be evaluated. """ parse content recursively, each variable and function in content will be evaluated.
@param (dict) content in any data structure @param (dict) content in any data structure
@@ -806,15 +806,15 @@ class TestcaseParser(object):
if isinstance(content, (list, tuple)): if isinstance(content, (list, tuple)):
return [ return [
self.parse_content_with_bindings(item) self.eval_content_with_bindings(item)
for item in content for item in content
] ]
if isinstance(content, dict): if isinstance(content, dict):
evaluated_data = {} evaluated_data = {}
for key, value in content.items(): for key, value in content.items():
eval_key = self.parse_content_with_bindings(key) eval_key = self.eval_content_with_bindings(key)
eval_value = self.parse_content_with_bindings(value) eval_value = self.eval_content_with_bindings(value)
evaluated_data[eval_key] = eval_value evaluated_data[eval_key] = eval_value
return evaluated_data return evaluated_data
@@ -826,10 +826,10 @@ class TestcaseParser(object):
content = "" if content is None else content.strip() content = "" if content is None else content.strip()
# replace functions with evaluated value # replace functions with evaluated value
# Notice: eval_content_functions must be called before eval_content_variables # Notice: _eval_content_functions must be called before _eval_content_variables
content = self.eval_content_functions(content) content = self._eval_content_functions(content)
# replace variables with binding value # replace variables with binding value
content = self.eval_content_variables(content) content = self._eval_content_variables(content)
return content return content

14
tests/test_context.py
View File

@@ -23,7 +23,7 @@ class VariableBindsUnittest(ApiServerUnittest):
{"timestamp10": "${get_timestamp(10)}"} {"timestamp10": "${get_timestamp(10)}"}
] ]
self.context.bind_variables(variables) self.context.bind_variables(variables)
context_variables = self.context.get_testcase_variables_mapping() context_variables = self.context.testcase_variables_mapping
self.assertEqual(len(context_variables["random"]), 5) self.assertEqual(len(context_variables["random"]), 5)
self.assertEqual(len(context_variables["timestamp10"]), 10) self.assertEqual(len(context_variables["timestamp10"]), 10)
@@ -44,7 +44,7 @@ class VariableBindsUnittest(ApiServerUnittest):
self.context.bind_variables(variables, level="testset") self.context.bind_variables(variables, level="testset")
testset_variables = self.context.testset_shared_variables_mapping testset_variables = self.context.testset_shared_variables_mapping
testcase_variables = self.context.get_testcase_variables_mapping() testcase_variables = self.context.testcase_variables_mapping
self.assertIn("GLOBAL_TOKEN", testset_variables) self.assertIn("GLOBAL_TOKEN", testset_variables)
self.assertIn("GLOBAL_TOKEN", testcase_variables) self.assertIn("GLOBAL_TOKEN", testcase_variables)
self.assertEqual(testset_variables["GLOBAL_TOKEN"], "debugtalk") self.assertEqual(testset_variables["GLOBAL_TOKEN"], "debugtalk")
@@ -66,7 +66,7 @@ class VariableBindsUnittest(ApiServerUnittest):
self.context.bind_variables(variables) self.context.bind_variables(variables)
testset_variables = self.context.testset_shared_variables_mapping testset_variables = self.context.testset_shared_variables_mapping
testcase_variables = self.context.get_testcase_variables_mapping() testcase_variables = self.context.testcase_variables_mapping
self.assertNotIn("GLOBAL_TOKEN", testset_variables) self.assertNotIn("GLOBAL_TOKEN", testset_variables)
self.assertIn("GLOBAL_TOKEN", testcase_variables) self.assertIn("GLOBAL_TOKEN", testcase_variables)
self.assertEqual(testcase_variables["GLOBAL_TOKEN"], "debugtalk") self.assertEqual(testcase_variables["GLOBAL_TOKEN"], "debugtalk")
@@ -94,7 +94,7 @@ class VariableBindsUnittest(ApiServerUnittest):
variables = testcase['variables'] variables = testcase['variables']
self.context.bind_variables(variables) self.context.bind_variables(variables)
context_variables = self.context.get_testcase_variables_mapping() context_variables = self.context.testcase_variables_mapping
self.assertIn("add1", context_variables) self.assertIn("add1", context_variables)
self.assertEqual(context_variables["add1"], 3) self.assertEqual(context_variables["add1"], 3)
self.assertIn("sum2nums", context_variables) self.assertIn("sum2nums", context_variables)
@@ -125,7 +125,7 @@ class VariableBindsUnittest(ApiServerUnittest):
variables = testcase['variables'] variables = testcase['variables']
self.context.bind_variables(variables) self.context.bind_variables(variables)
context_variables = self.context.get_testcase_variables_mapping() context_variables = self.context.testcase_variables_mapping
self.assertIn("TOKEN", context_variables) self.assertIn("TOKEN", context_variables)
TOKEN = context_variables["TOKEN"] TOKEN = context_variables["TOKEN"]
@@ -159,7 +159,7 @@ class VariableBindsUnittest(ApiServerUnittest):
variables = testcase['variables'] variables = testcase['variables']
self.context.bind_variables(variables) self.context.bind_variables(variables)
context_variables = self.context.get_testcase_variables_mapping() context_variables = self.context.testcase_variables_mapping
self.assertIn("TOKEN", context_variables) self.assertIn("TOKEN", context_variables)
TOKEN = context_variables["TOKEN"] TOKEN = context_variables["TOKEN"]
@@ -213,7 +213,7 @@ class VariableBindsUnittest(ApiServerUnittest):
test_runner = runner.Runner() test_runner = runner.Runner()
content = "${sleep(1)}" content = "${sleep(1)}"
start_time = time.time() start_time = time.time()
test_runner.context.exec_content_functions(content) test_runner.context.eval_content(content)
end_time = time.time() end_time = time.time()
elapsed_time = end_time - start_time elapsed_time = end_time - start_time
self.assertGreater(elapsed_time, 1) self.assertGreater(elapsed_time, 1)

58
tests/test_testcase.py
View File

@@ -247,51 +247,51 @@ class TestcaseParserUnittest(unittest.TestCase):
} }
testcase_parser = testcase.TestcaseParser(variables=variables) testcase_parser = testcase.TestcaseParser(variables=variables)
self.assertEqual( self.assertEqual(
testcase_parser.eval_content_variables("$var_1"), testcase_parser._eval_content_variables("$var_1"),
"abc" "abc"
) )
self.assertEqual( self.assertEqual(
testcase_parser.eval_content_variables("var_1"), testcase_parser._eval_content_variables("var_1"),
"var_1" "var_1"
) )
self.assertEqual( self.assertEqual(
testcase_parser.eval_content_variables("$var_1#XYZ"), testcase_parser._eval_content_variables("$var_1#XYZ"),
"abc#XYZ" "abc#XYZ"
) )
self.assertEqual( self.assertEqual(
testcase_parser.eval_content_variables("/$var_1/$var_2/var3"), testcase_parser._eval_content_variables("/$var_1/$var_2/var3"),
"/abc/def/var3" "/abc/def/var3"
) )
self.assertEqual( self.assertEqual(
testcase_parser.eval_content_variables("/$var_1/$var_2/$var_1"), testcase_parser._eval_content_variables("/$var_1/$var_2/$var_1"),
"/abc/def/abc" "/abc/def/abc"
) )
self.assertEqual( self.assertEqual(
testcase_parser.eval_content_variables("${func($var_1, $var_2, xyz)}"), testcase_parser._eval_content_variables("${func($var_1, $var_2, xyz)}"),
"${func(abc, def, xyz)}" "${func(abc, def, xyz)}"
) )
self.assertEqual( self.assertEqual(
testcase_parser.eval_content_variables("$var_3"), testcase_parser._eval_content_variables("$var_3"),
123 123
) )
self.assertEqual( self.assertEqual(
testcase_parser.eval_content_variables("$var_4"), testcase_parser._eval_content_variables("$var_4"),
{"a": 1} {"a": 1}
) )
self.assertEqual( self.assertEqual(
testcase_parser.eval_content_variables("$var_5"), testcase_parser._eval_content_variables("$var_5"),
True True
) )
self.assertEqual( self.assertEqual(
testcase_parser.eval_content_variables("abc$var_5"), testcase_parser._eval_content_variables("abc$var_5"),
"abcTrue" "abcTrue"
) )
self.assertEqual( self.assertEqual(
testcase_parser.eval_content_variables("abc$var_4"), testcase_parser._eval_content_variables("abc$var_4"),
"abc{'a': 1}" "abc{'a': 1}"
) )
self.assertEqual( self.assertEqual(
testcase_parser.eval_content_variables("$var_6"), testcase_parser._eval_content_variables("$var_6"),
None None
) )
@@ -299,10 +299,10 @@ class TestcaseParserUnittest(unittest.TestCase):
testcase_parser = testcase.TestcaseParser() testcase_parser = testcase.TestcaseParser()
with self.assertRaises(ParamsError): with self.assertRaises(ParamsError):
testcase_parser.eval_content_variables("/api/$SECRET_KEY") testcase_parser._eval_content_variables("/api/$SECRET_KEY")
testcase_parser.file_path = "tests/data/demo_testset_hardcode.yml" testcase_parser.file_path = "tests/data/demo_testset_hardcode.yml"
content = testcase_parser.eval_content_variables("/api/$SECRET_KEY") content = testcase_parser._eval_content_variables("/api/$SECRET_KEY")
self.assertEqual(content, "/api/DebugTalk") self.assertEqual(content, "/api/DebugTalk")
def test_parse_string_value(self): def test_parse_string_value(self):
@@ -345,23 +345,23 @@ class TestcaseParserUnittest(unittest.TestCase):
} }
testcase_parser = testcase.TestcaseParser(variables=variables) testcase_parser = testcase.TestcaseParser(variables=variables)
self.assertEqual( self.assertEqual(
testcase_parser.parse_content_with_bindings("$str_1"), testcase_parser.eval_content_with_bindings("$str_1"),
"str_value1" "str_value1"
) )
self.assertEqual( self.assertEqual(
testcase_parser.parse_content_with_bindings("123$str_1/456"), testcase_parser.eval_content_with_bindings("123$str_1/456"),
"123str_value1/456" "123str_value1/456"
) )
with self.assertRaises(ParamsError): with self.assertRaises(ParamsError):
testcase_parser.parse_content_with_bindings("$str_3") testcase_parser.eval_content_with_bindings("$str_3")
self.assertEqual( self.assertEqual(
testcase_parser.parse_content_with_bindings(["$str_1", "str3"]), testcase_parser.eval_content_with_bindings(["$str_1", "str3"]),
["str_value1", "str3"] ["str_value1", "str3"]
) )
self.assertEqual( self.assertEqual(
testcase_parser.parse_content_with_bindings({"key": "$str_1"}), testcase_parser.eval_content_with_bindings({"key": "$str_1"}),
{"key": "str_value1"} {"key": "str_value1"}
) )
@@ -373,7 +373,7 @@ class TestcaseParserUnittest(unittest.TestCase):
testcase_parser = testcase.TestcaseParser(variables=variables) testcase_parser = testcase.TestcaseParser(variables=variables)
content = "/users/$userid/training/$data?userId=$userid&data=$data" content = "/users/$userid/training/$data?userId=$userid&data=$data"
self.assertEqual( self.assertEqual(
testcase_parser.parse_content_with_bindings(content), testcase_parser.eval_content_with_bindings(content),
"/users/100/training/1498?userId=100&data=1498" "/users/100/training/1498?userId=100&data=1498"
) )
@@ -386,7 +386,7 @@ class TestcaseParserUnittest(unittest.TestCase):
testcase_parser = testcase.TestcaseParser(variables=variables) testcase_parser = testcase.TestcaseParser(variables=variables)
content = "/users/$user/$userid/$data?userId=$userid&data=$data" content = "/users/$user/$userid/$data?userId=$userid&data=$data"
self.assertEqual( self.assertEqual(
testcase_parser.parse_content_with_bindings(content), testcase_parser.eval_content_with_bindings(content),
"/users/100/1000/1498?userId=1000&data=1498" "/users/100/1000/1498?userId=1000&data=1498"
) )
@@ -398,17 +398,17 @@ class TestcaseParserUnittest(unittest.TestCase):
} }
testcase_parser = testcase.TestcaseParser(functions=functions) testcase_parser = testcase.TestcaseParser(functions=functions)
result = testcase_parser.parse_content_with_bindings("${gen_random_string(5)}") result = testcase_parser.eval_content_with_bindings("${gen_random_string(5)}")
self.assertEqual(len(result), 5) self.assertEqual(len(result), 5)
add_two_nums = lambda a, b=1: a + b add_two_nums = lambda a, b=1: a + b
functions["add_two_nums"] = add_two_nums functions["add_two_nums"] = add_two_nums
self.assertEqual( self.assertEqual(
testcase_parser.parse_content_with_bindings("${add_two_nums(1)}"), testcase_parser.eval_content_with_bindings("${add_two_nums(1)}"),
2 2
) )
self.assertEqual( self.assertEqual(
testcase_parser.parse_content_with_bindings("${add_two_nums(1, 2)}"), testcase_parser.eval_content_with_bindings("${add_two_nums(1, 2)}"),
3 3
) )
@@ -452,11 +452,11 @@ class TestcaseParserUnittest(unittest.TestCase):
} }
testcase_parser = testcase.TestcaseParser(functions=functions) testcase_parser = testcase.TestcaseParser(functions=functions)
self.assertEqual( self.assertEqual(
testcase_parser.eval_content_functions("${add_two_nums(1, 2)}"), testcase_parser._eval_content_functions("${add_two_nums(1, 2)}"),
3 3
) )
self.assertEqual( self.assertEqual(
testcase_parser.eval_content_functions("/api/${add_two_nums(1, 2)}"), testcase_parser._eval_content_functions("/api/${add_two_nums(1, 2)}"),
"/api/3" "/api/3"
) )
@@ -464,10 +464,10 @@ class TestcaseParserUnittest(unittest.TestCase):
testcase_parser = testcase.TestcaseParser() testcase_parser = testcase.TestcaseParser()
with self.assertRaises(ParamsError): with self.assertRaises(ParamsError):
testcase_parser.eval_content_functions("/api/${gen_md5(abc)}") testcase_parser._eval_content_functions("/api/${gen_md5(abc)}")
testcase_parser.file_path = "tests/data/demo_testset_hardcode.yml" testcase_parser.file_path = "tests/data/demo_testset_hardcode.yml"
content = testcase_parser.eval_content_functions("/api/${gen_md5(abc)}") content = testcase_parser._eval_content_functions("/api/${gen_md5(abc)}")
self.assertEqual(content, "/api/900150983cd24fb0d6963f7d28e17f72") self.assertEqual(content, "/api/900150983cd24fb0d6963f7d28e17f72")
def test_parse_content_with_bindings_testcase(self): def test_parse_content_with_bindings_testcase(self):
@@ -493,7 +493,7 @@ class TestcaseParserUnittest(unittest.TestCase):
"body": "$data" "body": "$data"
} }
parsed_testcase = testcase.TestcaseParser(variables, functions)\ parsed_testcase = testcase.TestcaseParser(variables, functions)\
.parse_content_with_bindings(testcase_template) .eval_content_with_bindings(testcase_template)
self.assertEqual( self.assertEqual(
parsed_testcase["url"], parsed_testcase["url"],