Coverage for tests/test_validation.py: 100%

1from typing import Optional 6 ctx1mnopqr

2 

3import pytest 6 ctx1mnopqr

4from pydantic.error_wrappers import ValidationError 6 ctx1mnopqr

5from sqlmodel import SQLModel 6 ctx1mnopqr

6 

7from .conftest import needs_pydanticv1, needs_pydanticv2 6 ctx1mnopqr

8 

9 

10@needs_pydanticv1 6 ctx1mnopqr

11def test_validation_pydantic_v1(clear_sqlmodel): 6 ctx1mnopqr

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

13 

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

15 # arguments that are not explicitly provided. 

16 

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

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

19 

20 """ 

21 from pydantic import validator 6 ctx1abcdef

22 

23 class Hero(SQLModel): 6 ctx1abcdef

24 name: Optional[str] = None 6 ctx1abcdef

25 secret_name: Optional[str] = None 6 ctx1abcdef

26 age: Optional[int] = None 6 ctx1abcdef

27 

28 @validator("name", "secret_name", "age") 6 ctx1abcdef

29 def reject_none(cls, v): 6 ctx1abcdef

30 assert v is not None 6 ctx1abcdef

31 return v 6 ctx1abcdef

32 

33 Hero.validate({"age": 25}) 6 ctx1abcdef

34 

35 with pytest.raises(ValidationError): 6 ctx1abcdef

36 Hero.validate({"name": None, "age": 25}) 6 ctx1abcdef

37 

38 

39@needs_pydanticv2 6 ctx1mnopqr

40def test_validation_pydantic_v2(clear_sqlmodel): 6 ctx1mnopqr

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

42 

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

44 # arguments that are not explicitly provided. 

45 

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

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

48 

49 """ 

50 from pydantic import field_validator 6 ctx1ghijkl

51 

52 class Hero(SQLModel): 6 ctx1ghijkl

53 name: Optional[str] = None 6 ctx1ghijkl

54 secret_name: Optional[str] = None 6 ctx1ghijkl

55 age: Optional[int] = None 6 ctx1ghijkl

56 

57 @field_validator("name", "secret_name", "age") 6 ctx1ghijkl

58 def reject_none(cls, v): 6 ctx1ghijkl

59 assert v is not None 6 ctx1ghijkl

60 return v 6 ctx1ghijkl

61 

62 Hero.model_validate({"age": 25}) 6 ctx1ghijkl

63 

64 with pytest.raises(ValidationError): 6 ctx1ghijkl

65 Hero.model_validate({"name": None, "age": 25}) 6 ctx1ghijkl