Coverage for tests/test_field_sa_relationship.py: 100%

1from typing import List, Optional 7 ctx1opqrstu

2 

3import pytest 7 ctx1opqrstu

4from sqlalchemy.orm import relationship 7 ctx1opqrstu

5from sqlmodel import Field, Relationship, SQLModel 7 ctx1opqrstu

6 

7 

8def test_sa_relationship_no_args() -> None: 7 ctx1opqrstu

9 with pytest.raises(RuntimeError): # pragma: no cover 7 ctx1cdaefgh

10 

11 class Team(SQLModel, table=True): 7 ctx1cdaefgh

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

13 name: str = Field(index=True) 7 ctx1cdaefgh

14 headquarters: str 7 ctx1cdaefgh

15 

16 heroes: List["Hero"] = Relationship( 7 ctx1cdaefgh

17 back_populates="team", 7 ctx1cdaefgh

18 sa_relationship_args=["Hero"], 7 ctx1cdaefgh

19 sa_relationship=relationship("Hero", back_populates="team"), 7 ctx1cdaefgh

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: 7 ctx1opqrstu

33 with pytest.raises(RuntimeError): # pragma: no cover 7 ctx1ijbklmn

34 

35 class Team(SQLModel, table=True): 7 ctx1ijbklmn

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

37 name: str = Field(index=True) 7 ctx1ijbklmn

38 headquarters: str 7 ctx1ijbklmn

39 

40 heroes: List["Hero"] = Relationship( 7 ctx1ijbklmn

41 back_populates="team", 7 ctx1ijbklmn

42 sa_relationship_kwargs={"lazy": "selectin"}, 7 ctx1ijbklmn

43 sa_relationship=relationship("Hero", back_populates="team"), 7 ctx1ijbklmn

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")