mirror of
https://github.com/httprunner/httprunner.git
synced 2026-05-12 02:21:29 +08:00
90 lines
2.9 KiB
Python
90 lines
2.9 KiB
Python
import unittest
|
|
|
|
import requests
|
|
|
|
from httprunner.parser import Parser
|
|
from httprunner.response import ResponseObject, uniform_validator
|
|
|
|
|
|
class TestResponse(unittest.TestCase):
|
|
def setUp(self) -> None:
|
|
resp = requests.post(
|
|
"https://httpbin.org/anything",
|
|
json={
|
|
"locations": [
|
|
{"name": "Seattle", "state": "WA"},
|
|
{"name": "New York", "state": "NY"},
|
|
{"name": "Bellevue", "state": "WA"},
|
|
{"name": "Olympia", "state": "WA"},
|
|
]
|
|
},
|
|
)
|
|
parser = Parser(
|
|
functions_mapping={"get_name": lambda: "name", "get_num": lambda x: x}
|
|
)
|
|
self.resp_obj = ResponseObject(resp, parser)
|
|
|
|
def test_extract(self):
|
|
variables_mapping = {"body": "body"}
|
|
extract_mapping = self.resp_obj.extract(
|
|
{
|
|
"var_1": "body.json.locations[0]",
|
|
"var_2": "body.json.locations[3].name",
|
|
"var_3": "$body.json.locations[3].name",
|
|
"var_4": "$body.json.locations[3].${get_name()}",
|
|
},
|
|
variables_mapping=variables_mapping,
|
|
)
|
|
self.assertEqual(extract_mapping["var_1"], {"name": "Seattle", "state": "WA"})
|
|
self.assertEqual(extract_mapping["var_2"], "Olympia")
|
|
self.assertEqual(extract_mapping["var_3"], "Olympia")
|
|
self.assertEqual(extract_mapping["var_4"], "Olympia")
|
|
|
|
def test_validate(self):
|
|
self.resp_obj.validate(
|
|
[
|
|
{"eq": ["body.json.locations[0].name", "Seattle"]},
|
|
{"eq": ["body.json.locations[0]", {"name": "Seattle", "state": "WA"}]},
|
|
],
|
|
)
|
|
|
|
def test_validate_variables(self):
|
|
variables_mapping = {"index": 1, "var_empty": ""}
|
|
self.resp_obj.validate(
|
|
[
|
|
{"eq": ["body.json.locations[$index].name", "New York"]},
|
|
{"eq": ["$var_empty", ""]},
|
|
],
|
|
variables_mapping=variables_mapping,
|
|
)
|
|
|
|
def test_validate_functions(self):
|
|
variables_mapping = {"index": 1}
|
|
self.resp_obj.validate(
|
|
[
|
|
{"eq": ["${get_num(0)}", 0]},
|
|
{"eq": ["${get_num($index)}", 1]},
|
|
],
|
|
variables_mapping=variables_mapping,
|
|
)
|
|
|
|
def test_uniform_validator(self):
|
|
validators = [
|
|
{
|
|
"check": "status_code",
|
|
"comparator": "eq",
|
|
"expect": 201,
|
|
"message": "test",
|
|
},
|
|
{"check": "status_code", "assert": "eq", "expect": 201, "msg": "test"},
|
|
{"eq": ["status_code", 201, "test"]},
|
|
]
|
|
expected = {
|
|
"check": "status_code",
|
|
"assert": "equal",
|
|
"expect": 201,
|
|
"message": "test",
|
|
}
|
|
for validator in validators:
|
|
self.assertEqual(uniform_validator(validator), expected)
|