Coverage for tests/test_field_sa_relationship.py: 100%

1from typing import List, Optional 6 ctx1mnopqr

2 

3import pytest 6 ctx1mnopqr

4from sqlalchemy.orm import relationship 6 ctx1mnopqr

5from sqlmodel import Field, Relationship, SQLModel 6 ctx1mnopqr

6 

7 

8def test_sa_relationship_no_args() -> None: 6 ctx1mnopqr

9 with pytest.raises(RuntimeError): # pragma: no cover 6 ctx1cdaefg

10 

11 class Team(SQLModel, table=True): 6 ctx1cdaefg

12 id: Optional[int] = Field(default=None, primary_key=True) 6 ctx1cdaefg

13 name: str = Field(index=True) 6 ctx1cdaefg

14 headquarters: str 6 ctx1cdaefg

15 

16 heroes: List["Hero"] = Relationship( 6 ctx1cdaefg

17 back_populates="team", 6 ctx1cdaefg

18 sa_relationship_args=["Hero"], 6 ctx1cdaefg

19 sa_relationship=relationship("Hero", back_populates="team"), 6 ctx1cdaefg

20 ) 

21 

22 class Hero(SQLModel, table=True): 1 ctx1a

23 id: Optional[int] = Field(default=None, primary_key=True) 

24 name: str = Field(index=True) 

25 secret_name: str 

26 age: Optional[int] = Field(default=None, index=True) 

27 

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

29 team: Optional[Team] = Relationship(back_populates="heroes") 

30 

31 

32def test_sa_relationship_no_kwargs() -> None: 6 ctx1mnopqr

33 with pytest.raises(RuntimeError): # pragma: no cover 6 ctx1hibjkl

34 

35 class Team(SQLModel, table=True): 6 ctx1hibjkl

36 id: Optional[int] = Field(default=None, primary_key=True) 6 ctx1hibjkl

37 name: str = Field(index=True) 6 ctx1hibjkl

38 headquarters: str 6 ctx1hibjkl

39 

40 heroes: List["Hero"] = Relationship( 6 ctx1hibjkl

41 back_populates="team", 6 ctx1hibjkl

42 sa_relationship_kwargs={"lazy": "selectin"}, 6 ctx1hibjkl

43 sa_relationship=relationship("Hero", back_populates="team"), 6 ctx1hibjkl

44 ) 

45 

46 class Hero(SQLModel, table=True): 1 ctx1b

47 id: Optional[int] = Field(default=None, primary_key=True) 

48 name: str = Field(index=True) 

49 secret_name: str 

50 age: Optional[int] = Field(default=None, index=True) 

51 

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

53 team: Optional[Team] = Relationship(back_populates="heroes")