python - 在集成测试 (Pytest) 和 Flask 应用程序之间创建不同的 SQLAlchemy 会话

Question

我有使用 flask-sqlalchemy lib 的烧瓶应用程序（RESTful API 后端）。集成测试使用带有固定装置的 Pytest，有些为测试目的创建记录。问题是，当测试预计会在数据库级别引发唯一约束失败的场景时，通过夹具为测试创建的记录都被回滚，导致其他测试失败并出现错误sqlalchemy.exc.InvalidRequestError: This Session's transaction has been rolled back due to a previous exception during flush. To begin a new transaction with this Session, first issue Session.rollback(). Original exception was: (sqlite3.IntegrityError) UNIQUE constraint failed: my_table.field1。

如何为测试创建不同的 SQLAlchemy 会话，以便测试记录可以提交到 DB 并且不受 Flask 请求生命周期内发生的错误的影响？

### globals.py ###

from flask_sqlalchemy import SQLAlchemy

db = SQLAlchemy()

### app.py ###

from sqlalchemy.exc import IntegrityError
from .globals import db

def handle_unique_constraint(error):
    return {"msg": "Duplicate"}, 409

def create_app(connection_string):
    app = Flask(__name__)
    app.register_error_handler(IntegrityError, handle_unique_constraint)
    app.config['SQLALCHEMY_DATABASE_URI'] = connection_string

    # register API blueprint ...
    # e.g. create new user record, done like this:
    #
    # db.session.add(User(**{'email': path_param}))
    # db.session.commit()

    db.init_app(app)
    return app

### conftest.py ###

from my_package.app import create_app
from my_package.globals import db as app_db
from my_package.models.user import User

@fixture(scope='session')
def app(request):
    app = create_app('https://localhost')
    app.debug = True

    with app.app_context():
        yield app

@fixture(scope='session')
def db(app):
    app_db.create_all()
    return app_db

@fixture(scope='session')
def client(app):

    with app.test_client() as client:
        yield client

@fixture(scope='function')
def test_user(db):
    user = User(**{'email': generate_random()})
    db.session.add(user)
    db.session.commit()
    db.session.refresh(user)

### test_user.py ###



# This test passses

def test_repeat_email(client, test_user):
    resp = client.post('/users/emails/{}'.format(test_user.email))
    assert resp.status_code == 409


# This test errors out during setting up test_user fixture
# with aforementioned error message

def test_good_email(client, test_user): # <- this 
    resp = client.post('/users/emails/{}'.format('[e-mail]'))
    assert resp.status_code == 201

标签： pythonflasksqlalchemyflask-sqlalchemypytest

Answer 1

您必须实现 asetUp和 a tearDown。

运行测试时，setUp将在每个测试开始时运行。将tearDown在每个结束时运行。

在setUp你将：初始化数据库

在tearnDown你将：关闭数据库连接，删除创建的项目......

---

我不熟悉pytest。根据this和this。

在yield它之前是setUp零件，在它之后是tearDown。

你应该有这样的东西：

@fixture(scope='session')
def app(request):
    app = create_app('https://localhost')
    app.debug = True
    
    ctx = app.app_context()
    ctx.push()

    yield app

    ctx.pop()


@fixture(scope='session')
def db(app):
    app_db.create_all()
    yield app_db
    app_db.drop_all()