[Example conftest.py for fastapi/sqlalchemy] #fastapi #pytest

conftest.py

# From @euri10 -- https://gitter.im/tiangolo/fastapi?at=5cd915ed56271260f95275ac

import asyncio

import pytest
from sqlalchemy import create_engine
from sqlalchemy_utils import create_database, database_exists, drop_database
from starlette.config import environ
from starlette.testclient import TestClient

# This sets `os.environ`, but provides some additional protection.
# If we placed it below the application import, it would raise an error
# informing us that 'TESTING' had already been read from the environment.

environ["TESTING"] = "True"
environ["EMAILS_ENABLED"] = "False"

from app.app import app  # isort:skip
from app.models import User, UserInCreate, clients, metadata, users  # isort:skip
from app.settings import database  # isort:skip

@pytest.fixture(scope="session", autouse=True)
def create_test_database():
    url = str(database.url)
    engine = create_engine(url)
    if database_exists(url):
        drop_database(url)
    create_database(url)  # Create the test database.
    metadata.create_all(engine)  # Create the tables.
    yield  # Run the tests.
    drop_database(url)  # Drop the test database.