Coverage for tests/test_main.py: 100%

1from typing import List, Optional 10 ctx1OPQRSTUVWX

2 

3import pytest 10 ctx1OPQRSTUVWX

4from sqlalchemy.exc import IntegrityError 10 ctx1OPQRSTUVWX

5from sqlalchemy.orm import RelationshipProperty 10 ctx1OPQRSTUVWX

6from sqlmodel import Field, Relationship, Session, SQLModel, create_engine, select 10 ctx1OPQRSTUVWX

7 

8 

9def test_should_allow_duplicate_row_if_unique_constraint_is_not_passed(clear_sqlmodel): 10 ctx1OPQRSTUVWX

10 class Hero(SQLModel, table=True): 10 ctx1abcdefghij

11 id: Optional[int] = Field(default=None, primary_key=True) 10 ctx1abcdefghij

12 name: str 10 ctx1abcdefghij

13 secret_name: str 10 ctx1abcdefghij

14 age: Optional[int] = None 10 ctx1abcdefghij

15 

16 hero_1 = Hero(name="Deadpond", secret_name="Dive Wilson") 10 ctx1abcdefghij

17 hero_2 = Hero(name="Deadpond", secret_name="Dive Wilson") 10 ctx1abcdefghij

18 

19 engine = create_engine("sqlite://") 10 ctx1abcdefghij

20 

21 SQLModel.metadata.create_all(engine) 10 ctx1abcdefghij

22 

23 with Session(engine) as session: 10 ctx1abcdefghij

24 session.add(hero_1) 10 ctx1abcdefghij

25 session.commit() 10 ctx1abcdefghij

26 session.refresh(hero_1) 10 ctx1abcdefghij

27 

28 with Session(engine) as session: 10 ctx1abcdefghij

29 session.add(hero_2) 10 ctx1abcdefghij

30 session.commit() 10 ctx1abcdefghij

31 session.refresh(hero_2) 10 ctx1abcdefghij

32 

33 with Session(engine) as session: 10 ctx1abcdefghij

34 heroes = session.exec(select(Hero)).all() 10 ctx1abcdefghij

35 assert len(heroes) == 2 10 ctx1abcdefghij

36 assert heroes[0].name == heroes[1].name 10 ctx1abcdefghij

37 

38 

39def test_should_allow_duplicate_row_if_unique_constraint_is_false(clear_sqlmodel): 10 ctx1OPQRSTUVWX

40 class Hero(SQLModel, table=True): 10 ctx1klmnopqrst

41 id: Optional[int] = Field(default=None, primary_key=True) 10 ctx1klmnopqrst

42 name: str 10 ctx1klmnopqrst

43 secret_name: str = Field(unique=False) 10 ctx1klmnopqrst

44 age: Optional[int] = None 10 ctx1klmnopqrst

45 

46 hero_1 = Hero(name="Deadpond", secret_name="Dive Wilson") 10 ctx1klmnopqrst

47 hero_2 = Hero(name="Deadpond", secret_name="Dive Wilson") 10 ctx1klmnopqrst

48 

49 engine = create_engine("sqlite://") 10 ctx1klmnopqrst

50 

51 SQLModel.metadata.create_all(engine) 10 ctx1klmnopqrst

52 

53 with Session(engine) as session: 10 ctx1klmnopqrst

54 session.add(hero_1) 10 ctx1klmnopqrst

55 session.commit() 10 ctx1klmnopqrst

56 session.refresh(hero_1) 10 ctx1klmnopqrst

57 

58 with Session(engine) as session: 10 ctx1klmnopqrst

59 session.add(hero_2) 10 ctx1klmnopqrst

60 session.commit() 10 ctx1klmnopqrst

61 session.refresh(hero_2) 10 ctx1klmnopqrst

62 

63 with Session(engine) as session: 10 ctx1klmnopqrst

64 heroes = session.exec(select(Hero)).all() 10 ctx1klmnopqrst

65 assert len(heroes) == 2 10 ctx1klmnopqrst

