conftest.py

import calendar

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

from app.config import PSQL_ENVIRONMENT
from app.database.models import Base

pytest_plugins = [
    "tests.user_fixture",
    "tests.event_fixture",
    "tests.dayview_fixture",
    "tests.invitation_fixture",
    "tests.association_fixture",
    "tests.client_fixture",
    "tests.asyncio_fixture",
    "tests.logger_fixture",
    "tests.category_fixture",
    "smtpdfix",
    "tests.quotes_fixture",
    "tests.zodiac_fixture",
    "tests.jokes_fixture",
    "tests.comment_fixture",
]

# When testing in a PostgreSQL environment please make sure that:
#   - Base string is a PSQL string
#   - app.config.PSQL_ENVIRONMENT is set to True

if PSQL_ENVIRONMENT:
    SQLALCHEMY_TEST_DATABASE_URL = (
        "postgresql://postgres:1234" "@localhost/postgres"
    )
    test_engine = create_engine(SQLALCHEMY_TEST_DATABASE_URL)

else:
    SQLALCHEMY_TEST_DATABASE_URL = "sqlite:///./test.db"
    test_engine = create_engine(
        SQLALCHEMY_TEST_DATABASE_URL,
        connect_args={"check_same_thread": False},
    )

TestingSessionLocal = sessionmaker(
    autocommit=False,
    autoflush=False,
    bind=test_engine,
)


def get_test_db():
    return TestingSessionLocal()


@pytest.fixture
def session():
    Base.metadata.create_all(bind=test_engine)
    session = get_test_db()
    yield session
    session.rollback()
    session.close()
    Base.metadata.drop_all(bind=test_engine)


@pytest.fixture
def sqlite_engine():
    SQLALCHEMY_TEST_DATABASE_URL = "sqlite:///./test.db"
    sqlite_test_engine = create_engine(
        SQLALCHEMY_TEST_DATABASE_URL,
        connect_args={"check_same_thread": False},
    )

    TestingSession = sessionmaker(
        autocommit=False,
        autoflush=False,
        bind=sqlite_test_engine,
    )

    yield sqlite_test_engine
    session = TestingSession()
    session.close()
    Base.metadata.drop_all(bind=sqlite_test_engine)


@pytest.fixture
def Calendar():
    return calendar.Calendar(0)