Coverage for tests / test_annotated_uuid.py: 100%

1import uuid 8 ctx1ijklmnop

2 

3from sqlmodel import Field, Session, SQLModel, create_engine, select 8 ctx1ijklmnop

4 

5 

6def test_annotated_optional_types(clear_sqlmodel) -> None: 8 ctx1ijklmnop

7 from pydantic import UUID4 8 ctx1abcdefgh

8 

9 class Hero(SQLModel, table=True): 8 ctx1abcdefgh

10 # Pydantic UUID4 is: Annotated[UUID, UuidVersion(4)] 

11 id: UUID4 | None = Field(default_factory=uuid.uuid4, primary_key=True) 8 ctx1abcdefgh

12 

13 engine = create_engine("sqlite:///:memory:") 8 ctx1abcdefgh

14 SQLModel.metadata.create_all(engine) 8 ctx1abcdefgh

15 with Session(engine) as db: 8 ctx1abcdefgh

16 hero = Hero() 8 ctx1abcdefgh

17 db.add(hero) 8 ctx1abcdefgh

18 db.commit() 8 ctx1abcdefgh

19 statement = select(Hero) 8 ctx1abcdefgh

20 result = db.exec(statement).all() 8 ctx1abcdefgh

21 assert len(result) == 1 8 ctx1abcdefgh

22 assert isinstance(hero.id, uuid.UUID) 8 ctx1abcdefgh