Coverage for tests/test_validation.py: 100%

1from typing import Optional 7 ctx1uopqrst

2 

3import pytest 7 ctx1uopqrst

4from pydantic.error_wrappers import ValidationError 7 ctx1uopqrst

5from sqlmodel import SQLModel 7 ctx1uopqrst

6 

7from .conftest import needs_pydanticv1, needs_pydanticv2 7 ctx1uopqrst

8 

9 

10@needs_pydanticv1 7 ctx1uopqrst

11def test_validation_pydantic_v1(clear_sqlmodel): 6 ctx1opqrst

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 7 ctx1mabcdef

22 

23 class Hero(SQLModel): 7 ctx1mabcdef

24 name: Optional[str] = None 7 ctx1mabcdef

25 secret_name: Optional[str] = None 7 ctx1mabcdef

26 age: Optional[int] = None 7 ctx1mabcdef

27 

28 @validator("name", "secret_name", "age") 7 ctx1mabcdef

29 def reject_none(cls, v): 6 ctx1abcdef

30 assert v is not None 7 ctx1mabcdef

31 return v 7 ctx1mabcdef

32 

33 Hero.validate({"age": 25}) 7 ctx1mabcdef

34 

35 with pytest.raises(ValidationError): 7 ctx1mabcdef

36 Hero.validate({"name": None, "age": 25}) 7 ctx1mabcdef

37 

38 

39@needs_pydanticv2 7 ctx1uopqrst

40def test_validation_pydantic_v2(clear_sqlmodel): 6 ctx1opqrst

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 7 ctx1nghijkl

51 

52 class Hero(SQLModel): 7 ctx1nghijkl

53 name: Optional[str] = None 7 ctx1nghijkl

54 secret_name: Optional[str] = None 7 ctx1nghijkl

55 age: Optional[int] = None 7 ctx1nghijkl

56 

57 @field_validator("name", "secret_name", "age") 7 ctx1nghijkl

58 def reject_none(cls, v): 6 ctx1ghijkl

59 assert v is not None 7 ctx1nghijkl

60 return v 7 ctx1nghijkl

61 

62 Hero.model_validate({"age": 25}) 7 ctx1nghijkl

63 

64 with pytest.raises(ValidationError): 7 ctx1nghijkl

65 Hero.model_validate({"name": None, "age": 25}) 7 ctx1nghijkl