Coverage for tests / test_query.py: 100%

1import pytest 8 ctx1ijklmnop

2from sqlmodel import Field, Session, SQLModel, create_engine 8 ctx1ijklmnop

3 

4 

5def test_query(clear_sqlmodel): 8 ctx1ijklmnop

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

7 id: int | None = Field(default=None, primary_key=True) 8 ctx1abcdefgh

8 name: str 8 ctx1abcdefgh

9 secret_name: str 8 ctx1abcdefgh

10 age: int | None = None 8 ctx1abcdefgh

11 

12 hero_1 = Hero(name="Deadpond", secret_name="Dive Wilson") 8 ctx1abcdefgh

13 

14 engine = create_engine("sqlite://") 8 ctx1abcdefgh

15 

16 SQLModel.metadata.create_all(engine) 8 ctx1abcdefgh

17 with Session(engine) as session: 8 ctx1abcdefgh

18 session.add(hero_1) 8 ctx1abcdefgh

19 session.commit() 8 ctx1abcdefgh

20 session.refresh(hero_1) 8 ctx1abcdefgh

21 

22 with Session(engine) as session: 8 ctx1abcdefgh

23 with pytest.warns(DeprecationWarning): 8 ctx1abcdefgh

24 query_hero = session.query(Hero).first() 8 ctx1abcdefgh

25 assert query_hero 8 ctx1abcdefgh

26 assert query_hero.name == hero_1.name 8 ctx1abcdefgh