Add new unit testing

Author: mrlockstar

manage_breast_screening/notifications/tests/services/test_queue.py

@@ -23,7 +23,16 @@ def mock_queue_client(self):
             queue_client.from_connection_string.return_value = mock_client
             yield mock_client
 
-    def test_queue_is_created(self):
+    @pytest.fixture
+    def mock_metrics(self):
+        with patch(
+            "manage_breast_screening.notifications.services.queue.Metrics"
+        ) as mock_metics:
+            mock_instance = MagicMock()
+            mock_metics.return_value = mock_instance
+            yield mock_metics, mock_instance
+
+    def test_queue_is_created(self, mock_metrics):
         with patch(
             "manage_breast_screening.notifications.services.queue.QueueClient"
         ) as queue_client:
@@ -36,12 +45,49 @@ def test_queue_is_created(self):
                 "qqq111", "new-queue"
             )
             mock_client.create_queue.assert_called_once()
+            mock_metrics[0].assert_called_once_with(
+                "new-queue", "messages", "Queue length"
+            )
+
+    def test_add_to_queue_records_metrics(self, mock_queue_client, mock_metrics):
+        _, mock_metrics_instance = mock_metrics
+
+        mock_queue_client.get_queue_properties.return_value.approximate_message_count = 82
+
+        queue = Queue("new-queue")
+        queue.client = mock_queue_client
+        queue.add("testing123")
+
+        mock_queue_client.send_message.assert_called_once_with("testing123")
+        mock_queue_client.get_queue_properties.assert_called_once()
+        mock_metrics_instance.add.assert_called_once_with(82)
+
+    def test_records_metrics_exception(self, mock_queue_client, mock_metrics):
+        _, mock_metrics_instance = mock_metrics
+        mock_queue_client.get_queue_properties.side_effect = Exception("An exception")
+
+        queue = Queue("new-queue")
+        queue.client = mock_queue_client
+        queue.recordMetrics()
+
+        mock_metrics_instance.add.assert_not_called()
+
+    def test_delete_from_queue_records_metrics(self, mock_queue_client, mock_metrics):
+        _, mock_metrics_instance = mock_metrics
+        mock_queue_client.get_queue_properties.return_value.approximate_message_count = 8
+
+        queue = Queue("new-queue")
+        queue.client = mock_queue_client
+        queue.add("testing123")
+
+        mock_queue_client.send_message.assert_called_once_with("testing123")
+        mock_queue_client.get_queue_properties.assert_called_once()
+        mock_metrics_instance.add.assert_called_once_with(8)
 
     def test_add_to_queue(self, mock_queue_client):
         Queue("new-queue").add("a message")
 
         mock_queue_client.send_message.assert_called_once_with("a message")
-        # mock_queue_client.
 
     def test_message_status_updates_queue(self):
         with patch(