didier/tests/conftest.py

63 lines
1.6 KiB
Python
Raw Normal View History

2022-06-22 02:05:04 +02:00
import asyncio
from typing import AsyncGenerator, Generator
2022-06-21 21:06:11 +02:00
from unittest.mock import MagicMock
2022-06-17 01:02:36 +02:00
2022-06-17 00:43:55 +02:00
import pytest
from alembic import command, config
2022-06-17 01:02:36 +02:00
from sqlalchemy.ext.asyncio import AsyncSession
from database.engine import engine
2022-06-21 21:06:11 +02:00
from didier import Didier
2022-06-17 00:43:55 +02:00
@pytest.fixture(scope="session")
2022-06-22 02:05:04 +02:00
def event_loop() -> Generator:
loop = asyncio.get_event_loop_policy().new_event_loop()
yield loop
loop.close()
@pytest.fixture(scope="session")
def tables(event_loop):
2022-06-17 00:43:55 +02:00
"""Initialize a database before the tests, and then tear it down again
Starts from an empty database and runs through all the migrations to check those as well
while we're at it
"""
alembic_config = config.Config("alembic.ini")
command.upgrade(alembic_config, "head")
yield
command.downgrade(alembic_config, "base")
2022-06-17 01:02:36 +02:00
@pytest.fixture
2022-06-22 02:05:04 +02:00
async def database_session(tables, event_loop) -> AsyncGenerator[AsyncSession, None]:
2022-06-17 01:02:36 +02:00
"""Fixture to create a session for every test
Rollbacks the transaction afterwards so that the future tests start with a clean database
"""
connection = await engine.connect()
transaction = await connection.begin()
session = AsyncSession(bind=connection, expire_on_commit=False)
yield session
# Clean up session & rollback transactions
await session.close()
if transaction.is_valid:
await transaction.rollback()
await connection.close()
2022-06-21 21:06:11 +02:00
@pytest.fixture
def mock_client() -> Didier:
"""Fixture to get a mock Didier instance
The mock uses 0 as the id
"""
mock_client = MagicMock()
mock_user = MagicMock()
mock_user.id = 0
mock_client.user = mock_user
return mock_client