@After
public void tearDown() throws Exception {
String status = VerifyUtils.getPipelineState(serverURI, getPipelineName(), getPipelineRev());
if ("RUNNING".equals(status) || "STARTING".equals(status)) {
miniSDC.stopPipeline();
VerifyUtils.waitForPipelineToStop(serverURI, getPipelineName(), getPipelineRev());
}
}
@Before
public void setUp() throws Exception {
miniSDC.startPipeline();
VerifyUtils.waitForPipelineToStart(serverURI, getPipelineName(), getPipelineRev());
}
@Test(timeout = 20000)
public void testRules() throws Exception {
// Data rule Ids from the rule definition json
String dataRuleId1 = "dataRule1437173920545";
String dataRuleId2 = "dataRule1437173981080";
String dataRuleId3 = "dataRule1437174013953";
// Expected alert names for the above rule definitions
String alertId1 = "alert.dataRule1437173920545.gauge";
String alertId2 = "alert.dataRule1437173981080.gauge";
String alertId3 = "alert.dataRule1437174013953.gauge";
// Expected alert count for the rules
int expectedCountAlert1 = 100;
int expectedCountAlert2 = 50;
int expectedCountAlert3 = 25;
Map<String, Map<String, Object>> gauges = null;
int alertsPresent = 0;
// Wait until all 3 alerts are fired.
while (alertsPresent != 3) {
gauges = VerifyUtils.getGaugesFromMetrics(serverURI, getPipelineName(), getPipelineRev());
Assert.assertNotNull(gauges);
alertsPresent = 0;
for (Map.Entry<String, Map<String, Object>> e : gauges.entrySet()) {
if (e.getKey().contains("alert.")) {
alertsPresent++;
}
}
}
// check each alert
Assert.assertTrue(gauges.containsKey(alertId1));
Map<String, Object> map = (Map<String, Object>) gauges.get(alertId1).get("value");
Assert.assertTrue((Integer) map.get("currentValue") >= expectedCountAlert1);
Assert.assertTrue(gauges.containsKey(alertId2));
map = (Map<String, Object>) gauges.get(alertId2).get("value");
Assert.assertTrue((Integer) map.get("currentValue") >= expectedCountAlert2);
Assert.assertTrue(gauges.containsKey(alertId3));
map = (Map<String, Object>) gauges.get(alertId3).get("value");
Assert.assertTrue((Integer) map.get("currentValue") >= expectedCountAlert3);
// change rule definitions to a large number before attempting to delete
ruleDefinitions =
ruleDefinitions.replaceAll(
"\"thresholdValue\" : \"100\"", "\"thresholdValue\" : \"10000000\"");
ruleDefinitions =
ruleDefinitions.replaceAll(
"\"thresholdValue\" : \"50\"", "\"thresholdValue\" : \"5000000\"");
ruleDefinitions =
ruleDefinitions.replaceAll(
"\"thresholdValue\" : \"25\"", "\"thresholdValue\" : \"2500000\"");
ruleDefinitions = miniSDC.createRules(getPipelineName(), getPipelineRev(), ruleDefinitions);
printGuages(gauges);
// delete alert and wait until alerts are cleared.
// It takes about 2 seconds for the rules config loader to load changes to rules.
boolean alertRemoved = false;
while (!alertRemoved) {
VerifyUtils.deleteAlert(serverURI, getPipelineName(), getPipelineRev(), dataRuleId1);
gauges = VerifyUtils.getGaugesFromMetrics(serverURI, getPipelineName(), getPipelineRev());
Assert.assertNotNull(gauges);
alertRemoved = !gauges.containsKey(alertId1);
}
printGuages(gauges);
// changes have been reflected, delete without waiting
Assert.assertTrue(gauges.containsKey(alertId2));
Assert.assertTrue(gauges.containsKey(alertId3));
VerifyUtils.deleteAlert(serverURI, getPipelineName(), getPipelineRev(), dataRuleId2);
gauges = VerifyUtils.getGaugesFromMetrics(serverURI, getPipelineName(), getPipelineRev());
printGuages(gauges);
Assert.assertNotNull(gauges);
Assert.assertFalse(gauges.containsKey(alertId1));
Assert.assertFalse(gauges.containsKey(alertId2));
Assert.assertTrue(gauges.containsKey(alertId3));
VerifyUtils.deleteAlert(serverURI, getPipelineName(), getPipelineRev(), dataRuleId3);
gauges = VerifyUtils.getGaugesFromMetrics(serverURI, getPipelineName(), getPipelineRev());
printGuages(gauges);
Assert.assertNotNull(gauges);
Assert.assertFalse(gauges.containsKey(alertId1));
Assert.assertFalse(gauges.containsKey(alertId2));
Assert.assertFalse(gauges.containsKey(alertId3));
// zero alerts should be present at this point present
alertsPresent = 0;
for (Map.Entry<String, Map<String, Object>> e : gauges.entrySet()) {
if (e.getKey().contains("alert.")) {
alertsPresent++;
}
}
Assert.assertEquals(0, alertsPresent);
// recreate new rule definitions
ruleDefinitions =
ruleDefinitions.replaceAll(
"\"thresholdValue\" : \"10000000\"",
"\"thresholdValue\" : \"" + expectedCountAlert1 + "\"");
ruleDefinitions =
ruleDefinitions.replaceAll(
"\"thresholdValue\" : \"5000000\"",
"\"thresholdValue\" : \"" + expectedCountAlert2 + "\"");
ruleDefinitions =
ruleDefinitions.replaceAll(
"\"thresholdValue\" : \"2500000\"",
"\"thresholdValue\" : \"" + expectedCountAlert3 + "\"");
ruleDefinitions = miniSDC.createRules(getPipelineName(), getPipelineRev(), ruleDefinitions);
alertsPresent = 0;
while (alertsPresent != 3) {
gauges = VerifyUtils.getGaugesFromMetrics(serverURI, getPipelineName(), getPipelineRev());
Assert.assertNotNull(gauges);
// Expected 3 alerts
alertsPresent = 0;
for (Map.Entry<String, Map<String, Object>> e : gauges.entrySet()) {
if (e.getKey().contains("alert.")) {
alertsPresent++;
}
}
}
// check for alerts
Assert.assertTrue(gauges.containsKey(alertId1));
map = (Map<String, Object>) gauges.get(alertId1).get("value");
Integer currentValue = (Integer) map.get("currentValue");
Assert.assertTrue("Current value is : " + currentValue, currentValue >= expectedCountAlert1);
Assert.assertTrue(gauges.containsKey(alertId2));
map = (Map<String, Object>) gauges.get(alertId2).get("value");
Assert.assertTrue((Integer) map.get("currentValue") >= expectedCountAlert2);
Assert.assertTrue(gauges.containsKey(alertId3));
map = (Map<String, Object>) gauges.get(alertId3).get("value");
Assert.assertTrue((Integer) map.get("currentValue") >= expectedCountAlert3);
}