mirror of
https://github.com/httprunner/httprunner.git
synced 2026-05-12 11:29:48 +08:00
104 lines
4.5 KiB
Python
104 lines
4.5 KiB
Python
import unittest
|
|
|
|
from httprunner import validator
|
|
|
|
|
|
class TestValidator(unittest.TestCase):
|
|
|
|
def test_get_uniform_comparator(self):
|
|
self.assertEqual(validator.get_uniform_comparator("eq"), "equals")
|
|
self.assertEqual(validator.get_uniform_comparator("=="), "equals")
|
|
self.assertEqual(validator.get_uniform_comparator("lt"), "less_than")
|
|
self.assertEqual(validator.get_uniform_comparator("le"), "less_than_or_equals")
|
|
self.assertEqual(validator.get_uniform_comparator("gt"), "greater_than")
|
|
self.assertEqual(validator.get_uniform_comparator("ge"), "greater_than_or_equals")
|
|
self.assertEqual(validator.get_uniform_comparator("ne"), "not_equals")
|
|
|
|
self.assertEqual(validator.get_uniform_comparator("str_eq"), "string_equals")
|
|
self.assertEqual(validator.get_uniform_comparator("len_eq"), "length_equals")
|
|
self.assertEqual(validator.get_uniform_comparator("count_eq"), "length_equals")
|
|
|
|
self.assertEqual(validator.get_uniform_comparator("len_gt"), "length_greater_than")
|
|
self.assertEqual(validator.get_uniform_comparator("count_gt"), "length_greater_than")
|
|
self.assertEqual(validator.get_uniform_comparator("count_greater_than"), "length_greater_than")
|
|
|
|
self.assertEqual(validator.get_uniform_comparator("len_ge"), "length_greater_than_or_equals")
|
|
self.assertEqual(validator.get_uniform_comparator("count_ge"), "length_greater_than_or_equals")
|
|
self.assertEqual(validator.get_uniform_comparator("count_greater_than_or_equals"), "length_greater_than_or_equals")
|
|
|
|
self.assertEqual(validator.get_uniform_comparator("len_lt"), "length_less_than")
|
|
self.assertEqual(validator.get_uniform_comparator("count_lt"), "length_less_than")
|
|
self.assertEqual(validator.get_uniform_comparator("count_less_than"), "length_less_than")
|
|
|
|
self.assertEqual(validator.get_uniform_comparator("len_le"), "length_less_than_or_equals")
|
|
self.assertEqual(validator.get_uniform_comparator("count_le"), "length_less_than_or_equals")
|
|
self.assertEqual(validator.get_uniform_comparator("count_less_than_or_equals"), "length_less_than_or_equals")
|
|
|
|
def test_parse_validator(self):
|
|
_validator = {"check": "status_code", "comparator": "eq", "expect": 201}
|
|
self.assertEqual(
|
|
validator.uniform_validator(_validator),
|
|
{"check": "status_code", "comparator": "equals", "expect": 201}
|
|
)
|
|
|
|
_validator = {'eq': ['status_code', 201]}
|
|
self.assertEqual(
|
|
validator.uniform_validator(_validator),
|
|
{"check": "status_code", "comparator": "equals", "expect": 201}
|
|
)
|
|
|
|
def test_extend_validators(self):
|
|
def_validators = [
|
|
{'eq': ['v1', 200]},
|
|
{"check": "s2", "expect": 16, "comparator": "len_eq"}
|
|
]
|
|
current_validators = [
|
|
{"check": "v1", "expect": 201},
|
|
{'len_eq': ['s3', 12]}
|
|
]
|
|
def_validators = [
|
|
validator.uniform_validator(_validator)
|
|
for _validator in def_validators
|
|
]
|
|
ref_validators = [
|
|
validator.uniform_validator(_validator)
|
|
for _validator in current_validators
|
|
]
|
|
|
|
extended_validators = validator.extend_validators(def_validators, ref_validators)
|
|
self.assertIn(
|
|
{"check": "v1", "expect": 201, "comparator": "equals"},
|
|
extended_validators
|
|
)
|
|
self.assertIn(
|
|
{"check": "s2", "expect": 16, "comparator": "length_equals"},
|
|
extended_validators
|
|
)
|
|
self.assertIn(
|
|
{"check": "s3", "expect": 12, "comparator": "length_equals"},
|
|
extended_validators
|
|
)
|
|
|
|
def test_extend_validators_with_dict(self):
|
|
def_validators = [
|
|
{'eq': ["a", {"v": 1}]},
|
|
{'eq': [{"b": 1}, 200]}
|
|
]
|
|
current_validators = [
|
|
{'len_eq': ['s3', 12]},
|
|
{'eq': [{"b": 1}, 201]}
|
|
]
|
|
def_validators = [
|
|
validator.uniform_validator(_validator)
|
|
for _validator in def_validators
|
|
]
|
|
ref_validators = [
|
|
validator.uniform_validator(_validator)
|
|
for _validator in current_validators
|
|
]
|
|
|
|
extended_validators = validator.extend_validators(def_validators, ref_validators)
|
|
self.assertEqual(len(extended_validators), 3)
|
|
self.assertIn({'check': {'b': 1}, 'expect': 201, 'comparator': 'equals'}, extended_validators)
|
|
self.assertNotIn({'check': {'b': 1}, 'expect': 200, 'comparator': 'equals'}, extended_validators)
|