teufa/tests/conftest.py

import pytest
from flask import Flask, g
from flask.testing import FlaskClient

from teufa import db as dbm
from teufa.app import create_app
from teufa.ext import db


@pytest.fixture
def app():
    app = create_app()
    with app.app_context():
        dbm.Base.metadata.create_all(db.engine)

        tenant = dbm.Tenant(
            name="Default",
            hostname="localhost",
        )
        db.session.add(tenant)
        db.session.commit()
        g.tenant = tenant

        yield app

        dbm.Base.metadata.drop_all(db.engine)


@pytest.fixture
def client(app: Flask) -> FlaskClient:
    return app.test_client()