From: qcha0 <agnewee@gmail.com>
Date: Fri, 3 Sep 2021 10:19:44 +0000 (+0800)
Subject: Fix websocket client already closed raise error does not match
X-Git-Tag: v6.2.0b1~27^2~1
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=14a52b3657faf24440ba7cb6d07e1faccad0e61f;p=thirdparty%2Ftornado.git

Fix websocket client already closed raise error does not match

As doc says "If the stream is closed, raises WebSocketClosedError"
---

diff --git a/tornado/websocket.py b/tornado/websocket.py
index c88cc2074..f2711c94f 100644
--- a/tornado/websocket.py
+++ b/tornado/websocket.py
@@ -1501,6 +1501,8 @@ class WebSocketClientConnection(simple_httpclient._HTTPConnection):
            Exception raised on a closed stream changed from `.StreamClosedError`
            to `WebSocketClosedError`.
         """
+        if self.protocol is None:
+            raise WebSocketClosedError("Client connection has been closed")
         return self.protocol.write_message(message, binary=binary)
 
     def read_message(