Coverage for tests/test_query.py: 100%

1from typing import Optional 6 ctx1ghijkl

2 

3import pytest 6 ctx1ghijkl

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

5 

6 

7def test_query(clear_sqlmodel): 6 ctx1ghijkl

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

9 id: Optional[int] = Field(default=None, primary_key=True) 6 ctx1abcdef

10 name: str 6 ctx1abcdef

11 secret_name: str 6 ctx1abcdef

12 age: Optional[int] = None 6 ctx1abcdef

13 

14 hero_1 = Hero(name="Deadpond", secret_name="Dive Wilson") 6 ctx1abcdef

15 

16 engine = create_engine("sqlite://") 6 ctx1abcdef

17 

18 SQLModel.metadata.create_all(engine) 6 ctx1abcdef

19 with Session(engine) as session: 6 ctx1abcdef

20 session.add(hero_1) 6 ctx1abcdef

21 session.commit() 6 ctx1abcdef

22 session.refresh(hero_1) 6 ctx1abcdef

23 

24 with Session(engine) as session: 6 ctx1abcdef

25 with pytest.warns(DeprecationWarning): 6 ctx1abcdef

26 query_hero = session.query(Hero).first() 6 ctx1abcdef

27 assert query_hero 6 ctx1abcdef

28 assert query_hero.name == hero_1.name 6 ctx1abcdef