import pytest

from sqlalchemy.future.engine import Engine
from sqlmodel import SQLModel, Session

from .config import config_fixture

from diyalgo import models
from diyalgo.db import init_db_engine, create_tables

@pytest.fixture()
def engine_fixture(config_fixture) -> Engine:
    """Create and engine and then create tables"""
    engine = init_db_engine(config_fixture)
    SQLModel.metadata.create_all(engine)

    yield engine

    engine.dispose()

@pytest.fixture()
def session_fixture(engine_fixture) -> Session:
    return Session(engine_fixture)