Websockets: Stop iteration when connection closes.

See issue aio-libs#1144.
frederikaalund · Oct 19, 2016 · 87fe03b · 87fe03b
1 parent 0033c48
commit 87fe03b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/aiohttp/web_ws.py b/aiohttp/web_ws.py
@@ -315,6 +315,6 @@ def __aiter__(self):
         @asyncio.coroutine
         def __anext__(self):
             msg = yield from self.receive()
-            if msg.type == WSMsgType.CLOSE:
+            if msg.type == WSMsgType.CLOSE or msg.type == WSMsgType.CLOSED:
                 raise StopAsyncIteration  # NOQA
             return msg