Coverage for tests/test_query.py: 100%

1from typing import Optional 10 ctx1klmnopqrst

2 

3import pytest 10 ctx1klmnopqrst

4from sqlmodel import Field, Session, SQLModel, create_engine 10 ctx1klmnopqrst

5 

6 

7def test_query(clear_sqlmodel): 10 ctx1klmnopqrst

8 class Hero(SQLModel, table=True): 10 ctx1abcdefghij

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

10 name: str 10 ctx1abcdefghij

11 secret_name: str 10 ctx1abcdefghij

12 age: Optional[int] = None 10 ctx1abcdefghij

13 

14 hero_1 = Hero(name="Deadpond", secret_name="Dive Wilson") 10 ctx1abcdefghij

15 

16 engine = create_engine("sqlite://") 10 ctx1abcdefghij

17 

18 SQLModel.metadata.create_all(engine) 10 ctx1abcdefghij

19 with Session(engine) as session: 10 ctx1abcdefghij

20 session.add(hero_1) 10 ctx1abcdefghij

21 session.commit() 10 ctx1abcdefghij

22 session.refresh(hero_1) 10 ctx1abcdefghij

23 

24 with Session(engine) as session: 10 ctx1abcdefghij

25 with pytest.warns(DeprecationWarning): 10 ctx1abcdefghij

26 query_hero = session.query(Hero).first() 10 ctx1abcdefghij

27 assert query_hero 10 ctx1abcdefghij

28 assert query_hero.name == hero_1.name 10 ctx1abcdefghij