Add a top-level exception class

src/simple_websocket/errors.py

@@ -1,14 +1,18 @@
 from wsproto.frame_protocol import CloseReason
 
 
-class ConnectionError(RuntimeError):  # pragma: no cover
+class SimpleWebsocketError(RuntimeError):
+    pass
+
+
+class ConnectionError(SimpleWebsocketError):
     """Connection error exception class."""
-    def __init__(self, status_code=None):
+    def __init__(self, status_code=None):  # pragma: no cover
         self.status_code = status_code
         super().__init__(f'Connection error: {status_code}')
 
 
-class ConnectionClosed(RuntimeError):
+class ConnectionClosed(SimpleWebsocketError):
     """Connection closed exception class."""
     def __init__(self, reason=CloseReason.NO_STATUS_RCVD, message=None):
         self.reason = reason

src/simple_websocket/ws.py

@@ -397,21 +397,18 @@ def __init__(self, url, subprotocols=None, headers=None,
         elif isinstance(headers, list):
             self.extra_headeers = headers
 
-        connection_args = None
-        for conn in socket.getaddrinfo(self.host, self.port):  # pragma: no branch
-            if conn[1] == socket.SOCK_STREAM:  # pragma: no branch
-                connection_args = conn
-                break
-        if connection_args is None:  # pragma: no cover
+        connection_args = socket.getaddrinfo(self.host, self.port,
+                                             type=socket.SOCK_STREAM)
+        if len(connection_args) == 0:  # pragma: no cover
             raise ConnectionError()
-        sock = socket.socket(connection_args[0], connection_args[1],
-                             connection_args[2])
+        sock = socket.socket(connection_args[0][0], connection_args[0][1],
+                             connection_args[0][2])
         if is_secure:  # pragma: no cover
             if ssl_context is None:
                 ssl_context = ssl.create_default_context(
                     purpose=ssl.Purpose.SERVER_AUTH)
             sock = ssl_context.wrap_socket(sock, server_hostname=self.host)
-        sock.connect(connection_args[4])
+        sock.connect(connection_args[0][4])
         super().__init__(sock, connection_type=ConnectionType.CLIENT,
                          receive_bytes=receive_bytes,
                          ping_interval=ping_interval,