python - 如果我在测试期间推送两个应用程序上下文，会发生什么不好的事情吗？

Question

我正在使用 Flask、Mongo(带有 Flask-PyMongo)和 py.test。我的应用程序有一个 py.test 固定装置，正如您所期望的:

@pytest.fixture
def app(request):
    from proj import app

    ctx = app.app_context()
    ctx.push()
    request.addfinalizer(lambda: ctx.pop())

    return app

我还有一个 Mongo 装置。现在，mongo 对象是使用 Flask-PyMongo 创建的，因此它需要应用程序上下文才能工作。我提出了以下建议:

@pytest.fixture
def mongo():
    from proj import app, mongo

    ctx = app.app_context()
    ctx.push()
    request.addfinalizer(lambda: ctx.pop())

    return mongo

这是有效的，因为如果我有这样的测试，它就会有效:

def test_mongo(mongo):
    assert mongo.db.coll.find({'foo': 'bar'}).count() > 0

但是，在某些情况下，我希望同时访问应用程序和数据库:

def test_app_and_mongo(app, mongo):
    assert serious_stuff(app, mongo)

这将导致创建两个应用程序上下文。现在它已经可以工作了，但考虑到我还不太了解 Flask，我很谨慎。

这会导致什么不好的事情发生吗？因此我会看到任何令人惊讶的行为吗？

<小时/>

作为引用，proj 的 __init__.py 类似于:

from flask import Flask
from flask.ext.pymongo import PyMongo

app = Flask(__name__)
# ... set config ... 
mongo = PyMongo(app)

Answer 1

它不应该影响任何事情，但没有必要。使 mongo 固定装置依赖于已设置上下文的 app 固定装置。

import pytest
from proj import app as flask_app, mongo as mongo_ext

@pytest.fixture
def app():
    with flask_app.app_context():
        yield flask_app

@pytest.fixture
def mongo(app):
    return mongo_ext

def test_with_app(app):
    ...

def test_with_mongo(mongo):
    ...

def test_with_both(app, mongo):
    ...

<小时/>

由于 mongo 固定装置没有执行任何其他操作，因此您可以将其删除。直接导入扩展对象，并使您的测试依赖于 app fixture 来设置上下文。 (尽管我假设在真实的装置中您正在执行一些数据库设置/清理。)