mirror of
https://github.com/PaiGramTeam/sqlmodel.git
synced 2024-11-25 17:35:40 +00:00
51 lines
1.3 KiB
Python
51 lines
1.3 KiB
Python
from typing import Optional
|
|
|
|
from sqlmodel import Field, Session, SQLModel, create_engine
|
|
|
|
|
|
def test_model_copy(clear_sqlmodel):
|
|
"""Test validation of implicit and explict None values.
|
|
|
|
# For consistency with pydantic, validators are not to be called on
|
|
# arguments that are not explicitly provided.
|
|
|
|
https://github.com/tiangolo/sqlmodel/issues/230
|
|
https://github.com/samuelcolvin/pydantic/issues/1223
|
|
|
|
"""
|
|
|
|
class Hero(SQLModel, table=True):
|
|
id: Optional[int] = Field(default=None, primary_key=True)
|
|
name: str
|
|
secret_name: str
|
|
age: Optional[int] = None
|
|
|
|
hero = Hero(name="Deadpond", secret_name="Dive Wilson", age=25)
|
|
|
|
engine = create_engine("sqlite://")
|
|
|
|
SQLModel.metadata.create_all(engine)
|
|
|
|
with Session(engine) as session:
|
|
session.add(hero)
|
|
session.commit()
|
|
session.refresh(hero)
|
|
|
|
model_copy = hero.model_copy(update={"name": "Deadpond Copy"})
|
|
|
|
assert (
|
|
model_copy.name == "Deadpond Copy"
|
|
and model_copy.secret_name == "Dive Wilson"
|
|
and model_copy.age == 25
|
|
)
|
|
|
|
db_hero = session.get(Hero, hero.id)
|
|
|
|
db_copy = db_hero.model_copy(update={"name": "Deadpond Copy"})
|
|
|
|
assert (
|
|
db_copy.name == "Deadpond Copy"
|
|
and db_copy.secret_name == "Dive Wilson"
|
|
and db_copy.age == 25
|
|
)
|