Coverage for tests/test_ondelete_raises.py: 100%

1from typing import Any, List, Union 10 ctx1klmnopqrst

2 

3import pytest 10 ctx1klmnopqrst

4from sqlmodel import Field, Relationship, SQLModel 10 ctx1klmnopqrst

5 

6 

7def test_ondelete_requires_nullable(clear_sqlmodel: Any) -> None: 10 ctx1klmnopqrst

8 with pytest.raises(RuntimeError) as exc: 10 ctx1abcdefghij

9 

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

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

12 

13 heroes: List["Hero"] = Relationship( 10 ctx1abcdefghij

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

15 ) 

16 

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

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

19 name: str = Field(index=True) 10 ctx1abcdefghij

20 secret_name: str 10 ctx1abcdefghij

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

22 

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

24 team: Team = Relationship(back_populates="heroes") 10 ctx1abcdefghij

25 

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

27 

28 

29def test_ondelete_requires_foreign_key(clear_sqlmodel: Any) -> None: 10 ctx1klmnopqrst

30 with pytest.raises(RuntimeError) as exc: 10 ctx1uvwxyzABCD

31 

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

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

34 

35 age: int = Field(ondelete="CASCADE") 10 ctx1uvwxyzABCD

36 

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