Coverage for tests/test_fields_set.py: 100%

1from datetime import datetime, timedelta 10 ctx1klmnopqrst

2 

3from sqlmodel import Field, SQLModel 10 ctx1klmnopqrst

4from sqlmodel._compat import get_fields_set 10 ctx1klmnopqrst

5 

6 

7def test_fields_set(): 10 ctx1klmnopqrst

8 class User(SQLModel): 10 ctx1abcdefghij

9 username: str 10 ctx1abcdefghij

10 email: str = "test@test.com" 10 ctx1abcdefghij

11 last_updated: datetime = Field(default_factory=datetime.now) 10 ctx1abcdefghij

12 

13 user = User(username="bob") 10 ctx1abcdefghij

14 assert get_fields_set(user) == {"username"} 10 ctx1abcdefghij

15 user = User(username="bob", email="bob@test.com") 10 ctx1abcdefghij

16 assert get_fields_set(user) == {"username", "email"} 10 ctx1abcdefghij

17 user = User( 10 ctx1abcdefghij

18 username="bob", 

19 email="bob@test.com", 

20 last_updated=datetime.now() - timedelta(days=1), 

21 ) 

22 assert get_fields_set(user) == {"username", "email", "last_updated"} 10 ctx1abcdefghij