websocket_connect: don't hang on HTTP errors

protoss-player · protoss-player · commit a69a43b5ef5c · 2013-04-02T03:03:12.000+04:00
diff --git a/tornado/websocket.py b/tornado/websocket.py
@@ -746,6 +746,9 @@ def __init__(self, io_loop, request):
     def _on_close(self):
         self.on_message(None)
 
+    def _on_body(self, body):
+        self.connect_future.set_exception(Exception('Could not connect.'))
+
     def _handle_1xx(self, code):
         assert code == 101
         assert self.headers['Upgrade'].lower() == 'websocket'
