Add request_ctx fixture to access request context

Author: vitalk

pytest_flask/fixtures.py

@@ -10,6 +10,8 @@
 except ImportError:
     from urllib.request import urlopen
 
+from flask import _request_ctx_stack
+
 
 @pytest.yield_fixture
 def client(app):
@@ -132,6 +134,14 @@ def config(app):
     return app.config
 
 
+@pytest.fixture
+def request_ctx(app):
+    """The request context which contains all request relevant information,
+    e.g. `session`, `g`, `flashes`, etc.
+    """
+    return _request_ctx_stack.top
+
+
 @pytest.fixture(params=['application/json', 'text/html'])
 def mimetype(request):
     return request.param

pytest_flask/plugin.py

@@ -13,7 +13,7 @@
 
 from .fixtures import (
     client, config, accept_json, accept_jsonp, accept_any, accept_mimetype,
-    client_class, live_server
+    client_class, live_server, request_ctx
 )
 
 

tests/test_fixtures.py

@@ -2,17 +2,14 @@
 # -*- coding: utf-8 -*-
 import pytest
 
-from flask import url_for
+from flask import request, url_for
 
 
 class TestFixtures:
 
     def test_config_access(self, config):
         assert config['SECRET_KEY'] == '42'
 
-    def test_application_has_pushed_context(self, app):
-        assert url_for('ping') == '/ping'
-
     def test_client(self, client):
         assert client.get(url_for('ping')).status == b'200 OK'
 
@@ -22,6 +19,13 @@ def test_accept_json(self, accept_json):
     def test_accept_jsonp(self, accept_jsonp):
         assert accept_jsonp == [('Accept', 'application/json-p')]
 
+    def test_request_ctx(self, app, request_ctx):
+        assert request_ctx.app is app
+
+    def test_request_ctx_is_kept_around(self, client):
+        res = client.get(url_for('index'), headers=[('X-Something', '42')])
+        assert request.headers['X-Something'] == '42'
+
 
 class TestJSONResponse: