"""Database tests."""


class TestDatabase:
    def test_init_db_creates_tables(self, db_module):
        from app.db.database import init_db
        init_db()  # Should not raise

    def test_get_db_yields_and_closes(self, db_module):
        gen = db_module.get_db()
        session = next(gen)
        assert session is not None
        session.close()
        try:
            next(gen)
        except StopIteration:
            pass