mirror of
https://github.com/httprunner/httprunner.git
synced 2026-05-30 12:59:39 +08:00
rename test folder to tests
This commit is contained in:
httprunner
218
tests/test_context.py
Normal file
218
tests/test_context.py
Normal file
@@ -0,0 +1,218 @@
|
||||
import os
|
||||
import unittest
|
||||
|
||||
from ate import utils, runner
|
||||
from ate.context import Context
|
||||
|
||||
|
||||
class VariableBindsUnittest(unittest.TestCase):
|
||||
|
||||
def setUp(self):
|
||||
self.context = Context()
|
||||
testcase_file_path = os.path.join(os.getcwd(), 'tests/data/demo_binds.yml')
|
||||
self.testcases = utils.load_testcases(testcase_file_path)
|
||||
|
||||
def test_context_register_variables(self):
|
||||
# testcase in JSON format
|
||||
testcase1 = {
|
||||
"variable_binds": [
|
||||
{"TOKEN": "debugtalk"},
|
||||
{"var": [1, 2, 3]},
|
||||
{"data": {'name': 'user', 'password': '123456'}}
|
||||
]
|
||||
}
|
||||
# testcase in YAML format
|
||||
testcase2 = self.testcases["register_variables"]
|
||||
|
||||
for testcase in [testcase1, testcase2]:
|
||||
variable_binds = testcase['variable_binds']
|
||||
self.context.register_variables_config(variable_binds)
|
||||
|
||||
context_variables = self.context._get_evaluated_testcase_variables()
|
||||
self.assertIn("TOKEN", context_variables)
|
||||
self.assertEqual(context_variables["TOKEN"], "debugtalk")
|
||||
self.assertIn("var", context_variables)
|
||||
self.assertEqual(context_variables["var"], [1, 2, 3])
|
||||
self.assertIn("data", context_variables)
|
||||
self.assertEqual(
|
||||
context_variables["data"],
|
||||
{'name': 'user', 'password': '123456'}
|
||||
)
|
||||
|
||||
def test_context_register_template_variables(self):
|
||||
testcase1 = {
|
||||
"variable_binds": [
|
||||
{"GLOBAL_TOKEN": "debugtalk"},
|
||||
{"token": "$GLOBAL_TOKEN"}
|
||||
]
|
||||
}
|
||||
testcase2 = self.testcases["register_template_variables"]
|
||||
|
||||
for testcase in [testcase1, testcase2]:
|
||||
variable_binds = testcase['variable_binds']
|
||||
self.context.register_variables_config(variable_binds)
|
||||
|
||||
context_variables = self.context._get_evaluated_testcase_variables()
|
||||
self.assertIn("GLOBAL_TOKEN", context_variables)
|
||||
self.assertEqual(context_variables["GLOBAL_TOKEN"], "debugtalk")
|
||||
self.assertIn("token", context_variables)
|
||||
self.assertEqual(context_variables["token"], "debugtalk")
|
||||
|
||||
def test_context_bind_lambda_functions(self):
|
||||
testcase1 = {
|
||||
"function_binds": {
|
||||
"add_one": lambda x: x + 1,
|
||||
"add_two_nums": lambda x, y: x + y
|
||||
},
|
||||
"variable_binds": [
|
||||
{"add1": "${add_one(2)}"},
|
||||
{"sum2nums": "${add_two_nums(2,3)}"}
|
||||
]
|
||||
}
|
||||
testcase2 = self.testcases["bind_lambda_functions"]
|
||||
|
||||
for testcase in [testcase1, testcase2]:
|
||||
function_binds = testcase.get('function_binds', {})
|
||||
self.context.bind_functions(function_binds)
|
||||
|
||||
variable_binds = testcase['variable_binds']
|
||||
self.context.register_variables_config(variable_binds)
|
||||
|
||||
context_variables = self.context._get_evaluated_testcase_variables()
|
||||
self.assertIn("add1", context_variables)
|
||||
self.assertEqual(context_variables["add1"], 3)
|
||||
self.assertIn("sum2nums", context_variables)
|
||||
self.assertEqual(context_variables["sum2nums"], 5)
|
||||
|
||||
def test_context_bind_lambda_functions_with_import(self):
|
||||
testcase1 = {
|
||||
"requires": ["random", "string", "hashlib"],
|
||||
"function_binds": {
|
||||
"gen_random_string": "lambda str_len: ''.join(random.choice(string.ascii_letters + string.digits) for _ in range(str_len))",
|
||||
"gen_md5": "lambda *str_args: hashlib.md5(''.join(str_args).encode('utf-8')).hexdigest()"
|
||||
},
|
||||
"variable_binds": [
|
||||
{"TOKEN": "debugtalk"},
|
||||
{"random": "${gen_random_string(5)}"},
|
||||
{"data": '{"name": "user", "password": "123456"}'},
|
||||
{"authorization": "${gen_md5($TOKEN, $data, $random)}"}
|
||||
]
|
||||
}
|
||||
testcase2 = self.testcases["bind_lambda_functions_with_import"]
|
||||
|
||||
for testcase in [testcase1, testcase2]:
|
||||
requires = testcase.get('requires', [])
|
||||
self.context.import_requires(requires)
|
||||
|
||||
function_binds = testcase.get('function_binds', {})
|
||||
self.context.bind_functions(function_binds)
|
||||
|
||||
variable_binds = testcase['variable_binds']
|
||||
self.context.register_variables_config(variable_binds)
|
||||
context_variables = self.context._get_evaluated_testcase_variables()
|
||||
|
||||
self.assertIn("TOKEN", context_variables)
|
||||
TOKEN = context_variables["TOKEN"]
|
||||
self.assertEqual(TOKEN, "debugtalk")
|
||||
self.assertIn("random", context_variables)
|
||||
self.assertIsInstance(context_variables["random"], str)
|
||||
self.assertEqual(len(context_variables["random"]), 5)
|
||||
random = context_variables["random"]
|
||||
self.assertIn("data", context_variables)
|
||||
data = context_variables["data"]
|
||||
self.assertIn("authorization", context_variables)
|
||||
self.assertEqual(len(context_variables["authorization"]), 32)
|
||||
authorization = context_variables["authorization"]
|
||||
self.assertEqual(utils.gen_md5(TOKEN, data, random), authorization)
|
||||
|
||||
def test_import_module_functions(self):
|
||||
testcase1 = {
|
||||
"import_module_functions": ["tests.data.custom_functions"],
|
||||
"variable_binds": [
|
||||
{"TOKEN": "debugtalk"},
|
||||
{"random": "${gen_random_string(5)}"},
|
||||
{"data": '{"name": "user", "password": "123456"}'},
|
||||
{"authorization": "${gen_md5($TOKEN, $data, $random)}"}
|
||||
]
|
||||
}
|
||||
testcase2 = self.testcases["bind_module_functions"]
|
||||
|
||||
for testcase in [testcase1, testcase2]:
|
||||
module_functions = testcase.get('import_module_functions', [])
|
||||
self.context.import_module_functions(module_functions)
|
||||
|
||||
variable_binds = testcase['variable_binds']
|
||||
self.context.register_variables_config(variable_binds)
|
||||
context_variables = self.context._get_evaluated_testcase_variables()
|
||||
|
||||
self.assertIn("TOKEN", context_variables)
|
||||
TOKEN = context_variables["TOKEN"]
|
||||
self.assertEqual(TOKEN, "debugtalk")
|
||||
self.assertIn("random", context_variables)
|
||||
self.assertIsInstance(context_variables["random"], str)
|
||||
self.assertEqual(len(context_variables["random"]), 5)
|
||||
random = context_variables["random"]
|
||||
self.assertIn("data", context_variables)
|
||||
data = context_variables["data"]
|
||||
self.assertIn("authorization", context_variables)
|
||||
self.assertEqual(len(context_variables["authorization"]), 32)
|
||||
authorization = context_variables["authorization"]
|
||||
self.assertEqual(utils.gen_md5(TOKEN, data, random), authorization)
|
||||
|
||||
def test_get_parsed_request(self):
|
||||
test_runner = runner.Runner()
|
||||
testcase = {
|
||||
"import_module_functions": ["tests.data.custom_functions"],
|
||||
"variable_binds": [
|
||||
{"TOKEN": "debugtalk"},
|
||||
{"random": "${gen_random_string(5)}"},
|
||||
{"data": '{"name": "user", "password": "123456"}'},
|
||||
{"authorization": "${gen_md5($TOKEN, $data, $random)}"}
|
||||
],
|
||||
"request": {
|
||||
"url": "http://127.0.0.1:5000/api/users/1000",
|
||||
"method": "POST",
|
||||
"headers": {
|
||||
"Content-Type": "application/json",
|
||||
"authorization": "$authorization",
|
||||
"random": "$random"
|
||||
},
|
||||
"data": "$data"
|
||||
}
|
||||
}
|
||||
test_runner.init_config(testcase, level="testcase")
|
||||
parsed_request = test_runner.context.get_parsed_request()
|
||||
self.assertIn("authorization", parsed_request["headers"])
|
||||
self.assertEqual(len(parsed_request["headers"]["authorization"]), 32)
|
||||
self.assertIn("random", parsed_request["headers"])
|
||||
self.assertEqual(len(parsed_request["headers"]["random"]), 5)
|
||||
self.assertIn("data", parsed_request)
|
||||
self.assertEqual(parsed_request["data"], testcase["variable_binds"][2]["data"])
|
||||
|
||||
def test_get_eval_value(self):
|
||||
self.context.testcase_variables_mapping = {
|
||||
"str_1": "str_value1",
|
||||
"str_2": "str_value2"
|
||||
}
|
||||
self.assertEqual(self.context.get_eval_value("$str_1"), "str_value1")
|
||||
self.assertEqual(self.context.get_eval_value("$str_2"), "str_value2")
|
||||
self.assertEqual(
|
||||
self.context.get_eval_value(["$str_1", "str3"]),
|
||||
["str_value1", "str3"]
|
||||
)
|
||||
self.assertEqual(
|
||||
self.context.get_eval_value({"key": "$str_1"}),
|
||||
{"key": "str_value1"}
|
||||
)
|
||||
|
||||
import random, string
|
||||
self.context.testcase_config["functions"]["gen_random_string"] = \
|
||||
lambda str_len: ''.join(random.choice(string.ascii_letters + string.digits) \
|
||||
for _ in range(str_len))
|
||||
result = self.context.get_eval_value("${gen_random_string(5)}")
|
||||
self.assertEqual(len(result), 5)
|
||||
|
||||
add_two_nums = lambda a, b=1: a + b
|
||||
self.context.testcase_config["functions"]["add_two_nums"] = add_two_nums
|
||||
self.assertEqual(self.context.get_eval_value("${add_two_nums(1)}"), 2)
|
||||
self.assertEqual(self.context.get_eval_value("${add_two_nums(1, 2)}"), 3)
|
||||
Reference in New Issue
Block a user