use Future.set_result() instead

tornadoweb · Aug 20, 2020 · 7b9e804 · 7b9e804
1 parent 4d5051a
commit 7b9e804
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tornado/websocket.py b/tornado/websocket.py
@@ -1435,12 +1435,12 @@ def close(self, code: Optional[int] = None, reason: Optional[str] = None) -> Non
             self.protocol = None  # type: ignore
 
     def on_connection_close(self) -> None:
-        if self.connect_future and not self.connect_future.done():
+        if not self.connect_future.done():
             self.connect_future.set_exception(StreamClosedError())
         self._on_message(None)
         self.tcp_client.close()
+        self.connect_future.set_result(None)
         super(WebSocketClientConnection, self).on_connection_close()
-        self.connect_future = None
 
     def on_ws_connection_close(
         self, close_code: Optional[int] = None, close_reason: Optional[str] = None