1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
# __init__.py -- The tests for prometheus_xmpp
# Copyright (C) 2018 Jelmer Vernooij <jelmer@jelmer.uk>
#
from datetime import datetime
import pytz
import unittest
from prometheus_xmpp import \
create_message_short, create_message_full, parse_timestring
class CreateMessageTests(unittest.TestCase):
message = {
"version": "4",
"groupKey": "test",
"status": "firing",
"receiver": "xmpp",
"groupLabels": {
"groupLabel1": "groupLabelValue1",
"groupLabel2": "groupLabelValue2"
},
"commonLabels": {
"commonLabel1": "commonLabelValue1",
"commonLabel2": "commonLabelValue2"
},
"commonAnnotations": {
"commonAnnotation1": "commonAnnotationValue1",
"commonAnnotation2": "commonAnnotationValue2"
},
"externalURL": "https://alertmanager.example.com",
"alerts": [
{
"status": "firing",
"labels": {
"test": "true",
"severity": "test"
},
"annotations": {
"summary": "Test Alert",
"description": "This is just a test alert."
},
"startsAt": "2019-04-12T23:20:50.123456789Z",
"endsAt": "2019-04-12T23:20:50.123456789Z",
"generatorURL": "curl"
}
]
}
def test_create_message_short(self):
self.assertEqual(
['FIRING, 2019-04-12T23:20:50+00:00, Test Alert'],
list(create_message_short(self.message)))
def test_create_message_full(self):
self.assertEqual(
['*[FIRING] Test Alert* (groupLabelValue1 groupLabelValue2)'
+ '\nThis is just a test alert.'
+ '\n*test:* true'
+ '\n*severity:* test'],
list(create_message_full(self.message)))
class ParseTimestringTests(unittest.TestCase):
def test_parse_with_nanoseconds(self):
self.assertEqual(
datetime(2019, 4, 27, 5, 33, 35, 739602, pytz.utc),
parse_timestring('2019-04-27T05:33:35.739602132Z'))
def test_parse_with_microseconds(self):
self.assertEqual(
datetime(2019, 4, 27, 5, 33, 35, 739602, pytz.utc),
parse_timestring('2019-04-27T05:33:35.739602Z'))
def test_parse_with_timezone(self):
self.assertEqual(
datetime(2019, 4, 27, 5, 33, 35, 739602, pytz.utc),
parse_timestring('2019-04-27T05:33:35.739602+00:00'))
def test_suite():
module_names = ['prometheus_xmpp.tests']
loader = unittest.TestLoader()
return loader.loadTestsFromNames(module_names)
|
