Coverage for tests/test_field_sa_relationship.py: 100%

1from typing import List, Optional 10 ctx1uvwxyzABCD

2 

3import pytest 10 ctx1uvwxyzABCD

4from sqlalchemy.orm import relationship 10 ctx1uvwxyzABCD

5from sqlmodel import Field, Relationship, SQLModel 10 ctx1uvwxyzABCD

6 

7 

8def test_sa_relationship_no_args() -> None: 10 ctx1uvwxyzABCD

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

10 

11 class Team(SQLModel, table=True): 10 ctx1cdoaepfghq

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

13 name: str = Field(index=True) 10 ctx1cdoaepfghq

14 headquarters: str 7 ctx1cdaefgh

15 

16 heroes: List["Hero"] = Relationship( 10 ctx1cdoaepfghq

17 back_populates="team", 10 ctx1cdoaepfghq

18 sa_relationship_args=["Hero"], 10 ctx1cdoaepfghq

19 sa_relationship=relationship("Hero", back_populates="team"), 10 ctx1cdoaepfghq

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: 10 ctx1uvwxyzABCD

33 with pytest.raises(RuntimeError): # pragma: no cover 10 ctx1ijrbkslmnt

34 

35 class Team(SQLModel, table=True): 10 ctx1ijrbkslmnt

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

37 name: str = Field(index=True) 10 ctx1ijrbkslmnt

38 headquarters: str 7 ctx1ijbklmn

39 

40 heroes: List["Hero"] = Relationship( 10 ctx1ijrbkslmnt

41 back_populates="team", 10 ctx1ijrbkslmnt

42 sa_relationship_kwargs={"lazy": "selectin"}, 10 ctx1ijrbkslmnt

43 sa_relationship=relationship("Hero", back_populates="team"), 10 ctx1ijrbkslmnt

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