def bind_variables(self, variable_binds, level="testcase"):
""" bind variables to testset context or current testcase context.
variables in testset context can be used in all testcases of current test suite.
@param (list) variable_binds, variable can be value or custom function.
if value is function, it will be called and bind result to variable.
e.g.
[
{"TOKEN": "debugtalk"},
{"random": "${gen_random_string(5)}"},
{"json": {'name': 'user', 'password': '******'}},
{"md5": "${gen_md5($TOKEN, $json, $random)}"}
]
"""
for variable_bind in variable_binds:
for variable_name, value in variable_bind.items():
variable_evale_value = testcase.parse_content_with_bindings(
value,
self.testcase_variables_mapping,
self.testcase_functions_config
)
if level == "testset":
self.testset_shared_variables_mapping[variable_name] = variable_evale_value
self.testcase_variables_mapping[variable_name] = variable_evale_value
def bind_variables(self, variable_binds, level="testcase"):
""" bind variables to testset context or current testcase context.
variables in testset context can be used in all testcases of current test suite.
@param (list) variable_binds, variable can be value or custom function.
if value is function, it will be called and bind result to variable.
e.g.
[
{"TOKEN": "debugtalk"},
{"random": "${gen_random_string(5)}"},
{"json": {'name': 'user', 'password': '******'}},
{"md5": "${gen_md5($TOKEN, $json, $random)}"}
]
"""
for variable_bind in variable_binds:
for variable_name, value in variable_bind.items():
variable_evale_value = testcase.parse_content_with_bindings(
value, self.testcase_variables_mapping,
self.testcase_functions_config)
if level == "testset":
self.testset_shared_variables_mapping[
variable_name] = variable_evale_value
self.testcase_variables_mapping[
variable_name] = variable_evale_value
def get_parsed_request(self):
""" get parsed request, with each variable replaced by bind value.
"""
parsed_request = testcase.parse_content_with_bindings(
self.testcase_request_config, self.testcase_variables_mapping,
self.testcase_functions_config)
return parsed_request
def test_parse_content_with_bindings_functions(self):
import random, string
functions_binds = {
"gen_random_string": lambda str_len: ''.join(random.choice(string.ascii_letters + string.digits) \
for _ in range(str_len))
}
result = testcase.parse_content_with_bindings(
"${gen_random_string(5)}", {}, functions_binds)
self.assertEqual(len(result), 5)
add_two_nums = lambda a, b=1: a + b
functions_binds["add_two_nums"] = add_two_nums
self.assertEqual(
testcase.parse_content_with_bindings("${add_two_nums(1)}", {},
functions_binds), 2)
self.assertEqual(
testcase.parse_content_with_bindings("${add_two_nums(1, 2)}", {},
functions_binds), 3)
def get_parsed_request(self):
""" get parsed request, with each variable replaced by bind value.
"""
parsed_request = testcase.parse_content_with_bindings(
self.testcase_request_config,
self.testcase_variables_mapping,
self.testcase_functions_config
)
return parsed_request
def test_parse_content_with_bindings_multiple_identical_variables(self):
variables_binds = {
"userid": 100,
"data": 1498
}
content = "/users/$userid/training/$data?userId=$userid&data=$data"
self.assertEqual(
testcase.parse_content_with_bindings(content, variables_binds, {}),
"/users/100/training/1498?userId=100&data=1498"
)
def test_parse_content_with_bindings_functions(self):
import random, string
functions_binds = {
"gen_random_string": lambda str_len: ''.join(random.choice(string.ascii_letters + string.digits) \
for _ in range(str_len))
}
result = testcase.parse_content_with_bindings("${gen_random_string(5)}", {}, functions_binds)
self.assertEqual(len(result), 5)
add_two_nums = lambda a, b=1: a + b
functions_binds["add_two_nums"] = add_two_nums
self.assertEqual(
testcase.parse_content_with_bindings("${add_two_nums(1)}", {}, functions_binds),
2
)
self.assertEqual(
testcase.parse_content_with_bindings("${add_two_nums(1, 2)}", {}, functions_binds),
3
)
def test_parse_content_with_bindings_variables(self):
variables_binds = {
"str_1": "str_value1",
"str_2": "str_value2"
}
self.assertEqual(
testcase.parse_content_with_bindings("$str_1", variables_binds, {}),
"str_value1"
)
self.assertEqual(
testcase.parse_content_with_bindings("123$str_1/456", variables_binds, {}),
"123str_value1/456"
)
with self.assertRaises(ParamsError):
testcase.parse_content_with_bindings("$str_3", variables_binds, {})
self.assertEqual(
testcase.parse_content_with_bindings(["$str_1", "str3"], variables_binds, {}),
["str_value1", "str3"]
)
self.assertEqual(
testcase.parse_content_with_bindings({"key": "$str_1"}, variables_binds, {}),
{"key": "str_value1"}
)
def test_parse_content_with_bindings_testcase(self):
variables_binds = {
"uid": "1000",
"random": "A2dEx",
"authorization": "a83de0ff8d2e896dbd8efb81ba14e17d",
"data": {"name": "user", "password": "******"},
"expected_status": 201,
"expected_success": True
}
functions_binds = {
"add_two_nums": lambda a, b=1: a + b
}
testcase_template = {
"url": "http://127.0.0.1:5000/api/users/$uid",
"method": "POST",
"headers": {
"Content-Type": "application/json",
"authorization": "$authorization",
"random": "$random",
"sum": "${add_two_nums(1, 2)}"
},
"body": "$data"
}
parsed_testcase = testcase.parse_content_with_bindings(
testcase_template, variables_binds, functions_binds)
self.assertEqual(
parsed_testcase["url"],
"http://127.0.0.1:5000/api/users/%s" % variables_binds["uid"]
)
self.assertEqual(
parsed_testcase["headers"]["authorization"],
variables_binds["authorization"]
)
self.assertEqual(
parsed_testcase["headers"]["random"],
variables_binds["random"]
)
self.assertEqual(
parsed_testcase["body"],
variables_binds["data"]
)
self.assertEqual(
parsed_testcase["headers"]["sum"],
3
)
