# !/usr/bin/python3
|
# -*- coding: utf-8 -*-
|
"""
|
Built-in dependent functions used in YAML/JSON testcases.
|
"""
|
|
import re
|
|
import pydash
|
|
from httprunner.compat import basestring, builtin_str, integer_types
|
|
"""
|
built-in comparators
|
"""
|
|
|
def equals(check_value, expect_value):
|
assert check_value == expect_value
|
|
|
def less_than(check_value, expect_value):
|
assert check_value < expect_value
|
|
|
def less_than_or_equals(check_value, expect_value):
|
assert check_value <= expect_value
|
|
|
def greater_than(check_value, expect_value):
|
assert check_value > expect_value
|
|
|
def greater_than_or_equals(check_value, expect_value):
|
assert check_value >= expect_value
|
|
|
def not_equals(check_value, expect_value):
|
assert check_value != expect_value
|
|
|
def string_equals(check_value, expect_value):
|
assert builtin_str(check_value) == builtin_str(expect_value)
|
|
|
def length_equals(check_value, expect_value):
|
assert isinstance(expect_value, integer_types)
|
assert len(check_value) == expect_value
|
|
|
def length_greater_than(check_value, expect_value):
|
assert isinstance(expect_value, integer_types)
|
assert len(check_value) > expect_value
|
|
|
def length_greater_than_or_equals(check_value, expect_value):
|
assert isinstance(expect_value, integer_types)
|
assert len(check_value) >= expect_value
|
|
|
def length_less_than(check_value, expect_value):
|
assert isinstance(expect_value, integer_types)
|
assert len(check_value) < expect_value
|
|
|
def length_less_than_or_equals(check_value, expect_value):
|
assert isinstance(expect_value, integer_types)
|
assert len(check_value) <= expect_value
|
|
|
def contains(check_value, expect_value):
|
assert isinstance(check_value, (list, tuple, dict, basestring))
|
assert expect_value in check_value
|
|
|
def not_contains(check_value, expect_value):
|
assert isinstance(check_value, (list, tuple, dict, basestring))
|
assert expect_value not in check_value
|
|
|
def contained_by(check_value, expect_value):
|
assert isinstance(expect_value, (list, tuple, dict, basestring))
|
assert check_value in expect_value
|
|
|
def _get_expression(item, expression, expect_value, jsonpath):
|
parsed_expression = None
|
if isinstance(item, dict):
|
item_value = pydash.get(item, jsonpath)
|
|
if isinstance(item_value, (int, float, list, dict, bool, type(None))):
|
parsed_expression = f"{item_value} {expression} {expect_value}"
|
else:
|
parsed_expression = (
|
f"'{pydash.get(item, jsonpath)}' {expression} '{expect_value}'"
|
)
|
|
if isinstance(item, str):
|
parsed_expression = f"{item} {expression} {expect_value}"
|
|
if parsed_expression is None:
|
raise AssertionError(f"list的元素只能是dict或者string")
|
|
return parsed_expression
|
|
|
def list_any_item_contains(check_value: list, jsonpath_expression_value):
|
assert isinstance(check_value, list)
|
jsonpath, expression, expect_value = jsonpath_expression_value.split(" ")
|
for item in check_value:
|
parsed_expression = _get_expression(
|
item=item,
|
expression=expression,
|
expect_value=expect_value,
|
jsonpath=jsonpath,
|
)
|
try:
|
if eval(parsed_expression) is True:
|
break
|
except Exception as e:
|
raise e
|
else:
|
raise AssertionError(f"{check_value} {expression} {expect_value}")
|
|
|
def list_all_item_contains(check_value: list, jsonpath_expression_value):
|
assert isinstance(check_value, list)
|
jsonpath, expression, expect_value = jsonpath_expression_value.split(" ")
|
for item in check_value:
|
parsed_expression = _get_expression(
|
item=item,
|
expression=expression,
|
expect_value=expect_value,
|
jsonpath=jsonpath,
|
)
|
try:
|
if eval(parsed_expression) is False:
|
raise AssertionError(f"{check_value} {expression} {expect_value}")
|
except Exception as e:
|
raise e
|
|
|
def type_match(check_value, expect_value):
|
def get_type(name):
|
if isinstance(name, type):
|
return name
|
elif isinstance(name, basestring):
|
try:
|
return __builtins__[name]
|
except KeyError:
|
raise ValueError(name)
|
else:
|
raise ValueError(name)
|
|
assert isinstance(check_value, get_type(expect_value))
|
|
|
def regex_match(check_value, expect_value):
|
assert isinstance(expect_value, basestring)
|
assert isinstance(check_value, basestring)
|
assert re.match(expect_value, check_value)
|
|
|
def startswith(check_value, expect_value):
|
assert builtin_str(check_value).startswith(builtin_str(expect_value))
|
|
|
def endswith(check_value, expect_value):
|
assert builtin_str(check_value).endswith(builtin_str(expect_value))
|
