Coverage for tests/test_validation.py: 100%

1from typing import Optional 6 ctx1fabcde

2 

3import pytest 6 ctx1fabcde

4from pydantic.error_wrappers import ValidationError 6 ctx1fabcde

5from sqlmodel import SQLModel 6 ctx1fabcde

6 

7from .conftest import needs_pydanticv1, needs_pydanticv2 6 ctx1fabcde

8 

9 

10@needs_pydanticv1 6 ctx1fabcde

11def test_validation_pydantic_v1(clear_sqlmodel): 5 ctx1abcde

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 2 ctx1ae

22 

23 class Hero(SQLModel): 2 ctx1ae

24 name: Optional[str] = None 2 ctx1ae

25 secret_name: Optional[str] = None 2 ctx1ae

26 age: Optional[int] = None 2 ctx1ae

27 

28 @validator("name", "secret_name", "age") 2 ctx1ae

29 def reject_none(cls, v): 2 ctx1ae

30 assert v is not None 2 ctx1ae

31 return v 2 ctx1ae

32 

33 Hero.validate({"age": 25}) 2 ctx1ae

34 

35 with pytest.raises(ValidationError): 2 ctx1ae

36 Hero.validate({"name": None, "age": 25}) 2 ctx1ae

37 

38 

39@needs_pydanticv2 6 ctx1fabcde

40def test_validation_pydantic_v2(clear_sqlmodel): 5 ctx1abcde

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 4 ctx1fbcd

51 

52 class Hero(SQLModel): 4 ctx1fbcd

53 name: Optional[str] = None 4 ctx1fbcd

54 secret_name: Optional[str] = None 4 ctx1fbcd

55 age: Optional[int] = None 4 ctx1fbcd

56 

57 @field_validator("name", "secret_name", "age") 4 ctx1fbcd

58 def reject_none(cls, v): 3 ctx1bcd

59 assert v is not None 4 ctx1fbcd

60 return v 4 ctx1fbcd

61 

62 Hero.model_validate({"age": 25}) 4 ctx1fbcd

63 

64 with pytest.raises(ValidationError): 4 ctx1fbcd

65 Hero.model_validate({"name": None, "age": 25}) 4 ctx1fbcd