Coverage for tests/test_nullable.py: 100%

1from typing import Optional 6 ctx1mnopqr

2 

3import pytest 6 ctx1mnopqr

4from sqlalchemy.exc import IntegrityError 6 ctx1mnopqr

5from sqlmodel import Field, Session, SQLModel, create_engine 6 ctx1mnopqr

6 

7 

8def test_nullable_fields(clear_sqlmodel, caplog): 6 ctx1mnopqr

9 class Hero(SQLModel, table=True): 6 ctx1abcdef

10 primary_key: Optional[int] = Field( 6 ctx1abcdef

11 default=None, 

12 primary_key=True, 

13 ) 

14 required_value: str 6 ctx1abcdef

15 optional_default_ellipsis: Optional[str] = Field(default=...) 6 ctx1abcdef

16 optional_default_none: Optional[str] = Field(default=None) 6 ctx1abcdef

17 optional_non_nullable: Optional[str] = Field( 6 ctx1abcdef

18 nullable=False, 

19 ) 

20 optional_nullable: Optional[str] = Field( 6 ctx1abcdef

21 nullable=True, 

22 ) 

23 optional_default_ellipses_non_nullable: Optional[str] = Field( 6 ctx1abcdef

24 default=..., 

25 nullable=False, 

26 ) 

27 optional_default_ellipses_nullable: Optional[str] = Field( 6 ctx1abcdef

28 default=..., 

29 nullable=True, 

30 ) 

31 optional_default_none_non_nullable: Optional[str] = Field( 6 ctx1abcdef

32 default=None, 

33 nullable=False, 

34 ) 

35 optional_default_none_nullable: Optional[str] = Field( 6 ctx1abcdef

36 default=None, 

37 nullable=True, 

38 ) 

39 default_ellipses_non_nullable: str = Field(default=..., nullable=False) 6 ctx1abcdef

40 optional_default_str: Optional[str] = "default" 6 ctx1abcdef

41 optional_default_str_non_nullable: Optional[str] = Field( 6 ctx1abcdef

42 default="default", nullable=False 

43 ) 

44 optional_default_str_nullable: Optional[str] = Field( 6 ctx1abcdef

45 default="default", nullable=True 

46 ) 

47 str_default_str: str = "default" 6 ctx1abcdef

48 str_default_str_non_nullable: str = Field(default="default", nullable=False) 6 ctx1abcdef

49 str_default_str_nullable: str = Field(default="default", nullable=True) 6 ctx1abcdef

50 str_default_ellipsis_non_nullable: str = Field(default=..., nullable=False) 6 ctx1abcdef

51 str_default_ellipsis_nullable: str = Field(default=..., nullable=True) 6 ctx1abcdef

52 

53 engine = create_engine("sqlite://", echo=True) 6 ctx1abcdef

54 SQLModel.metadata.create_all(engine) 6 ctx1abcdef

55 

56 create_table_log = [ 6 ctx1abcdef

57 message for message in caplog.messages if "CREATE TABLE hero" in message 

58 ][0] 

59 assert "primary_key INTEGER NOT NULL," in create_table_log 6 ctx1abcdef

60 assert "required_value VARCHAR NOT NULL," in create_table_log 6 ctx1abcdef

61 assert "optional_default_ellipsis VARCHAR," in create_table_log 6 ctx1abcdef

62 assert "optional_default_none VARCHAR," in create_table_log 6 ctx1abcdef

63 assert "optional_non_nullable VARCHAR NOT NULL," in create_table_log 6 ctx1abcdef

64 assert "optional_nullable VARCHAR," in create_table_log 6 ctx1abcdef

65 assert ( 6 ctx1abcdef

66 "optional_default_ellipses_non_nullable VARCHAR NOT NULL," in create_table_log 

67 ) 

68 assert "optional_default_ellipses_nullable VARCHAR," in create_table_log 6 ctx1abcdef

69 assert "optional_default_none_non_nullable VARCHAR NOT NULL," in create_table_log 6 ctx1abcdef

70 assert "optional_default_none_nullable VARCHAR," in create_table_log 6 ctx1abcdef

71 assert "default_ellipses_non_nullable VARCHAR NOT NULL," in create_table_log 6 ctx1abcdef

72 assert "optional_default_str VARCHAR," in create_table_log 6 ctx1abcdef

73 assert "optional_default_str_non_nullable VARCHAR NOT NULL," in create_table_log 6 ctx1abcdef

74 assert "optional_default_str_nullable VARCHAR," in create_table_log 6 ctx1abcdef

75 assert "str_default_str VARCHAR NOT NULL," in create_table_log 6 ctx1abcdef

76 assert "str_default_str_non_nullable VARCHAR NOT NULL," in create_table_log 6 ctx1abcdef

77 assert "str_default_str_nullable VARCHAR," in create_table_log 6 ctx1abcdef

78 assert "str_default_ellipsis_non_nullable VARCHAR NOT NULL," in create_table_log 6 ctx1abcdef

79 assert "str_default_ellipsis_nullable VARCHAR," in create_table_log 6 ctx1abcdef

80 

81 

82# Test for regression in https://github.com/tiangolo/sqlmodel/issues/420 

83def test_non_nullable_optional_field_with_no_default_set(clear_sqlmodel, caplog): 6 ctx1mnopqr

84 class Hero(SQLModel, table=True): 6 ctx1ghijkl

85 primary_key: Optional[int] = Field( 6 ctx1ghijkl

86 default=None, 

87 primary_key=True, 

88 ) 

89 

90 optional_non_nullable_no_default: Optional[str] = Field(nullable=False) 6 ctx1ghijkl

91 

92 engine = create_engine("sqlite://", echo=True) 6 ctx1ghijkl

93 SQLModel.metadata.create_all(engine) 6 ctx1ghijkl

94 

95 create_table_log = [ 6 ctx1ghijkl

96 message for message in caplog.messages if "CREATE TABLE hero" in message 

97 ][0] 

98 assert "primary_key INTEGER NOT NULL," in create_table_log 6 ctx1ghijkl

99 assert "optional_non_nullable_no_default VARCHAR NOT NULL," in create_table_log 6 ctx1ghijkl

100 

101 # We can create a hero with `None` set for the optional non-nullable field 

102 hero = Hero(primary_key=123, optional_non_nullable_no_default=None) 6 ctx1ghijkl

103 # But we cannot commit it. 

104 with Session(engine) as session: 6 ctx1ghijkl

105 session.add(hero) 6 ctx1ghijkl

106 with pytest.raises(IntegrityError): 6 ctx1ghijkl

107 session.commit() 6 ctx1ghijkl

108 

109 

110def test_nullable_primary_key(clear_sqlmodel, caplog): 6 ctx1mnopqr

111 # Probably the weirdest corner case, it shouldn't happen anywhere, but let's test it 

112 class Hero(SQLModel, table=True): 6 ctx1stuvwx

113 nullable_integer_primary_key: Optional[int] = Field( 6 ctx1stuvwx

114 default=None, 

115 primary_key=True, 

116 nullable=True, 

117 ) 

118 

119 engine = create_engine("sqlite://", echo=True) 6 ctx1stuvwx

120 SQLModel.metadata.create_all(engine) 6 ctx1stuvwx

121 

122 create_table_log = [ 6 ctx1stuvwx

123 message for message in caplog.messages if "CREATE TABLE hero" in message 

124 ][0] 

125 assert "nullable_integer_primary_key INTEGER," in create_table_log 6 ctx1stuvwx