Skip to content

Commit

Permalink
yield _fix_request_current_app middleware uncoditionally; update test…
Browse files Browse the repository at this point in the history
…; (see #2550)
  • Loading branch information
popravich committed Nov 30, 2017
1 parent 81594ec commit 75ba698
Show file tree
Hide file tree
Showing 2 changed files with 26 additions and 13 deletions.
3 changes: 1 addition & 2 deletions aiohttp/web.py
Original file line number Diff line number Diff line change
Expand Up @@ -268,8 +268,7 @@ def _prepare_middleware(self):
'see #2252'.format(m),
DeprecationWarning, stacklevel=2)
yield m, False
if self._middlewares:
yield _fix_request_current_app(self), True
yield _fix_request_current_app(self), True

async def _handle(self, request):
match_info = await self._router.resolve(request)
Expand Down
36 changes: 25 additions & 11 deletions tests/test_web_functional.py
Original file line number Diff line number Diff line change
Expand Up @@ -1290,32 +1290,46 @@ async def on_signal(app):
assert [app, subapp1, subapp2] == order


@pytest.mark.parametrize('route,expected', [
('/sub/', ['app see root', 'subapp see sub']),
('/', ['app see root']),
@pytest.mark.parametrize('route,expected,middlewares', [
('/sub/', ['A: root', 'C: sub', 'D: sub'], 'AC'),
('/', ['A: root', 'B: root'], 'AC'),
('/sub/', ['A: root', 'D: sub'], 'A'),
('/', ['A: root', 'B: root'], 'A'),
('/sub/', ['C: sub', 'D: sub'], 'C'),
('/', ['B: root'], 'C'),
('/sub/', ['D: sub'], ''),
('/', ['B: root'], ''),
])
async def test_subapp_middleware_context(loop, test_client, route, expected):
async def test_subapp_middleware_context(
loop, test_client, route, expected, middlewares):
values = []

def show_app_context(appname):
@web.middleware
async def middleware(request, handler):
values.append('{} see {}'.format(appname, request.app['my_value']))
values.append('{}: {}'.format(
appname, request.app['my_value']))
return await handler(request)
return middleware

async def handler(request):
return web.Response(text='Ok')
def make_handler(appname):
async def handler(request):
values.append('{}: {}'.format(
appname, request.app['my_value']))
return web.Response(text='Ok')
return handler

app = web.Application()
app['my_value'] = 'root'
app.middlewares.append(show_app_context('app'))
app.router.add_get('/', handler)
if 'A' in middlewares:
app.middlewares.append(show_app_context('A'))
app.router.add_get('/', make_handler('B'))

subapp = web.Application()
subapp['my_value'] = 'sub'
subapp.middlewares.append(show_app_context('subapp'))
subapp.router.add_get('/', handler)
if 'C' in middlewares:
subapp.middlewares.append(show_app_context('C'))
subapp.router.add_get('/', make_handler('D'))
app.add_subapp('/sub/', subapp)

client = await test_client(app)
Expand Down

0 comments on commit 75ba698

Please sign in to comment.