Coverage for tests/test_annotated_uuid.py: 100%

1import uuid 6 ctx1ghijkl

2from typing import Optional 6 ctx1ghijkl

3 

4from sqlmodel import Field, Session, SQLModel, create_engine, select 6 ctx1ghijkl

5 

6from tests.conftest import needs_pydanticv2 6 ctx1ghijkl

7 

8 

9@needs_pydanticv2 6 ctx1ghijkl

10def test_annotated_optional_types(clear_sqlmodel) -> None: 6 ctx1ghijkl

11 from pydantic import UUID4 6 ctx1abcdef

12 

13 class Hero(SQLModel, table=True): 6 ctx1abcdef

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

15 id: Optional[UUID4] = Field(default_factory=uuid.uuid4, primary_key=True) 6 ctx1abcdef

16 

17 engine = create_engine("sqlite:///:memory:") 6 ctx1abcdef

18 SQLModel.metadata.create_all(engine) 6 ctx1abcdef

19 with Session(engine) as db: 6 ctx1abcdef

20 hero = Hero() 6 ctx1abcdef

21 db.add(hero) 6 ctx1abcdef

22 db.commit() 6 ctx1abcdef

23 statement = select(Hero) 6 ctx1abcdef

24 result = db.exec(statement).all() 6 ctx1abcdef

25 assert len(result) == 1 6 ctx1abcdef

26 assert isinstance(hero.id, uuid.UUID) 6 ctx1abcdef