1 files changed, 131 insertions, 0 deletions
diff --git a/src/pybind/mgr/dashboard/tests/test_prometheus.py b/src/pybind/mgr/dashboard/tests/test_prometheus.py
new file mode 100644
index 000000000..cd2fb3e8d
--- /dev/null
+++ b/src/pybind/mgr/dashboard/tests/test_prometheus.py
@@ -0,0 +1,131 @@
+# -*- coding: utf-8 -*-
+# pylint: disable=protected-access
+try:
+    from mock import patch
+except ImportError:
+    from unittest.mock import patch
+
+from .. import mgr
+from ..controllers.prometheus import Prometheus, PrometheusNotifications, PrometheusReceiver
+from ..tests import ControllerTestCase
+
+
+class PrometheusControllerTest(ControllerTestCase):
+    alert_host = 'http://alertmanager:9093/mock'
+    alert_host_api = alert_host + '/api/v1'
+
+    prometheus_host = 'http://prometheus:9090/mock'
+    prometheus_host_api = prometheus_host + '/api/v1'
+
+    @classmethod
+    def setup_server(cls):
+        settings = {
+            'ALERTMANAGER_API_HOST': cls.alert_host,
+            'PROMETHEUS_API_HOST': cls.prometheus_host
+        }
+        mgr.get_module_option.side_effect = settings.get
+        cls.setup_controllers([Prometheus, PrometheusNotifications, PrometheusReceiver])
+
+    def test_rules(self):
+        with patch('requests.request') as mock_request:
+            self._get('/api/prometheus/rules')
+            mock_request.assert_called_with('GET', self.prometheus_host_api + '/rules',
+                                            json=None, params={}, verify=True)
+
+    def test_list(self):
+        with patch('requests.request') as mock_request:
+            self._get('/api/prometheus')
+            mock_request.assert_called_with('GET', self.alert_host_api + '/alerts',
+                                            json=None, params={}, verify=True)
+
+    def test_get_silences(self):
+        with patch('requests.request') as mock_request:
+            self._get('/api/prometheus/silences')
+            mock_request.assert_called_with('GET', self.alert_host_api + '/silences',
+                                            json=None, params={}, verify=True)
+
+    def test_add_silence(self):
+        with patch('requests.request') as mock_request:
+            self._post('/api/prometheus/silence', {'id': 'new-silence'})
+            mock_request.assert_called_with('POST', self.alert_host_api + '/silences',
+                                            params=None, json={'id': 'new-silence'},
+                                            verify=True)
+
+    def test_update_silence(self):
+        with patch('requests.request') as mock_request:
+            self._post('/api/prometheus/silence', {'id': 'update-silence'})
+            mock_request.assert_called_with('POST', self.alert_host_api + '/silences',
+                                            params=None, json={'id': 'update-silence'},
+                                            verify=True)
+
+    def test_expire_silence(self):
+        with patch('requests.request') as mock_request:
+            self._delete('/api/prometheus/silence/0')
+            mock_request.assert_called_with('DELETE', self.alert_host_api + '/silence/0',
+                                            json=None, params=None, verify=True)
+
+    def test_silences_empty_delete(self):
+        with patch('requests.request') as mock_request:
+            self._delete('/api/prometheus/silence')
+            mock_request.assert_not_called()
+
+    def test_post_on_receiver(self):
+        PrometheusReceiver.notifications = []
+        self._post('/api/prometheus_receiver', {'name': 'foo'})
+        self.assertEqual(len(PrometheusReceiver.notifications), 1)
+        notification = PrometheusReceiver.notifications[0]
+        self.assertEqual(notification['name'], 'foo')
+        self.assertTrue(len(notification['notified']) > 20)
+
+    def test_get_empty_list_with_no_notifications(self):
+        PrometheusReceiver.notifications = []
+        self._get('/api/prometheus/notifications')
+        self.assertStatus(200)
+        self.assertJsonBody([])
+        self._get('/api/prometheus/notifications?from=last')
+        self.assertStatus(200)
+        self.assertJsonBody([])
+
+    def test_get_all_notification(self):
+        PrometheusReceiver.notifications = []
+        self._post('/api/prometheus_receiver', {'name': 'foo'})
+        self._post('/api/prometheus_receiver', {'name': 'bar'})
+        self._get('/api/prometheus/notifications')
+        self.assertStatus(200)
+        self.assertJsonBody(PrometheusReceiver.notifications)
+
+    def test_get_last_notification_with_use_of_last_keyword(self):
+        PrometheusReceiver.notifications = []
+        self._post('/api/prometheus_receiver', {'name': 'foo'})
+        self._post('/api/prometheus_receiver', {'name': 'bar'})
+        self._get('/api/prometheus/notifications?from=last')
+        self.assertStatus(200)
+        last = PrometheusReceiver.notifications[1]
+        self.assertJsonBody([last])
+
+    def test_get_no_notification_with_unknown_id(self):
+        PrometheusReceiver.notifications = []
+        self._post('/api/prometheus_receiver', {'name': 'foo'})
+        self._post('/api/prometheus_receiver', {'name': 'bar'})
+        self._get('/api/prometheus/notifications?from=42')
+        self.assertStatus(200)
+        self.assertJsonBody([])
+
+    def test_get_no_notification_since_with_last_notification(self):
+        PrometheusReceiver.notifications = []
+        self._post('/api/prometheus_receiver', {'name': 'foo'})
+        notification = PrometheusReceiver.notifications[0]
+        self._get('/api/prometheus/notifications?from=' + notification['id'])
+        self.assertStatus(200)
+        self.assertJsonBody([])
+
+    def test_get_notifications_since_last_notification(self):
+        PrometheusReceiver.notifications = []
+        self._post('/api/prometheus_receiver', {'name': 'foobar'})
+        next_to_last = PrometheusReceiver.notifications[0]
+        self._post('/api/prometheus_receiver', {'name': 'foo'})
+        self._post('/api/prometheus_receiver', {'name': 'bar'})
+        self._get('/api/prometheus/notifications?from=' + next_to_last['id'])
+        forelast = PrometheusReceiver.notifications[1]
+        last = PrometheusReceiver.notifications[2]
+        self.assertEqual(self.json_body(), [forelast, last])