24 lines
436 B
Python
24 lines
436 B
Python
from pytest import fixture
|
|
from sqlalchemy import Integer, String, Column
|
|
|
|
from zvk.util.db import Database, DBBase
|
|
|
|
|
|
|
|
|
|
class SomeTable(DBBase):
|
|
id = Column(Integer, primary_key=True)
|
|
|
|
s = Column(String)
|
|
|
|
|
|
def test_db(db):
|
|
db.create_all()
|
|
|
|
with db as session:
|
|
session.add(SomeTable(s='123'))
|
|
session.add(SomeTable(s='132'))
|
|
session.add(SomeTable(s='1232132132'))
|
|
|
|
assert len(session.query(SomeTable).all()) == 3
|