Coverage for tests / test_validation.py: 100%

1from typing import Optional 9 ctx1jklmnopqr

2 

3import pytest 9 ctx1jklmnopqr

4from pydantic.error_wrappers import ValidationError 9 ctx1jklmnopqr

5from sqlmodel import SQLModel 9 ctx1jklmnopqr

6 

7 

8def test_validation_pydantic_v2(clear_sqlmodel): 9 ctx1jklmnopqr

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

10 

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

12 # arguments that are not explicitly provided. 

13 

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

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

16 

17 """ 

18 from pydantic import field_validator 9 ctx1abcdefghi

19 

20 class Hero(SQLModel): 9 ctx1abcdefghi

21 name: Optional[str] = None 9 ctx1abcdefghi

22 secret_name: Optional[str] = None 9 ctx1abcdefghi

23 age: Optional[int] = None 9 ctx1abcdefghi

24 

25 @field_validator("name", "secret_name", "age") 9 ctx1abcdefghi

26 def reject_none(cls, v): 9 ctx1abcdefghi

27 assert v is not None 9 ctx1abcdefghi

28 return v 9 ctx1abcdefghi

29 

30 Hero.model_validate({"age": 25}) 9 ctx1abcdefghi

31 

32 with pytest.raises(ValidationError): 9 ctx1abcdefghi

33 Hero.model_validate({"name": None, "age": 25}) 9 ctx1abcdefghi