Coverage for tests/test_validation.py: 100%

1from typing import Optional 6 ctx1rmnopq

2 

3import pytest 6 ctx1rmnopq

4from pydantic.error_wrappers import ValidationError 6 ctx1rmnopq

5from sqlmodel import SQLModel 6 ctx1rmnopq

6 

7from .conftest import needs_pydanticv1, needs_pydanticv2 6 ctx1rmnopq

8 

9 

10@needs_pydanticv1 6 ctx1rmnopq

11def test_validation_pydantic_v1(clear_sqlmodel): 5 ctx1mnopq

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 ctx1kabcde

22 

23 class Hero(SQLModel): 6 ctx1kabcde

24 name: Optional[str] = None 6 ctx1kabcde

25 secret_name: Optional[str] = None 6 ctx1kabcde

26 age: Optional[int] = None 6 ctx1kabcde

27 

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

29 def reject_none(cls, v): 5 ctx1abcde

30 assert v is not None 6 ctx1kabcde

31 return v 6 ctx1kabcde

32 

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

34 

35 with pytest.raises(ValidationError): 6 ctx1kabcde

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

37 

38 

39@needs_pydanticv2 6 ctx1rmnopq

40def test_validation_pydantic_v2(clear_sqlmodel): 5 ctx1mnopq

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 ctx1lfghij

51 

52 class Hero(SQLModel): 6 ctx1lfghij

53 name: Optional[str] = None 6 ctx1lfghij

54 secret_name: Optional[str] = None 6 ctx1lfghij

55 age: Optional[int] = None 6 ctx1lfghij

56 

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

58 def reject_none(cls, v): 5 ctx1fghij

59 assert v is not None 6 ctx1lfghij

60 return v 6 ctx1lfghij

61 

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

63 

64 with pytest.raises(ValidationError): 6 ctx1lfghij

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