Coverage for tests/test_annotated_uuid.py: 100%

1import uuid 7 ctx1hijklmn

2from typing import Optional 7 ctx1hijklmn

3 

4from sqlmodel import Field, Session, SQLModel, create_engine, select 7 ctx1hijklmn

5 

6from tests.conftest import needs_pydanticv2 7 ctx1hijklmn

7 

8 

9@needs_pydanticv2 7 ctx1hijklmn

10def test_annotated_optional_types(clear_sqlmodel) -> None: 7 ctx1hijklmn

11 from pydantic import UUID4 7 ctx1abcdefg

12 

13 class Hero(SQLModel, table=True): 7 ctx1abcdefg

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

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

16 

17 engine = create_engine("sqlite:///:memory:") 7 ctx1abcdefg

18 SQLModel.metadata.create_all(engine) 7 ctx1abcdefg

19 with Session(engine) as db: 7 ctx1abcdefg

20 hero = Hero() 7 ctx1abcdefg

21 db.add(hero) 7 ctx1abcdefg

22 db.commit() 7 ctx1abcdefg

23 statement = select(Hero) 7 ctx1abcdefg

24 result = db.exec(statement).all() 7 ctx1abcdefg

25 assert len(result) == 1 7 ctx1abcdefg

26 assert isinstance(hero.id, uuid.UUID) 7 ctx1abcdefg