Skip to content

Commit a836948

Browse files
committed
update test
1 parent 2ac82ca commit a836948

File tree

1 file changed

+12
-9
lines changed

1 file changed

+12
-9
lines changed

tests/serializers/test_model.py

+12-9
Original file line numberDiff line numberDiff line change
@@ -211,19 +211,22 @@ def test_exclude_if():
211211
{
212212
'a': core_schema.model_field(core_schema.int_schema(), exclude_if=lambda x: x > 1),
213213
'b': core_schema.model_field(core_schema.str_schema(), exclude_if=lambda x: 'foo' in x),
214+
'c': core_schema.model_field(
215+
core_schema.str_schema(), serialization_exclude=True, exclude_if=lambda x: 'foo' in x
216+
),
214217
}
215218
),
216219
)
217220
)
218-
assert s.to_python(BasicModel(a=0, b='bar')) == {'a': 0, 'b': 'bar'}
219-
assert s.to_python(BasicModel(a=2, b='bar')) == {'b': 'bar'}
220-
assert s.to_python(BasicModel(a=0, b='foo')) == {'a': 0}
221-
assert s.to_python(BasicModel(a=2, b='foo')) == {}
222-
223-
assert s.to_json(BasicModel(a=0, b='bar')) == b'{"a":0,"b":"bar"}'
224-
assert s.to_json(BasicModel(a=2, b='bar')) == b'{"b":"bar"}'
225-
assert s.to_json(BasicModel(a=0, b='foo')) == b'{"a":0}'
226-
assert s.to_json(BasicModel(a=2, b='foo')) == b'{}'
221+
assert s.to_python(BasicModel(a=0, b='bar', c='bar')) == {'a': 0, 'b': 'bar'}
222+
assert s.to_python(BasicModel(a=2, b='bar', c='bar')) == {'b': 'bar'}
223+
assert s.to_python(BasicModel(a=0, b='foo', c='bar')) == {'a': 0}
224+
assert s.to_python(BasicModel(a=2, b='foo', c='bar')) == {}
225+
226+
assert s.to_json(BasicModel(a=0, b='bar', c='bar')) == b'{"a":0,"b":"bar"}'
227+
assert s.to_json(BasicModel(a=2, b='bar', c='bar')) == b'{"b":"bar"}'
228+
assert s.to_json(BasicModel(a=0, b='foo', c='bar')) == b'{"a":0}'
229+
assert s.to_json(BasicModel(a=2, b='foo', c='bar')) == b'{}'
227230

228231

229232
def test_alias():

0 commit comments

Comments
 (0)