def test_alerts(self):
alert1 = {
'name': 'AlertName1',
'uuid': '11'
}
alert2 = {
'name': 'AlertName2',
'uuid': '12'
}
alert3 = {
'name': 'AlertName3',
'uuid': '13'
}
alert4 = {
'name': 'AlertName4',
'uuid': '14'
}
controller = AlertCollector()
controller._AlertCollector__buckets = {
'TestCluster1': {
'AlertName1': alert1,
'AlertName2': alert2
},
'TestCluster2': {
'AlertName3': alert3,
'AlertName4': alert4
}
}
list = controller.alerts()
self.assertEquals(controller._AlertCollector__buckets, {})
self.assertEquals(list.sort(), [alert1, alert2, alert3, alert4].sort())
def test_put_clusterExists(self):
cluster = 'TestCluster'
alert = {
'name': 'AlertName',
'uuid': '12'
}
collector = AlertCollector()
collector._AlertCollector__buckets = {
'TestCluster': {}
}
collector.put(cluster, alert)
self.assertEquals(collector._AlertCollector__buckets, {'TestCluster': {'AlertName': alert}})
def test_remove_noAlert(self):
alert2 = {
'name': 'AlertName2',
'uuid': '12'
}
controller = AlertCollector()
controller._AlertCollector__buckets = {
'TestCluster2': {
'AlertName2': alert2
}
}
controller.remove('TestCluster', 'AlertName1')
self.assertEquals(controller._AlertCollector__buckets, {'TestCluster2': {'AlertName2': alert2}})
def test_remove_by_uuid_absent(self):
alert1 = {
'name': 'AlertName1',
'uuid': '11'
}
alert2 = {
'name': 'AlertName2',
'uuid': '12'
}
controller = AlertCollector()
controller._AlertCollector__buckets = {
'TestCluster': {
'AlertName1': alert1,
'AlertName2': alert2
}
}
controller.remove_by_uuid('13')
self.assertEquals(controller._AlertCollector__buckets, {'TestCluster': {'AlertName1': alert1, 'AlertName2': alert2}})
