feat(tests): refactor connection state

signalrcore/helpers.py

@@ -10,7 +10,7 @@ def configure_logger(level=logging.INFO, handler=None):
             handler = logging.StreamHandler()
             handler.setFormatter(
                 logging.Formatter(
-                    '%(asctime)s - %(name)s - %(levelname)s - %(message)s'))
+                    '%(asctime)s [%(filename)s:%(lineno)d] %(name)s - %(levelname)s - %(message)s'))
             handler.setLevel(level)
         logger.addHandler(handler)
         logger.setLevel(level)

signalrcore/hub/errors.py

@@ -10,3 +10,11 @@ class HubConnectionError(ValueError):
     """Hub connection error
     """
     pass
+
+
+class HubReconnectingError(HubError):
+    pass
+
+
+class HandShakeError(HubError):
+    pass

signalrcore/transport/websockets/connection.py

@@ -2,7 +2,6 @@
 
 
 class ConnectionState(Enum):
-    connecting = 0
-    connected = 1
-    reconnecting = 2
-    disconnected = 4
+    disconnected = 0
+    connecting = 1
+    connected = 2

signalrcore/transport/websockets/reconnection.py

@@ -43,6 +43,9 @@ def __init__(self):
         self.reconnecting = False
         self.attempt_number = 0
         self.last_attempt = time.time()
+
+    def can_reconnect(self) -> bool:
+        raise NotImplementedError()
 
     def next(self):
         raise NotImplementedError()
@@ -58,6 +61,9 @@ def __init__(self, sleep_time, max_attempts):
         self.sleep_time = sleep_time
         self.max_reconnection_attempts = max_attempts
 
+    def can_reconnect(self) -> bool:
+        return False
+
     def next(self):
         self.reconnecting = True
         if self.max_reconnection_attempts is not None:
@@ -76,6 +82,9 @@ class IntervalReconnectionHandler(ReconnectionHandler):
     def __init__(self, intervals):
         super(IntervalReconnectionHandler, self).__init__()
         self._intervals = intervals
+
+    def can_reconnect(self) -> bool:
+        return self.attempt_number >= len(self._intervals)
 
     def next(self):
         self.reconnecting = True