Coverage for tests / test_validation.py: 100%

1import pytest 8 ctx1ijklmnop

2from pydantic.error_wrappers import ValidationError 8 ctx1ijklmnop

3from sqlmodel import SQLModel 8 ctx1ijklmnop

4 

5 

6def test_validation_pydantic_v2(clear_sqlmodel): 8 ctx1ijklmnop

7 """Test validation of implicit and explicit None values. 

8 

9 # For consistency with pydantic, validators are not to be called on 

10 # arguments that are not explicitly provided. 

11 

12 https://github.com/tiangolo/sqlmodel/issues/230 

13 https://github.com/samuelcolvin/pydantic/issues/1223 

14 

15 """ 

16 from pydantic import field_validator 8 ctx1abcdefgh

17 

18 class Hero(SQLModel): 8 ctx1abcdefgh

19 name: str | None = None 8 ctx1abcdefgh

20 secret_name: str | None = None 8 ctx1abcdefgh

21 age: int | None = None 8 ctx1abcdefgh

22 

23 @field_validator("name", "secret_name", "age") 8 ctx1abcdefgh

24 def reject_none(cls, v): 8 ctx1abcdefgh

25 assert v is not None 8 ctx1abcdefgh

26 return v 8 ctx1abcdefgh

27 

28 Hero.model_validate({"age": 25}) 8 ctx1abcdefgh

29 

30 with pytest.raises(ValidationError): 8 ctx1abcdefgh

31 Hero.model_validate({"name": None, "age": 25}) 8 ctx1abcdefgh