Fix redirects

aiohttp_debugtoolbar/middlewares.py

@@ -123,7 +123,9 @@ def toolbar_middleware(request):
         if intercept_redirects:
             # Intercept http redirect codes and display an html page with a
             # link to the target.
-            if response.status in REDIRECT_CODES and response.location:
+            if response.status in REDIRECT_CODES \
+                    and getattr(response, 'location', None):
+
                 context = {'redirect_to': response.location,
                            'redirect_code': response.status}
 

tests/test_middleware.py

@@ -65,6 +65,24 @@ def handler(request):
     assert 'Redirect intercepted' in txt
 
 
+@asyncio.coroutine
+def test_no_location_no_intercept(create_server, test_client):
+
+    @asyncio.coroutine
+    def handler(request):
+        return web.Response(text="no location", status=301)
+
+    app = yield from create_server()
+    app.router.add_route('GET', '/', handler)
+    client = yield from test_client(app)
+
+    resp = yield from client.get('/', allow_redirects=False)
+    txt = yield from resp.text()
+    assert 301 == resp.status
+    assert 'location' not in resp.headers
+    assert 'no location' in txt
+
+
 @asyncio.coroutine
 def test_intercept_redirects_disabled(create_server, test_client):
     @asyncio.coroutine