Coverage for tests / test_ondelete_raises.py: 100%

1from typing import Any, Union 9 ctx1jklmnopqr

2 

3import pytest 9 ctx1jklmnopqr

4from sqlmodel import Field, Relationship, SQLModel 9 ctx1jklmnopqr

5 

6 

7def test_ondelete_requires_nullable(clear_sqlmodel: Any) -> None: 9 ctx1jklmnopqr

8 with pytest.raises(RuntimeError) as exc: 9 ctx1abcdefghi

9 

10 class Team(SQLModel, table=True): 9 ctx1abcdefghi

11 id: Union[int, None] = Field(default=None, primary_key=True) 9 ctx1abcdefghi

12 

13 heroes: list["Hero"] = Relationship( 9 ctx1abcdefghi

14 back_populates="team", passive_deletes="all" 

15 ) 

16 

17 class Hero(SQLModel, table=True): 9 ctx1abcdefghi

18 id: Union[int, None] = Field(default=None, primary_key=True) 9 ctx1abcdefghi

19 name: str = Field(index=True) 9 ctx1abcdefghi

20 secret_name: str 9 ctx1abcdefghi

21 age: Union[int, None] = Field(default=None, index=True) 9 ctx1abcdefghi

22 

23 team_id: int = Field(foreign_key="team.id", ondelete="SET NULL") 9 ctx1abcdefghi

24 team: Team = Relationship(back_populates="heroes") 9 ctx1abcdefghi

25 

26 assert 'ondelete="SET NULL" requires nullable=True' in str(exc.value) 9 ctx1abcdefghi

27 

28 

29def test_ondelete_requires_foreign_key(clear_sqlmodel: Any) -> None: 9 ctx1jklmnopqr

30 with pytest.raises(RuntimeError) as exc: 9 ctx1stuvwxyzA

31 

32 class Team(SQLModel, table=True): 9 ctx1stuvwxyzA

33 id: Union[int, None] = Field(default=None, primary_key=True) 9 ctx1stuvwxyzA

34 

35 age: int = Field(ondelete="CASCADE") 9 ctx1stuvwxyzA

36 

37 assert "ondelete can only be used with foreign_key" in str(exc.value) 9 ctx1stuvwxyzA