Coverage for tests/test_query.py: 100%

1from typing import Optional 7 ctx1hijklmn

2 

3import pytest 7 ctx1hijklmn

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

5 

6 

7def test_query(clear_sqlmodel): 7 ctx1hijklmn

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

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

10 name: str 7 ctx1abcdefg

11 secret_name: str 7 ctx1abcdefg

12 age: Optional[int] = None 7 ctx1abcdefg

13 

14 hero_1 = Hero(name="Deadpond", secret_name="Dive Wilson") 7 ctx1abcdefg

15 

16 engine = create_engine("sqlite://") 7 ctx1abcdefg

17 

18 SQLModel.metadata.create_all(engine) 7 ctx1abcdefg

19 with Session(engine) as session: 7 ctx1abcdefg

20 session.add(hero_1) 7 ctx1abcdefg

21 session.commit() 7 ctx1abcdefg

22 session.refresh(hero_1) 7 ctx1abcdefg

23 

24 with Session(engine) as session: 7 ctx1abcdefg

25 with pytest.warns(DeprecationWarning): 7 ctx1abcdefg

26 query_hero = session.query(Hero).first() 7 ctx1abcdefg

27 assert query_hero 7 ctx1abcdefg

28 assert query_hero.name == hero_1.name 7 ctx1abcdefg