Coverage for tests/test_field_sa_relationship.py: 100%
6 statements
« prev ^ index » next coverage.py v7.6.1, created at 2024-08-12 19:13 +0000
1from typing import List, Optional 1bcadef
3import pytest 1bcadef
4from sqlalchemy.orm import relationship 1bcadef
5from sqlmodel import Field, Relationship, SQLModel 1bcadef
8def test_sa_relationship_no_args() -> None: 1bcadef
9 with pytest.raises(RuntimeError): # pragma: no cover 1bcadef
11 class Team(SQLModel, table=True): 1bcadef
12 id: Optional[int] = Field(default=None, primary_key=True) 1bcadef
13 name: str = Field(index=True) 1bcadef
14 headquarters: str 1bcadef
16 heroes: List["Hero"] = Relationship( 1bcadef
17 back_populates="team", 1bcadef
18 sa_relationship_args=["Hero"], 1bcadef
19 sa_relationship=relationship("Hero", back_populates="team"), 1bcadef
20 )
22 class Hero(SQLModel, table=True): 1a
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)
28 team_id: Optional[int] = Field(default=None, foreign_key="team.id")
29 team: Optional[Team] = Relationship(back_populates="heroes")
32def test_sa_relationship_no_kwargs() -> None: 1bcadef
33 with pytest.raises(RuntimeError): # pragma: no cover 1bcadef
35 class Team(SQLModel, table=True): 1bcadef
36 id: Optional[int] = Field(default=None, primary_key=True) 1bcadef
37 name: str = Field(index=True) 1bcadef
38 headquarters: str 1bcadef
40 heroes: List["Hero"] = Relationship( 1bcadef
41 back_populates="team", 1bcadef
42 sa_relationship_kwargs={"lazy": "selectin"}, 1bcadef
43 sa_relationship=relationship("Hero", back_populates="team"), 1bcadef
44 )
46 class Hero(SQLModel, table=True): 1a
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)
52 team_id: Optional[int] = Field(default=None, foreign_key="team.id")
53 team: Optional[Team] = Relationship(back_populates="heroes")