Coverage for tests/test_ondelete_raises.py: 100%

1from typing import Any, List, Union 6 ctx1abcdef

2 

3import pytest 6 ctx1abcdef

4from sqlmodel import Field, Relationship, SQLModel 6 ctx1abcdef

5 

6 

7def test_ondelete_requires_nullable(clear_sqlmodel: Any) -> None: 6 ctx1abcdef

8 with pytest.raises(RuntimeError) as exc: 6 ctx1abcdef

9 

10 class Team(SQLModel, table=True): 6 ctx1abcdef

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

12 

13 heroes: List["Hero"] = Relationship( 6 ctx1abcdef

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

15 ) 

16 

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

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

19 name: str = Field(index=True) 6 ctx1abcdef

20 secret_name: str 6 ctx1abcdef

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

22 

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

24 team: Team = Relationship(back_populates="heroes") 6 ctx1abcdef

25 

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

27 

28 

29def test_ondelete_requires_foreign_key(clear_sqlmodel: Any) -> None: 6 ctx1abcdef

30 with pytest.raises(RuntimeError) as exc: 6 ctx1abcdef

31 

32 class Team(SQLModel, table=True): 6 ctx1abcdef

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

34 

35 age: int = Field(ondelete="CASCADE") 6 ctx1abcdef

36 

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