1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
|
import pytest
from pydantic_core import PydanticOmit, SchemaError, SchemaValidator, ValidationError, core_schema
from pydantic_core import core_schema as cs
from .conftest import PyAndJson
def test_isinstance():
v = SchemaValidator(cs.int_schema())
assert v.validate_python(123) == 123
assert v.isinstance_python(123) is True
assert v.validate_python('123') == 123
assert v.isinstance_python('123') is True
with pytest.raises(ValidationError, match='Input should be a valid integer'):
v.validate_python('foo')
assert v.isinstance_python('foo') is False
def test_isinstance_strict():
v = SchemaValidator(cs.int_schema(strict=True))
assert v.validate_python(123) == 123
assert v.isinstance_python(123) is True
with pytest.raises(ValidationError, match='Input should be a valid integer'):
v.validate_python('123')
assert v.isinstance_python('123') is False
def test_isinstance_forbid_extra_fn_override():
v = SchemaValidator(cs.typed_dict_schema({'f': cs.typed_dict_field(cs.str_schema())}))
with pytest.raises(ValidationError, match='Extra inputs are not permitted'):
v.validate_python({'f': 'x', 'extra_field': '123'}, extra='forbid')
assert v.isinstance_python({'f': 'x', 'extra_field': '123'}, extra='forbid') is False
def test_internal_error():
v = SchemaValidator(
cs.model_schema(cls=int, schema=cs.model_fields_schema(fields={'f': cs.model_field(schema=cs.int_schema())}))
)
with pytest.raises(AttributeError, match="'int' object has no attribute '__dict__'"):
v.validate_python({'f': 123})
with pytest.raises(AttributeError, match="'int' object has no attribute '__dict__'"):
v.validate_json('{"f": 123}')
with pytest.raises(AttributeError, match="'int' object has no attribute '__dict__'"):
v.isinstance_python({'f': 123})
def test_omit(py_and_json: PyAndJson):
def omit(v, info):
if v == 'omit':
raise PydanticOmit
elif v == 'error':
raise ValueError('error')
else:
return v
v = py_and_json(core_schema.with_info_plain_validator_function(omit))
assert v.validate_test('foo') == 'foo'
if v.validator_type == 'python':
assert v.isinstance_test('foo') is True
if v.validator_type == 'python':
assert v.isinstance_test('error') is False
with pytest.raises(SchemaError, match='Uncaught Omit error, please check your usage of `default` validators.'):
v.validate_test('omit')
|
