system-pri/backend/tests/conftest.py

39 lines
861 B
Python

from typing import Generator
import pytest
from apiflask import APIFlask
from flask import Flask
from flask.testing import FlaskClient
from flask.ctx import AppContext
from ..app import create_app
from ..app.dependencies import db
@pytest.fixture()
def test_app() -> Generator[Flask, None, None]:
yield create_app("testing")
@pytest.fixture()
def test_app_ctx_with_db(test_app) -> Generator[AppContext, None, None]:
with test_app.app_context() as ctx:
db.create_all()
yield ctx
@pytest.fixture()
def test_client() -> FlaskClient:
app = create_app("testing")
with app.app_context():
db.create_all()
return app.test_client()
@pytest.fixture()
def test_app_with_context() -> Generator[APIFlask, None, None]:
app = create_app("testing")
with app.app_context():
db.create_all()
yield app