mirror of
https://github.com/fastapi/sqlmodel.git
synced 2025-10-31 18:15:44 +08:00
129 lines
4.1 KiB
Python
129 lines
4.1 KiB
Python
from typing import List, Optional
|
|
|
|
import pytest
|
|
from sqlalchemy.exc import IntegrityError
|
|
from sqlalchemy.orm import RelationshipProperty
|
|
from sqlmodel import Field, Relationship, Session, SQLModel, create_engine
|
|
|
|
|
|
def test_should_allow_duplicate_row_if_unique_constraint_is_not_passed(clear_sqlmodel):
|
|
class Hero(SQLModel, table=True):
|
|
id: Optional[int] = Field(default=None, primary_key=True)
|
|
name: str
|
|
secret_name: str
|
|
age: Optional[int] = None
|
|
|
|
hero_1 = Hero(name="Deadpond", secret_name="Dive Wilson")
|
|
hero_2 = Hero(name="Deadpond", secret_name="Dive Wilson")
|
|
|
|
engine = create_engine("sqlite://")
|
|
|
|
SQLModel.metadata.create_all(engine)
|
|
|
|
with Session(engine) as session:
|
|
session.add(hero_1)
|
|
session.commit()
|
|
session.refresh(hero_1)
|
|
|
|
with Session(engine) as session:
|
|
session.add(hero_2)
|
|
session.commit()
|
|
session.refresh(hero_2)
|
|
|
|
with Session(engine) as session:
|
|
heroes = session.query(Hero).all()
|
|
assert len(heroes) == 2
|
|
assert heroes[0].name == heroes[1].name
|
|
|
|
|
|
def test_should_allow_duplicate_row_if_unique_constraint_is_false(clear_sqlmodel):
|
|
class Hero(SQLModel, table=True):
|
|
id: Optional[int] = Field(default=None, primary_key=True)
|
|
name: str
|
|
secret_name: str = Field(unique=False)
|
|
age: Optional[int] = None
|
|
|
|
hero_1 = Hero(name="Deadpond", secret_name="Dive Wilson")
|
|
hero_2 = Hero(name="Deadpond", secret_name="Dive Wilson")
|
|
|
|
engine = create_engine("sqlite://")
|
|
|
|
SQLModel.metadata.create_all(engine)
|
|
|
|
with Session(engine) as session:
|
|
session.add(hero_1)
|
|
session.commit()
|
|
session.refresh(hero_1)
|
|
|
|
with Session(engine) as session:
|
|
session.add(hero_2)
|
|
session.commit()
|
|
session.refresh(hero_2)
|
|
|
|
with Session(engine) as session:
|
|
heroes = session.query(Hero).all()
|
|
assert len(heroes) == 2
|
|
assert heroes[0].name == heroes[1].name
|
|
|
|
|
|
def test_should_raise_exception_when_try_to_duplicate_row_if_unique_constraint_is_true(
|
|
clear_sqlmodel,
|
|
):
|
|
class Hero(SQLModel, table=True):
|
|
id: Optional[int] = Field(default=None, primary_key=True)
|
|
name: str
|
|
secret_name: str = Field(unique=True)
|
|
age: Optional[int] = None
|
|
|
|
hero_1 = Hero(name="Deadpond", secret_name="Dive Wilson")
|
|
hero_2 = Hero(name="Deadpond", secret_name="Dive Wilson")
|
|
|
|
engine = create_engine("sqlite://")
|
|
|
|
SQLModel.metadata.create_all(engine)
|
|
|
|
with Session(engine) as session:
|
|
session.add(hero_1)
|
|
session.commit()
|
|
session.refresh(hero_1)
|
|
|
|
with pytest.raises(IntegrityError):
|
|
with Session(engine) as session:
|
|
session.add(hero_2)
|
|
session.commit()
|
|
session.refresh(hero_2)
|
|
|
|
|
|
def test_sa_relationship_property(clear_sqlmodel):
|
|
"""Test https://github.com/tiangolo/sqlmodel/issues/315#issuecomment-1272122306"""
|
|
|
|
class Team(SQLModel, table=True):
|
|
id: Optional[int] = Field(default=None, primary_key=True)
|
|
name: str = Field(unique=True)
|
|
heroes: List["Hero"] = Relationship( # noqa: F821
|
|
sa_relationship=RelationshipProperty("Hero", back_populates="team")
|
|
)
|
|
|
|
class Hero(SQLModel, table=True):
|
|
id: Optional[int] = Field(default=None, primary_key=True)
|
|
name: str = Field(unique=True)
|
|
team_id: Optional[int] = Field(default=None, foreign_key="team.id")
|
|
team: Optional[Team] = Relationship(
|
|
sa_relationship=RelationshipProperty("Team", back_populates="heroes")
|
|
)
|
|
|
|
team_preventers = Team(name="Preventers")
|
|
hero_rusty_man = Hero(name="Rusty-Man", team=team_preventers)
|
|
|
|
engine = create_engine("sqlite://", echo=True)
|
|
|
|
SQLModel.metadata.create_all(engine)
|
|
|
|
with Session(engine) as session:
|
|
session.add(hero_rusty_man)
|
|
session.commit()
|
|
session.refresh(hero_rusty_man)
|
|
# The next statement should not raise an AttributeError
|
|
assert hero_rusty_man.team
|
|
assert hero_rusty_man.team.name == "Preventers"
|