From 3a1594b597d6667da4bc2e2323ea321ce3aaa2bf Mon Sep 17 00:00:00 2001
From: Leszek Hanusz <leszek.hanusz@gmail.com>
Date: Sun, 14 Jul 2024 18:20:11 +0200
Subject: [PATCH] Implement fix

---
 gql/transport/websockets_base.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/gql/transport/websockets_base.py b/gql/transport/websockets_base.py
index 45c96d3e..5c7713e9 100644
--- a/gql/transport/websockets_base.py
+++ b/gql/transport/websockets_base.py
@@ -512,7 +512,11 @@ async def connect(self) -> None:
                 await self._initialize()
             except ConnectionClosed as e:
                 raise e
-            except (TransportProtocolError, asyncio.TimeoutError) as e:
+            except (
+                TransportProtocolError,
+                TransportServerError,
+                asyncio.TimeoutError,
+            ) as e:
                 await self._fail(e, clean_close=False)
                 raise e