import pytest
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker

from coldcall_lti.db import Base


@pytest.fixture
def db_session():
    engine = create_engine("sqlite://")
    Base.metadata.create_all(engine)
    session = sessionmaker(bind=engine)()
    yield session
    session.close()
    engine.dispose()