🐛 Fix fields marked as "set" in models (#117)

Co-authored-by: Michael Statt <michael.statt@modelyst.io>
Co-authored-by: Sebastián Ramírez <tiangolo@gmail.com>
This commit is contained in:
statt8900
2022-08-27 18:59:09 -04:00
committed by GitHub
parent 9c68ce12ec
commit 680602b7eb
2 changed files with 22 additions and 1 deletions

21
tests/test_fields_set.py Normal file
View File

@@ -0,0 +1,21 @@
from datetime import datetime, timedelta
from sqlmodel import Field, SQLModel
def test_fields_set():
class User(SQLModel):
username: str
email: str = "test@test.com"
last_updated: datetime = Field(default_factory=datetime.now)
user = User(username="bob")
assert user.__fields_set__ == {"username"}
user = User(username="bob", email="bob@test.com")
assert user.__fields_set__ == {"username", "email"}
user = User(
username="bob",
email="bob@test.com",
last_updated=datetime.now() - timedelta(days=1),
)
assert user.__fields_set__ == {"username", "email", "last_updated"}