使用app工厂时,在pytest测试中访问Flask测试客户端会话

我正在尝试使用pytest和app工厂对应用程序进行单元测试,但我似乎无法在我的测试中访问客户端会话对象.我确定有一些背景我不是在推动某个地方.我在我的’app’夹具中推送应用程序上下文.我应该在某处推送请求上下文吗？

以下是MWE.

mwe.py：

from flask import Flask, session


def create_app():
    app = Flask(__name__)
    app.secret_key = 'top secret'

    @app.route('/set')
    def session_set():
        session['key'] = 'value'
        return 'Set'

    @app.route('/check')
    def session_check():
        return str('key' in session)

    @app.route('/clear')
    def session_clear():
        session.pop('key', None)
        return 'Cleared'

    return app


if __name__ == "__main__":
    mwe = create_app()
    mwe.run()

conftest.py：

import pytest
from mwe import create_app


@pytest.fixture(scope='session')
def app(request):
    app = create_app()

    ctx = app.app_context()
    ctx.push()

    def teardown():
        ctx.pop()

    request.addfinalizer(teardown)
    return app


@pytest.fixture(scope='function')
def client(app):
    return app.test_client()

test_session.py：

import pytest
from flask import session


def test_value_set_for_client_request(client):  # PASS
    client.get('/set')
    r = client.get('/check')
    assert 'True' in r.data


def test_value_set_in_session(client):  # FAIL
    client.get('/set')
    assert 'key' in session


def test_value_set_in_session_transaction(client):  # FAIL
    with client.session_transaction() as sess:
        client.get('/set')
        assert 'key' in sess

请注意,直接运行它可以正常工作,我可以跳转/设置,/检查,/清除它,它的行为符合预期.类似地,仅使用测试客户端来获取页面的测试按预期工作.但是,似乎没有直接访问会话.

解决方法:

从cookie000-flask中的conftest.py开始查看以下内容.它可能会给你一些想法.

@pytest.yield_fixture(scope='function') def app():
    """An application for the tests."""
    _app = create_app(TestConfig)
    ctx = _app.test_request_context()
    ctx.push()

    yield _app

    ctx.pop()


@pytest.fixture(scope='function') def testapp(app):
    """A Webtest app."""
    return TestApp(app)