conftest.py

import calendar
from datetime import datetime

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

from app.config import PSQL_ENVIRONMENT
from app.database.models import Base
from app.routers.event import create_event

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

# 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)


# the following fixtures are meant to replace user1, user2,user3
# fixtures inside test_user. they do the same but those have more indicative
# name, so after the feature freeze ill make sure to replace them completely.
# did not delete user1, user2 and user3 since some people
# use them in testing now.


@pytest.fixture
def event_example(session, event_owning_user):
    data = {
        "title": "test event title",
        "start": datetime.strptime("2021-05-05 14:59", "%Y-%m-%d %H:%M"),
        "end": datetime.strptime("2021-05-05 15:01", "%Y-%m-%d %H:%M"),
        "location": "https://us02web.zoom.us/j/87538459r6",
        "content": "content",
        "owner_id": event_owning_user.id,
    }

    event = create_event(session, **data)
    return event


nest_asyncio.apply()