Handle new non-numeric sasl events

hifi · Aug 15, 2023 · 1191c70 · 1191c70
1 parent 73435c9
commit 1191c70
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/heisenbridge/network_room.py b/heisenbridge/network_room.py
@@ -1494,8 +1494,10 @@ async def _connect(self) -> None:
                             self.conn.send_items("AUTHENTICATE", base64.b64encode(sasl.encode("utf8")).decode("utf8"))
                         else:
                             self.conn.send_items("AUTHENTICATE", "+")
-                        (connection, event) = await self.conn.expect(["903", "904", "908"])
-                        if event.type != "903":
+                        (connection, event) = await self.conn.expect(
+                            ["903", "904", "908", "saslsuccess", "saslfail", "saslmechs"]
+                        )
+                        if event.type not in ["903", "saslsuccess"]:
                             raise irc.client.ServerConnectionError(event.arguments[0])
 
                     except asyncio.TimeoutError: