conftest.py

import pytest
from alembic.command import upgrade
from yarl import URL

from staff.main import create_app
from staff.utils import alembic_config_from_url, tmp_database


@pytest.fixture(scope='session')
def migrated_postgres_template(pg_url):
    """
    Creates temporary database and applies migrations.
    Database can be used as template to fast creation databases for tests.

    Has "session" scope, so is called only once per tests run.
    """
    with tmp_database(pg_url, 'template') as tmp_url:
        alembic_config = alembic_config_from_url(tmp_url)
        upgrade(alembic_config, 'head')
        yield tmp_url


@pytest.fixture
def migrated_postgres(pg_url, migrated_postgres_template):
    """
    Quickly creates clean migrated database using temporary database as base.
    Use this fixture in tests that require migrated database.
    """
    template_db = URL(migrated_postgres_template).name
    with tmp_database(pg_url, 'pytest', template=template_db) as tmp_url:
        yield tmp_url


@pytest.fixture
async def api_client(migrated_postgres, aiohttp_client):
    """
    Creates aiohttp application and client for it.
    """
    app = create_app(migrated_postgres)
    client = await aiohttp_client(app)
    try:
        yield client
    finally:
        await client.close()