Coverage for tests / test_field_sa_relationship.py: 100%

1from typing import Optional 9 ctx1stuvwxyzA

2 

3import pytest 9 ctx1stuvwxyzA

4from sqlalchemy.orm import relationship 9 ctx1stuvwxyzA

5from sqlmodel import Field, Relationship, SQLModel 9 ctx1stuvwxyzA

6 

7 

8def test_sa_relationship_no_args() -> None: 9 ctx1stuvwxyzA

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

10 

11 class Team(SQLModel, table=True): 9 ctx1cdmaenfgo

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

13 name: str = Field(index=True) 9 ctx1cdmaenfgo

14 headquarters: str 6 ctx1cdaefg

15 

16 heroes: list["Hero"] = Relationship( 9 ctx1cdmaenfgo

17 back_populates="team", 9 ctx1cdmaenfgo

18 sa_relationship_args=["Hero"], 9 ctx1cdmaenfgo

19 sa_relationship=relationship("Hero", back_populates="team"), 9 ctx1cdmaenfgo

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: 9 ctx1stuvwxyzA

33 with pytest.raises(RuntimeError): # pragma: no cover 9 ctx1hipbjqklr

34 

35 class Team(SQLModel, table=True): 9 ctx1hipbjqklr

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

37 name: str = Field(index=True) 9 ctx1hipbjqklr

38 headquarters: str 6 ctx1hibjkl

39 

40 heroes: list["Hero"] = Relationship( 9 ctx1hipbjqklr

41 back_populates="team", 9 ctx1hipbjqklr

42 sa_relationship_kwargs={"lazy": "selectin"}, 9 ctx1hipbjqklr

43 sa_relationship=relationship("Hero", back_populates="team"), 9 ctx1hipbjqklr

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