teufa/tests/conftest.py
2024-05-30 22:05:13 -06:00

32 lines
633 B
Python

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