Update ct-app/core/components/graphql_providers.py

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
hoprnet · May 2, 2024 · c9ff47d · c9ff47d
1 parent 48372f0
commit c9ff47d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/ct-app/core/components/graphql_providers.py b/ct-app/core/components/graphql_providers.py
@@ -55,7 +55,11 @@ async def _test_query(self, key: str, **kwargs) -> bool:
         vars.update(kwargs)
 
         # call `self._execute(self._sku_query, vars)` with a timeout
-        response = await asyncio.wait_for(self._execute(self._sku_query, vars), timeout=30)
+        try:
+            response = await asyncio.wait_for(self._execute(self._sku_query, vars), timeout=30)
+        except asyncio.TimeoutError:
+            self.error("Query timeout occurred")
+            return False
 
         return response and key in response