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