feat/58 - improve Backpressure algorithm

tests/service/test_amq_service.py

@@ -1,8 +1,7 @@
 import unittest
-from unittest.mock import AsyncMock, MagicMock, Mock, patch
+from unittest.mock import AsyncMock, Mock, patch
 
 from amqp.config.amq_configuration import AMQAdapter, AMQConfiguration
-from amqp.model.model import DataMessage
 from amqp.service.amq_service import AMQService
 
 
@@ -14,6 +13,11 @@ class TestAMQService(unittest.IsolatedAsyncioTestCase):
         self.mock_adapter.route_mapping = {}
         self.mock_adapter.swarm_service_id = "test-swarm-id"
         self.mock_adapter.swarm_task_id = "test-task-id"
+        self.mock_adapter.consul_host = "consul"
+        self.mock_adapter.consul_port = 4500
+        self.mock_adapter.consul_counter_key = "services/ids/counter"
+        self.mock_adapter.consul_max_retries = 5
+        self.mock_adapter.consul_initial_retry_delay = 0.2
 
         self.mock_config = Mock(spec=AMQConfiguration)
         self.mock_config.amq_adapter = self.mock_adapter
@@ -29,7 +33,7 @@ class TestAMQService(unittest.IsolatedAsyncioTestCase):
 
         # Add backpressure configuration
         self.mock_backpressure = Mock()
-        self.mock_backpressure.threshold_threads = 10
+        self.mock_backpressure.threshold_load = 10
         self.mock_config.backpressure = self.mock_backpressure
 
         # Create service instance with mocked configuration
@@ -145,22 +149,6 @@ class TestAMQService(unittest.IsolatedAsyncioTestCase):
         # Verify that client was reinitialized
         mock_client.init.assert_called_once()
 
-    def test_send_message(self):
-        # Create mock message and future
-        mock_message = Mock(spec=DataMessage)
-        mock_message.id.return_value = "test-id"
-        mock_future = MagicMock()
-
-        # Setup mock message handler
-        self.service.amq_data_message_handler = Mock()
-        self.service.amq_data_message_handler.outstanding = {}
-
-        # Send message
-        self.service.send_message(mock_message, mock_future)
-
-        # Verify message was added to outstanding
-        self.assertEqual(self.service.amq_data_message_handler.outstanding["test-id"], mock_future)
-
     def test_remove_outstanding_future(self):
         # Setup mock message handler with outstanding message
         self.service.amq_data_message_handler = Mock()