Coverage for tests/test_ondelete_raises.py: 100%

1from typing import Any, List, Union 7 ctx1hijklmn

2 

3import pytest 7 ctx1hijklmn

4from sqlmodel import Field, Relationship, SQLModel 7 ctx1hijklmn

5 

6 

7def test_ondelete_requires_nullable(clear_sqlmodel: Any) -> None: 7 ctx1hijklmn

8 with pytest.raises(RuntimeError) as exc: 7 ctx1abcdefg

9 

10 class Team(SQLModel, table=True): 7 ctx1abcdefg

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

12 

13 heroes: List["Hero"] = Relationship( 7 ctx1abcdefg

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

15 ) 

16 

17 class Hero(SQLModel, table=True): 7 ctx1abcdefg

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

19 name: str = Field(index=True) 7 ctx1abcdefg

20 secret_name: str 7 ctx1abcdefg

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

22 

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

24 team: Team = Relationship(back_populates="heroes") 7 ctx1abcdefg

25 

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

27 

28 

29def test_ondelete_requires_foreign_key(clear_sqlmodel: Any) -> None: 7 ctx1hijklmn

30 with pytest.raises(RuntimeError) as exc: 7 ctx1opqrstu

31 

32 class Team(SQLModel, table=True): 7 ctx1opqrstu

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

34 

35 age: int = Field(ondelete="CASCADE") 7 ctx1opqrstu

36 

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