pytest тестирование Flask приложения с тестовой базой данных

import os
 
import pytest
 
from config import basedir
from webapp import create_app
from webapp import db as _db
from webapp.users.models import PermissionModel, RoleModel, UserModel
 
 
TESTDB = "test.db"
TESTDB_PATH = os.path.join(basedir, TESTDB)
TEST_DATABASE_URI = "sqlite:///" + TESTDB_PATH
 
 
@pytest.fixture(scope="session")
def app(request):
    """Session-wide test `Flask` application."""
    settings_override = {
        "TESTING": True,
        "SQLALCHEMY_DATABASE_URI": TEST_DATABASE_URI,
    }
    app = create_app(settings_override)
 
    # Establish an application context before running the tests.
    ctx = app.app_context()
    ctx.push()
 
    def teardown():
        ctx.pop()
 
    request.addfinalizer(teardown)
    return app
 
 
@pytest.fixture(scope="session")
def db(app, request):
    """Session-wide test database."""
    if os.path.exists(TESTDB_PATH):
        os.unlink(TESTDB_PATH)
 
    def teardown():
        _db.drop_all()
        try:
            os.unlink(TESTDB_PATH)
        except FileNotFoundError:
            pass
 
    _db.app = app
    _db.create_all()
 
    request.addfinalizer(teardown)
    return _db
 
 
@pytest.fixture(scope="function")
def session(db, request):
    """Creates a new database session for a test."""
    connection = db.engine.connect()
    transaction = connection.begin()
 
    options = dict(bind=connection, binds={})
    session = db.create_scoped_session(options=options)
 
    db.session = session
 
    def teardown():
        transaction.rollback()
        connection.close()
        session.remove()
 
    request.addfinalizer(teardown)
    return session

def test_new_permission(session):
    permission_title = "can_search_articles"
    permission = PermissionModel(permission_title)
    session.add(permission)
    session.commit()
 
    assert permission.title == permission_title