【问题标题】:Will anything bad happen if I push two app contexts during testing?如果我在测试期间推送两个应用程序上下文,会发生什么不好的事情吗?
【发布时间】:2016-07-03 01:51:52
【问题描述】:

我正在使用 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)

【问题讨论】:

    标签: python mongodb flask pytest fixtures


    【解决方案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 夹具来设置上下文。 (虽然我假设在真正的夹具中你正在做一些数据库设置/清理。)

    【讨论】:

      猜你喜欢
      • 2011-06-02
      • 1970-01-01
      • 2017-04-18
      • 2021-06-02
      • 1970-01-01
      • 2013-08-11
      • 1970-01-01
      • 1970-01-01
      • 2019-01-13
      相关资源
      最近更新 更多