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}})