Update test

tests/test_web_websocket_functional.py

@@ -80,6 +80,8 @@ def connect_ws(self, url, protocol='chat'):
 
     def test_send_recv_text(self):
 
+        closed = asyncio.Future(loop=self.loop)
+
         @asyncio.coroutine
         def handler(request):
             ws = web.WebSocketResponse()
@@ -88,6 +90,10 @@ def handler(request):
             msg = yield from ws.receive()
             ws.send_str(msg+'/answer')
             ws.close()
+            try:
+                yield from ws.receive()
+            except web.WebSocketDisconnectedError:
+                closed.set_result(None)
             return ws
 
         @asyncio.coroutine
@@ -104,6 +110,10 @@ def go():
             self.assertEqual(msg.data, 1000)
             self.assertEqual(msg.extra, b'')
 
+            writer.close()
+
+            yield from closed
+
         self.loop.run_until_complete(go())
 
     def test_send_recv_bytes(self):