Add test for the websocket connection-refused fix.

diff --git a/tornado/test/websocket_test.py b/tornado/test/websocket_test.py
index 28375cd69ac7d842293740f6e613a492b8ee96f2..3c6e3a9a2cc06a4bb9b865cf59b60188cec133a9 100644 (file)
--- a/tornado/test/websocket_test.py
+++ b/tornado/test/websocket_test.py
@@ -86,7 +86,7 @@ class WebSocketTest(AsyncHTTPTestCase):
                 io_loop=self.io_loop)
 
     @gen_test
-    def test_websocket_network_fail(self):
+    def test_websocket_network_timeout(self):
         sock, port = bind_unused_port()
         sock.close()
         with self.assertRaises(HTTPError) as cm:
@@ -97,6 +97,18 @@ class WebSocketTest(AsyncHTTPTestCase):
                     connect_timeout=0.01)
         self.assertEqual(cm.exception.code, 599)
 
+    @gen_test
+    def test_websocket_network_fail(self):
+        sock, port = bind_unused_port()
+        sock.close()
+        with self.assertRaises(HTTPError) as cm:
+            with ExpectLog(gen_log, ".*"):
+                yield websocket_connect(
+                    'ws://localhost:%d/' % port,
+                    io_loop=self.io_loop,
+                    connect_timeout=3600)
+        self.assertEqual(cm.exception.code, 599)
+
     @gen_test
     def test_websocket_close_buffered_data(self):
         ws = yield websocket_connect(