adding more tests

LifeguardSystem · Jun 21, 2021 · 13bcb40 · 13bcb40
1 parent 20d0975
commit 13bcb40
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -1,3 +1,6 @@
+"""
+Lifeguard RabbitMQ Plugin
+"""
 from setuptools import find_packages, setup
 
 setup(

diff --git a/tests/test_validations.py b/tests/test_validations.py
@@ -100,9 +100,11 @@ def test_consumers_running_validation_when_problem(self, mock_count_consumers):
 
     @patch("lifeguard_rabbitmq.validations.count_consumers")
     @patch("lifeguard_rabbitmq.validations.logger")
+    @patch("lifeguard_rabbitmq.validations.traceback")
     def test_consumers_running_validation_when_problem_because_integration_error(
-        self, mock_logger, mock_count_consumers
+        self, mock_traceback, mock_logger, mock_count_consumers
     ):
+        mock_traceback.format_exc.return_value = "traceback"
         mock_count_consumers.side_effect = [Exception("error")]
 
         response = consumers_running_validation()
@@ -121,6 +123,9 @@ def test_consumers_running_validation_when_problem_because_integration_error(
                 ]
             },
         )
+        mock_logger.error.assert_called_with(
+            "error on recover queue infos %s", "error", extra={"traceback": "traceback"}
+        )
 
     @patch("lifeguard_rabbitmq.validations.number_of_messages")
     @patch("lifeguard_rabbitmq.validations.logger")