minitor/tests/alert_test.py

from datetime import datetime
from unittest.mock import patch

import pytest

from minitor.main import Alert
from minitor.main import Monitor
from tests.util import assert_called_once_with


class TestAlert(object):

    @pytest.fixture
    def monitor(self):
        return Monitor({
            'name': 'Dummy Monitor',
            'command': ['echo', 'foo'],
        })

    @pytest.fixture
    def echo_alert(self):
        return Alert(
            'log',
            {
                'command': [
                    'echo', (
                        '{monitor_name} has failed {failure_count} time(s)!\n'
                        'We have alerted {alert_count} time(s)\n'
                        'Last success was {last_success}\n'
                        'Last output was: {last_output}'
                    )
                ]
            }
        )

    @pytest.mark.parametrize(
        'last_success,expected_success',
        [
            (None, 'Never'),
            (datetime(2018, 4, 10), '2018-04-10T00:00:00')
        ]
    )
    def test_simple_alert(
        self,
        monitor,
        echo_alert,
        last_success,
        expected_success
    ):
        monitor.alert_count = 1
        monitor.last_output = 'beep boop'
        monitor.last_success = last_success
        monitor.total_failure_count = 1
        with patch.object(echo_alert._logger, 'error') as mock_error:
            echo_alert.alert('Exception message', monitor)
            assert_called_once_with(
                mock_error,
                'Dummy Monitor has failed 1 time(s)!\n'
                'We have alerted 1 time(s)\n'
                'Last success was ' + expected_success + '\n'
                'Last output was: beep boop'
            )