conftest.py

import pytest
from sqlalchemy.orm import sessionmaker
from sqlalchemy_utils import create_database, database_exists, drop_database

from ezrules.backend import ezruleapp, ezrulevalapp
from ezrules.models.backend_core import Organisation, User
from ezrules.models.database import Base, engine
from ezrules.models.history_meta import versioned_session


@pytest.fixture(scope="session")
def logged_out_manager_client():
    ezruleapp.app.config["TESTING"] = True
    ezruleapp.app.config["WTF_CSRF_METHODS"] = []

    with ezruleapp.app.test_client() as client:
        yield client


@pytest.fixture(scope="session")
def logged_out_eval_client():
    ezrulevalapp.app.config["TESTING"] = True

    with ezrulevalapp.app.test_client() as client:
        yield client


@pytest.fixture(scope="session")
def engine_fix():
    if database_exists(engine.url):
        drop_database(engine.url)
    create_database(engine.url)

    Base.metadata.create_all(
        engine
    )  # Assuming Base is the declarative base from your models

    yield engine

    Base.metadata.drop_all(engine)
    drop_database(engine.url)


@pytest.fixture(scope="session")
def connection(engine_fix):
    connection = engine_fix.connect()
    yield connection
    connection.close()


@pytest.fixture(scope="function")
def session(connection):
    transaction = connection.begin()
    Session = sessionmaker(bind=connection)
    session = Session()
    versioned_session(session)

    org = Organisation(name="test_org")
    session.add(org)
    session.commit()

    admin_email = f"admin@test_org.com"
    admin_password = f"12345678"
    session.add(
        User(
            email=admin_email,
            password=admin_password,
            active=True,
            fs_uniquifier=admin_email,
        )
    )
    session.commit()

    ezruleapp.fsrm.db = session
    ezruleapp.fsrm.o_id = org.o_id
    ezruleapp.rule_engine_config_producer.db = session
    ezruleapp.rule_engine_config_producer.o_id = org.o_id
    ezrulevalapp.lre.db = session
    ezrulevalapp.lre.o_id = org.o_id

    yield session

    session.close()
    transaction.rollback()


@pytest.fixture
def logged_in_manager_client(session, logged_out_manager_client):
    # Log in the test user
    logged_out_manager_client.post(
        "/login",
        data=dict(email="admin@test_org.com", password="12345678"),
        follow_redirects=True,
    )
    yield logged_out_manager_client