Merge branch 'feat/hitless-upgrade-sync-standalone' into ps_add_fail_over_events_handling

Author: petyaslavova

redis/maintenance_events.py

@@ -15,6 +15,7 @@ class MaintenanceState(enum.Enum):
     FAILING_OVER = "failing_over"
 
 
+
 if TYPE_CHECKING:
     from redis.connection import (
         BlockingConnectionPool,
@@ -581,11 +582,13 @@ def handle_maintenance_start_event(self, maintenance_state: MaintenanceState):
             or not self.config.is_relax_timeouts_enabled()
         ):
             return
+
         self.connection.maintenance_state = maintenance_state
         self.connection.set_tmp_settings(tmp_relax_timeout=self.config.relax_timeout)
         # extend the timeout for all created connections
         self.connection.update_current_socket_timeout(self.config.relax_timeout)
 
+
     def handle_maintenance_completed_event(self):
         # Only reset timeouts if state is not MOVING and relax timeouts are enabled
         if (

tests/test_maintenance_events.py

@@ -283,7 +283,6 @@ def test_equality_and_hash(self):
         assert hash(event1) == hash(event2)
         assert hash(event1) != hash(event3)
 
-
 class TestNodeFailingOverEvent:
     """Test the NodeFailingOverEvent class."""
 
@@ -361,7 +360,6 @@ def test_equality_and_hash(self):
         assert hash(event1) == hash(event2)
         assert hash(event1) != hash(event3)
 
-
 class TestMaintenanceEventsConfig:
     """Test the MaintenanceEventsConfig class."""
 
@@ -564,6 +562,7 @@ def test_handle_event_migrating(self):
             self.handler.handle_event(event)
             mock_handle.assert_called_once_with(MaintenanceState.MIGRATING)
 
+
     def test_handle_event_migrated(self):
         """Test handling of NodeMigratedEvent."""
         event = NodeMigratedEvent(id=1)
@@ -668,6 +667,7 @@ def test_handle_maintenance_completed_event_success(self):
         self.handler.handle_maintenance_completed_event()
 
         assert self.mock_connection.maintenance_state == MaintenanceState.NONE
+
         self.mock_connection.update_current_socket_timeout.assert_called_once_with(-1)
         self.mock_connection.reset_tmp_settings.assert_called_once_with(
             reset_relax_timeout=True