fa2f178b8a
Co-authored-by: Mohamed Farahat <farahats9@yahoo.com> Co-authored-by: Stefan Borer <stefan.borer@gmail.com> Co-authored-by: Peter Landry <peter.landry@gmail.com> Co-authored-by: Anton De Meester <antondemeester+github@gmail.com>
36 lines
1.0 KiB
Python
36 lines
1.0 KiB
Python
from typing import Optional
|
|
|
|
import pytest
|
|
from pydantic import ValidationError
|
|
from sqlmodel import Field, Session, SQLModel, create_engine, select
|
|
|
|
|
|
def test_allow_instantiation_without_arguments(clear_sqlmodel):
|
|
class Item(SQLModel, table=True):
|
|
id: Optional[int] = Field(default=None, primary_key=True)
|
|
name: str
|
|
description: Optional[str] = None
|
|
|
|
engine = create_engine("sqlite:///:memory:")
|
|
SQLModel.metadata.create_all(engine)
|
|
with Session(engine) as db:
|
|
item = Item()
|
|
item.name = "Rick"
|
|
db.add(item)
|
|
db.commit()
|
|
statement = select(Item)
|
|
result = db.exec(statement).all()
|
|
assert len(result) == 1
|
|
assert isinstance(item.id, int)
|
|
SQLModel.metadata.clear()
|
|
|
|
|
|
def test_not_allow_instantiation_without_arguments_if_not_table():
|
|
class Item(SQLModel):
|
|
id: Optional[int] = Field(default=None, primary_key=True)
|
|
name: str
|
|
description: Optional[str] = None
|
|
|
|
with pytest.raises(ValidationError):
|
|
Item()
|