| 3 | |
| 4 | |
| 5 | def test_query(clear_sqlmodel): |
| 6 | class Hero(SQLModel, table=True): |
| 7 | id: int | None = Field(default=None, primary_key=True) |
| 8 | name: str |
| 9 | secret_name: str |
| 10 | age: int | None = None |
| 11 | |
| 12 | hero_1 = Hero(name="Deadpond", secret_name="Dive Wilson") |
| 13 | |
| 14 | engine = create_engine("sqlite://") |
| 15 | |
| 16 | SQLModel.metadata.create_all(engine) |
| 17 | with Session(engine) as session: |
| 18 | session.add(hero_1) |
| 19 | session.commit() |
| 20 | session.refresh(hero_1) |
| 21 | |
| 22 | with Session(engine) as session: |
| 23 | with pytest.warns(DeprecationWarning): |
| 24 | query_hero = session.query(Hero).first() |
| 25 | assert query_hero |
| 26 | assert query_hero.name == hero_1.name |