add JSON termcolor tests

tenable · Oct 1, 2020 · c2a3b43 · c2a3b43
1 parent 77be67d
commit c2a3b43
Showing 1 changed file with 61 additions and 0 deletions.
diff --git a/pkg/termcolor/writer_test.go b/pkg/termcolor/writer_test.go
@@ -57,6 +57,8 @@ func verifyLineWithStringIsColorized(s string, buf string, t *testing.T) {
     }
 }
 
+///////////  YAML
+
 func TestYAMLBogusSeverityIsNotColorized(t *testing.T) {
     re := regexp.MustCompile(`(?m)^(.*severity.*)$`)
     m := re.FindString(yamlData.String())
@@ -113,3 +115,62 @@ func TestYAMLCountMediumIsColorized(t *testing.T) {
 func TestYAMLCountHighIsColorized(t *testing.T) {
     verifyLineWithStringIsColorized("high", yamlData.String(), t)
 }
+
+///////////  JSON
+
+func TestJSONBogusSeverityIsNotColorized(t *testing.T) {
+    re := regexp.MustCompile(`(?m)^(.*severity.*)$`)
+    m := re.FindString(jsonData.String())
+    if strings.Contains(m, ColorPrefix) {
+        t.Errorf("severity is colorized [%v]\n%s", m, jsonData.String())
+    }
+}
+
+func TestJSONRuleNameIsColorized(t *testing.T) {
+    verifyLineWithStringIsColorized("rule_name", jsonData.String(), t)
+}
+func TestJSONDescriptionIsColorized(t *testing.T) {
+    verifyLineWithStringIsColorized("description", jsonData.String(), t)
+}
+func TestJSONSeverityIsColorized(t *testing.T) {
+
+    res := buildStore()
+    yw := &strings.Builder{}
+    w := NewColorizedWriter(yw)
+
+    // HIGH, MEDIUM, LOW
+    testSeverity := func(sev string) {
+        res.Violations[0].Severity = sev
+        yw.Reset()
+        err := writer.Write("json", res, w)
+        if err != nil {
+            panic(err)
+        }
+        verifyLineWithStringIsColorized("severity", yw.String(), t)
+    }
+    testSeverity("HIGH")
+    testSeverity("MEDIUM")
+    testSeverity("LOW")
+}
+func TestJSONResourceNameIsColorized(t *testing.T) {
+    verifyLineWithStringIsColorized("resource_name", jsonData.String(), t)
+}
+func TestJSONResourceTypeIsColorized(t *testing.T) {
+    verifyLineWithStringIsColorized("resource_type", jsonData.String(), t)
+}
+func TestJSONFileIsColorized(t *testing.T) {
+    verifyLineWithStringIsColorized("file", jsonData.String(), t)
+}
+
+func TestJSONCountIsColorized(t *testing.T) {
+    verifyLineWithStringIsColorized("count", jsonData.String(), t)
+}
+func TestJSONCountLowIsColorized(t *testing.T) {
+    verifyLineWithStringIsColorized("low", jsonData.String(), t)
+}
+func TestJSONCountMediumIsColorized(t *testing.T) {
+    verifyLineWithStringIsColorized("medium", jsonData.String(), t)
+}
+func TestJSONCountHighIsColorized(t *testing.T) {
+    verifyLineWithStringIsColorized("high", jsonData.String(), t)
+}