def test_repeat_as_as_any_of():
with pytest.raises(QueryValidatorStructureError):
Query.has("test").as_(str).as_any_of({int, str})
with pytest.raises(QueryValidatorStructureError):
Query.has("test").as_any_of({int, str}).as_(str)
with pytest.raises(QueryValidatorStructureError):
Query.can_have("test").as_(str).as_any_of({int, str})
with pytest.raises(QueryValidatorStructureError):
Query.can_have("test").as_any_of({int, str}).as_(str)
def test_default_supplier():
x = 0
def supplier(query, context):
nonlocal x
x += 1
return "test"
valid = Query.can_have("test").with_default(supplier, str)
query = {"test": "dog"}
assert valid(query)
assert query == {"test": "dog"}
assert x == 0
with pytest.raises(QueryValidationError):
valid({"test": 1})
assert x == 0
query = {}
assert valid(query)
assert query == {"test": "test"}
assert x == 1
query = {}
assert valid(query)
assert query == {"test": "test"}
assert x == 2
def test_can_have_as_any_of_and():
valid = Query.can_have("test").as_any_of({str, int}).and_("dog").as_any_of(
{str, int})
with pytest.raises(QueryValidationError):
valid({"test": 0.0})
with pytest.raises(QueryValidationError):
valid({"test": 0})
with pytest.raises(QueryValidationError):
valid({"test": "test"})
with pytest.raises(QueryValidationError):
valid({"dog": 0.0})
with pytest.raises(QueryValidationError):
valid({"dog": 0})
with pytest.raises(QueryValidationError):
valid({"dog": "cat"})
with pytest.raises(QueryValidationError):
valid({"test": 0.0, "dog": 0.0})
assert valid({})
assert valid({"test": "test", "dog": "cat"})
assert valid({"test": "test", "dog": 0})
assert valid({"test": 0, "dog": "cat"})
assert valid({"test": 0, "dog": 0})
def test_can_have():
valid = Query.can_have("test")
assert valid({})
assert valid({"dog": "cat"})
assert valid({"test": "test"})
assert valid({"test": 0})
assert valid({"test": "test", "dog": "cat"})
def test_can_have_or():
valid = Query.can_have("test").or_("dog").or_("foo")
assert valid({})
assert valid({"test": "test"})
assert valid({"dog": "cat"})
assert valid({"foo": "bar"})
assert valid({"test": 0})
assert valid({"test": "test", "dog": "cat", "foo": "bar"})
def test_can_have_as():
valid = Query.can_have("test").as_(str)
with pytest.raises(QueryValidationError):
valid({"test": 0})
assert valid({})
assert valid({"dog": "cat"})
assert valid({"test": "test"})
assert valid({"test": "test", "dog": "cat"})
def test_can_have_nested_or_and():
valid = Query.can_have("test").or_("cat").and_("dog")
with pytest.raises(QueryValidationError):
valid({"dog": "cat"})
with pytest.raises(QueryValidationError):
valid({"cat": "dog"})
assert valid({})
assert valid({"test": "test"})
assert valid({"dog": "cat", "cat": "dog"})
assert valid({"test": "test", "foo": "bar"})
assert valid({"test": "test", "cat": "dog"})
assert valid({"test": "test", "dog": "cat"})
assert valid({"test": "test", "dog": "cat", "cat": "dog"})
def test_no_default_type():
valid = Query.can_have("test").with_default("test")
query = {"test": "dog"}
assert valid(query)
assert query == {"test": "dog"}
def test_wrong_default_type():
valid = Query.can_have("test").with_default("test")
with pytest.raises(QueryValidationError):
valid({"test": 1})
def test_default():
valid = Query.can_have("test").with_default("test")
query = {}
assert valid(query)
assert query == {"test": "test"}
def test_repeat_have_can_have():
with pytest.raises(QueryValidatorStructureError):
Query.has("test").can_have("dog")
with pytest.raises(QueryValidatorStructureError):
Query.can_have("test").has("dog")
