initial tweaks on mocks

meedan · Nov 13, 2024 · 32ef971 · 32ef971
1 parent 06e5f2b
commit 32ef971
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/test/lib/queue/test_processor.py b/test/lib/queue/test_processor.py
@@ -16,7 +16,7 @@ def setUp(self, mock_get_env_setting, mock_boto_resource):
         self.mock_sqs_resource = MagicMock()
         self.mock_input_queue = MagicMock()
         self.mock_input_queue.url = "http://queue/mean_tokens__Model"
-        self.mock_sqs_resource.queues.filter.return_value = [self.mock_input_queue]
+        self.mock_sqs_resource.get_queue_by_name.return_value = self.mock_input_queue
         mock_boto_resource.return_value = self.mock_sqs_resource
 
         # Initialize the QueueProcessor instance

diff --git a/test/lib/queue/test_queue.py b/test/lib/queue/test_queue.py
@@ -49,7 +49,7 @@ def setUp(self, mock_log_execution_time, mock_get_env_setting, mock_boto_resourc
         self.mock_dlq_queue = MagicMock()
         self.mock_dlq_queue.url = f"http://queue/{self.queue_name_dlq}"
         self.mock_dlq_queue.attributes = {"QueueArn": f"queue:{self.queue_name_dlq}"}
-        self.mock_sqs_resource.queues.filter.return_value = [self.mock_input_queue, self.mock_output_queue, self.mock_dlq_queue]
+        self.mock_sqs_resource.get_queue_by_name.return_value = self.mock_input_queue#[, self.mock_output_queue, self.mock_dlq_queue]
         mock_boto_resource.return_value = self.mock_sqs_resource
 
         # Initialize the QueueWorker instance