def test_send_notification_sends_to_slack_and_telegram_when_options_are_set(self): rmqa = rabbitmqalert.RabbitMQAlert() options = self.construct_options() optionsresolver.OptionsResover.setup_options = mock.MagicMock(return_value=options) rabbitmqalert.smtplib = mock.MagicMock() rabbitmqalert.urllib2 = mock.MagicMock() rmqa.send_notification(options, "") rabbitmqalert.urllib2.urlopen.assert_called()
def test_send_notification_not_call_login_when_email_password_not_set(self): rmqa = rabbitmqalert.RabbitMQAlert() options = self.construct_options() optionsresolver.OptionsResover.setup_options = mock.MagicMock(return_value=options) rabbitmqalert.smtplib = mock.MagicMock() rabbitmqalert.urllib2 = mock.MagicMock() rmqa.send_notification(options, "") rabbitmqalert.smtplib.SMTP().login.assert_not_called()
def test_send_notification_sends_email_when_email_to_is_set(self): rmqa = rabbitmqalert.RabbitMQAlert() options = self.construct_options() optionsresolver.OptionsResover.setup_options = mock.MagicMock(return_value=options) rabbitmqalert.smtplib = mock.MagicMock() rabbitmqalert.urllib2 = mock.MagicMock() rmqa.send_notification(options, "") rabbitmqalert.smtplib.SMTP().sendmail.assert_called_once()
def test_check_connection_conditions_open_connections_not_send_notification_when_exceeding_option(self): rmqa = rabbitmqalert.RabbitMQAlert() response = self.construct_response_connection() rmqa.send_request = mock.MagicMock(return_value=response) options = self.construct_options() optionsresolver.OptionsResover.setup_options = mock.MagicMock(return_value=options) rmqa.send_notification = mock.MagicMock() rmqa.check_connection_conditions(options) rmqa.send_notification.assert_not_called()
def test_send_notification_not_send_to_telegram_when_any_option_not_set(self): rmqa = rabbitmqalert.RabbitMQAlert() options = self.construct_options() options["telegram_bot_id"] = None optionsresolver.OptionsResover.setup_options = mock.MagicMock(return_value=options) rabbitmqalert.smtplib = mock.MagicMock() rabbitmqalert.urllib2 = mock.MagicMock() rmqa.send_notification(options, "") # Assure only slack is called rabbitmqalert.urllib2.urlopen.assert_called_once()
def test_check_queue_conditions_messages_unacknowledged_send_notification_when_exceeding_option(self): rmqa = rabbitmqalert.RabbitMQAlert() response = self.construct_response_queue() response["messages_unacknowledged"] = 2 rmqa.send_request = mock.MagicMock(return_value=response) options = self.construct_options() optionsresolver.OptionsResover.setup_options = mock.MagicMock(return_value=options) rmqa.send_notification = mock.MagicMock() rmqa.check_queue_conditions(options) rmqa.send_notification.assert_called_once()
def test_check_node_conditions_send_notification_when_node_memory_exceeding_option(self): rmqa = rabbitmqalert.RabbitMQAlert() response = self.construct_response_node() response[0]["mem_used"] = 2000000 rmqa.send_request = mock.MagicMock(return_value=response) options = self.construct_options() optionsresolver.OptionsResover.setup_options = mock.MagicMock(return_value=options) rmqa.send_notification = mock.MagicMock() rmqa.check_node_conditions(options) rmqa.send_notification.assert_called_once()
def test_check_connection_conditions_open_connections_send_notification_when_beneath_option(self): rmqa = rabbitmqalert.RabbitMQAlert() response = self.construct_response_connection() response.pop("connection_foo") response.pop("connection_bar") rmqa.send_request = mock.MagicMock(return_value=response) options = self.construct_options() optionsresolver.OptionsResover.setup_options = mock.MagicMock(return_value=options) rmqa.send_notification = mock.MagicMock() rmqa.check_connection_conditions(options) rmqa.send_notification.assert_called_once()