azkaban-developers

Details

diff --git a/azkaban-common/src/test/java/azkaban/project/validator/ValidationReportTest.java b/azkaban-common/src/test/java/azkaban/project/validator/ValidationReportTest.java
new file mode 100644
index 0000000..a01ae80
--- /dev/null
+++ b/azkaban-common/src/test/java/azkaban/project/validator/ValidationReportTest.java
@@ -0,0 +1,59 @@
+package azkaban.project.validator;
+
+import static org.junit.Assert.*;
+
+import java.util.HashSet;
+import java.util.Set;
+
+import org.junit.Test;
+
+/**
+ * Test
+ */
+public class ValidationReportTest {
+
+  @Test
+  public void testAddWarnLevelInfoMsg() {
+    ValidationReport report = new ValidationReport();
+    String msg = "test warn level info message.";
+    report.addWarnLevelInfoMsg(msg);
+    for (String info : report.getInfoMsgs()) {
+      assertEquals("Info message added through addWarnLevelInfoMsg should have level set to WARN",
+          ValidationReport.getInfoMsgLevel(info), ValidationStatus.WARN);
+      assertEquals("Retrieved info message does not match the original one.",
+          ValidationReport.getInfoMsg(info), msg);
+    }
+  }
+
+  @Test
+  public void testAddErrorLevelInfoMsg() {
+    ValidationReport report = new ValidationReport();
+    String msg = "test error level error message.";
+    report.addErrorLevelInfoMsg(msg);
+    for (String info : report.getInfoMsgs()) {
+      assertEquals("Info message added through addErrorLevelInfoMsg should have level set to ERROR",
+          ValidationReport.getInfoMsgLevel(info), ValidationStatus.ERROR);
+      assertEquals("Retrieved info message does not match the original one.",
+          ValidationReport.getInfoMsg(info), msg);
+    }
+  }
+
+  @Test
+  public void testAddMsgs() {
+    ValidationReport report = new ValidationReport();
+    Set<String> msgs = new HashSet<String>();
+    msgs.add("test msg 1.");
+    msgs.add("test msg 2.");
+    report.addWarningMsgs(msgs);
+    assertEquals("Level of severity is not warn.",
+        report.getStatus(), ValidationStatus.WARN);
+    report.addErrorMsgs(msgs);
+    assertEquals("Number of error messages retrieved does not match.",
+        report.getErrorMsgs().size(), 2);
+    assertEquals("Number of warn messages retrieved does not match.",
+        report.getWarningMsgs().size(), 2);
+    assertEquals("Level of severity is not error.",
+        report.getStatus(), ValidationStatus.ERROR);
+  }
+
+}