Add CLOSED as an expected type (home-assistant#83180)

nnkostov · Dec 3, 2022 · 0dfc0af · 0dfc0af
1 parent 6cef376
commit 0dfc0af
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/homeassistant/components/websocket_api/http.py b/homeassistant/components/websocket_api/http.py
@@ -214,7 +214,7 @@ def handle_hass_stop(event: Event) -> None:
                 disconnect_warn = "Did not receive auth message within 10 seconds"
                 raise Disconnect from err
 
-            if msg.type in (WSMsgType.CLOSE, WSMsgType.CLOSING):
+            if msg.type in (WSMsgType.CLOSE, WSMsgType.CLOSED, WSMsgType.CLOSING):
                 raise Disconnect
 
             if msg.type != WSMsgType.TEXT:
@@ -238,7 +238,7 @@ def handle_hass_stop(event: Event) -> None:
             while not wsock.closed:
                 msg = await wsock.receive()
 
-                if msg.type in (WSMsgType.CLOSE, WSMsgType.CLOSING):
+                if msg.type in (WSMsgType.CLOSE, WSMsgType.CLOSED, WSMsgType.CLOSING):
                     break
 
                 if msg.type != WSMsgType.TEXT: