Fixing tests

Author: petyaslavova

tests/test_maintenance_events.py

@@ -443,7 +443,7 @@ def test_handle_node_moving_event_success(self):
             assert event in self.handler._processed_events
 
             # Verify pool methods were called
-            self.mock_pool.add_tmp_config_to_connection_kwargs.assert_called_once()
+            self.mock_pool.update_connections_settings.assert_called_once()
 
     def test_handle_node_moved_event(self):
         """Test handling of node moved event (cleanup)."""
@@ -454,7 +454,7 @@ def test_handle_node_moved_event(self):
         self.handler.handle_node_moved_event(event)
 
         # Verify cleanup methods were called
-        self.mock_pool.remove_tmp_config_from_connection_kwargs.assert_called_once()
+        self.mock_pool.update_connections_settings.assert_called_once()
 
 
 class TestMaintenanceEventConnectionHandler: