Minor fixes for networking

kratman · Nov 15, 2024 · d894dca · d894dca
1 parent f211c11
commit d894dca
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/shared.py b/tests/shared.py
@@ -336,12 +336,12 @@ def no_internet_connection():
         conn = socket.create_connection((host, 80), 2)
         conn.close()
         return False
-    except socket.gaierror:
+    except (socket.gaierror, TimeoutError):
         return True
 
 
 def assert_domain_equal(a, b):
-    "Check that two domains are equal, ignoring empty domains"
+    """Check that two domains are equal, ignoring empty domains"""
     a_dict = {k: v for k, v in a.items() if v != []}
     b_dict = {k: v for k, v in b.items() if v != []}
     assert a_dict == b_dict