66 assert heroes[0].name == heroes[1].name 10 ctx1klmnopqrst

67 

68 

69def test_should_raise_exception_when_try_to_duplicate_row_if_unique_constraint_is_true( 10 ctx1OPQRSTUVWX

70 clear_sqlmodel, 

71): 

72 class Hero(SQLModel, table=True): 10 ctx1EFGHIJKLMN

73 id: Optional[int] = Field(default=None, primary_key=True) 10 ctx1EFGHIJKLMN

74 name: str 10 ctx1EFGHIJKLMN

75 secret_name: str = Field(unique=True) 10 ctx1EFGHIJKLMN

76 age: Optional[int] = None 10 ctx1EFGHIJKLMN

77 

78 hero_1 = Hero(name="Deadpond", secret_name="Dive Wilson") 10 ctx1EFGHIJKLMN

79 hero_2 = Hero(name="Deadpond", secret_name="Dive Wilson") 10 ctx1EFGHIJKLMN

80 

81 engine = create_engine("sqlite://") 10 ctx1EFGHIJKLMN

82 

83 SQLModel.metadata.create_all(engine) 10 ctx1EFGHIJKLMN

84 

85 with Session(engine) as session: 10 ctx1EFGHIJKLMN

86 session.add(hero_1) 10 ctx1EFGHIJKLMN

87 session.commit() 10 ctx1EFGHIJKLMN

88 session.refresh(hero_1) 10 ctx1EFGHIJKLMN

89 

90 with pytest.raises(IntegrityError): 10 ctx1EFGHIJKLMN

91 with Session(engine) as session: 10 ctx1EFGHIJKLMN

92 session.add(hero_2) 10 ctx1EFGHIJKLMN

93 session.commit() 10 ctx1EFGHIJKLMN

94 

95 

96def test_sa_relationship_property(clear_sqlmodel): 10 ctx1OPQRSTUVWX

97 """Test https://github.com/tiangolo/sqlmodel/issues/315#issuecomment-1272122306""" 

98 

99 class Team(SQLModel, table=True): 10 ctx1uvwxyzABCD

100 id: Optional[int] = Field(default=None, primary_key=True) 10 ctx1uvwxyzABCD

101 name: str = Field(unique=True) 10 ctx1uvwxyzABCD

102 heroes: List["Hero"] = Relationship( # noqa: F821 10 ctx1uvwxyzABCD

103 sa_relationship=RelationshipProperty("Hero", back_populates="team") 

104 ) 

105 

106 class Hero(SQLModel, table=True): 10 ctx1uvwxyzABCD

107 id: Optional[int] = Field(default=None, primary_key=True) 10 ctx1uvwxyzABCD

108 name: str = Field(unique=True) 10 ctx1uvwxyzABCD

109 team_id: Optional[int] = Field(default=None, foreign_key="team.id") 10 ctx1uvwxyzABCD

110 team: Optional[Team] = Relationship( 10 ctx1uvwxyzABCD

111 sa_relationship=RelationshipProperty("Team", back_populates="heroes") 

112 ) 

113 

114 team_preventers = Team(name="Preventers") 10 ctx1uvwxyzABCD

115 hero_rusty_man = Hero(name="Rusty-Man", team=team_preventers) 10 ctx1uvwxyzABCD

116 

117 engine = create_engine("sqlite://", echo=True) 10 ctx1uvwxyzABCD

118 

119 SQLModel.metadata.create_all(engine) 10 ctx1uvwxyzABCD

120 

121 with Session(engine) as session: 10 ctx1uvwxyzABCD

122 session.add(hero_rusty_man) 10 ctx1uvwxyzABCD

123 session.commit() 10 ctx1uvwxyzABCD

124 session.refresh(hero_rusty_man) 10 ctx1uvwxyzABCD

125 # The next statement should not raise an AttributeError 

126 assert hero_rusty_man.team 10 ctx1uvwxyzABCD

127 assert hero_rusty_man.team.name == "Preventers" 10 ctx1uvwxyzABCD