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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
import pytest
from pydantic_core import SchemaValidator, ValidationError, core_schema
from .conftest import PyAndJson
def test_after(py_and_json: PyAndJson):
def f(input_value, info):
return input_value + f'| context: {info.context}'
v = py_and_json(core_schema.with_info_after_validator_function(f, core_schema.str_schema()))
assert v.validate_test('foobar') == 'foobar| context: None'
assert v.validate_test('foobar', None, {1: 10}) == 'foobar| context: {1: 10}'
assert v.validate_test('foobar', None, 'frogspawn') == 'foobar| context: frogspawn'
def test_mutable_context(py_and_json: PyAndJson):
def f(input_value, info):
info.context['foo'] = input_value
return input_value
v = py_and_json(core_schema.with_info_before_validator_function(f, core_schema.str_schema()))
mutable_context = {}
assert v.validate_test('foobar', None, mutable_context) == 'foobar'
assert mutable_context == {'foo': 'foobar'}
def test_typed_dict(py_and_json: PyAndJson):
def f1(input_value, info):
info.context['f1'] = input_value
return input_value + f'| context: {info.context}'
def f2(input_value, info):
info.context['f2'] = input_value
return input_value + f'| context: {info.context}'
v = py_and_json(
core_schema.typed_dict_schema(
{
'f1': core_schema.typed_dict_field(core_schema.with_info_plain_validator_function(f1)),
'f2': core_schema.typed_dict_field(core_schema.with_info_plain_validator_function(f2)),
}
)
)
assert v.validate_test({'f1': '1', 'f2': '2'}, None, {'x': 'y'}) == {
'f1': "1| context: {'x': 'y', 'f1': '1'}",
'f2': "2| context: {'x': 'y', 'f1': '1', 'f2': '2'}",
}
def test_wrap(py_and_json: PyAndJson):
def f(input_value, validator, info):
return validator(input_value) + f'| context: {info.context}'
v = py_and_json(core_schema.with_info_wrap_validator_function(f, core_schema.str_schema()))
assert v.validate_test('foobar') == 'foobar| context: None'
assert v.validate_test('foobar', None, {1: 10}) == 'foobar| context: {1: 10}'
assert v.validate_test('foobar', None, 'frogspawn') == 'foobar| context: frogspawn'
def test_isinstance(py_and_json: PyAndJson):
def f(input_value, validator, info):
if 'error' in info.context:
raise ValueError('wrong')
return validator(input_value)
v = py_and_json(core_schema.with_info_wrap_validator_function(f, core_schema.str_schema()))
assert v.validate_python('foobar', None, {}) == 'foobar'
with pytest.raises(TypeError):
v.validate_test('foobar')
with pytest.raises(TypeError):
v.isinstance_test('foobar')
with pytest.raises(ValidationError, match=r'Value error, wrong \[type=value_error,'):
v.validate_test('foobar', None, {'error'})
assert v.isinstance_test('foobar', None, {}) is True
with pytest.raises(TypeError):
v.isinstance_test('foobar')
assert v.isinstance_test('foobar', None, {'error'}) is False
def test_validate_assignment_with_context():
def f1(input_value, info):
info.context['f1'] = input_value
return input_value + f'| context: {info.context}'
def f2(input_value, info):
info.context['f2'] = input_value
return input_value + f'| context: {info.context}'
v = SchemaValidator(
core_schema.model_fields_schema(
{
'f1': core_schema.model_field(core_schema.with_info_plain_validator_function(f1)),
'f2': core_schema.model_field(core_schema.with_info_plain_validator_function(f2)),
}
)
)
m1, model_extra, fields_set = v.validate_python({'f1': '1', 'f2': '2'}, strict=None, context={'x': 'y'})
assert m1 == {'f1': "1| context: {'x': 'y', 'f1': '1'}", 'f2': "2| context: {'x': 'y', 'f1': '1', 'f2': '2'}"}
assert model_extra is None
assert fields_set == {'f1', 'f2'}
m2, model_extra, fields_set = v.validate_assignment(m1, 'f1', '3', context={'x': 'y'})
assert m2 == {'f1': "3| context: {'x': 'y', 'f1': '3'}", 'f2': "2| context: {'x': 'y', 'f1': '1', 'f2': '2'}"}
assert model_extra is None
assert fields_set == {'f1'}
|
