diff options
author | Paul Duncan <pabs@pablotron.org> | 2022-02-01 13:49:24 -0500 |
---|---|---|
committer | Paul Duncan <pabs@pablotron.org> | 2022-02-01 13:49:24 -0500 |
commit | 806a74bdff6cf2f014bb93d01d84611c9ef26e04 (patch) | |
tree | b8aa5bb2c9c57a1991b4ed82f02a3082982e8ee7 /internal/feed/severity_test.go | |
parent | ee4857cdd5416686e36624621b3460f1bc334c54 (diff) | |
download | cvez-806a74bdff6cf2f014bb93d01d84611c9ef26e04.tar.bz2 cvez-806a74bdff6cf2f014bb93d01d84611c9ef26e04.zip |
internal/feed: add severity, severity tests
Diffstat (limited to 'internal/feed/severity_test.go')
-rw-r--r-- | internal/feed/severity_test.go | 83 |
1 files changed, 83 insertions, 0 deletions
diff --git a/internal/feed/severity_test.go b/internal/feed/severity_test.go new file mode 100644 index 0000000..75aec72 --- /dev/null +++ b/internal/feed/severity_test.go @@ -0,0 +1,83 @@ +package feed + +import ( + "encoding/json" + "testing" +) + +func TestSeverityUnmarshalInvalidData(t *testing.T) { + test := []byte(`{}`) + var val Severity + + if err := json.Unmarshal(test, &val); err == nil { + t.Errorf("got \"%s\", exp error", val) + } +} + +func TestSeverityUnmarshalUnknown(t *testing.T) { + test := []byte(`"foo"`) + exp := "unknown severity: foo" + var val Severity + + err := json.Unmarshal(test, &val) + if err == nil { + t.Errorf("got \"%s\", exp error", val) + return + } + + if err.Error() != exp { + t.Errorf("got \"%s\", exp \"%s\"", err.Error(), exp) + } +} + +func TestSeverityUnmarshalValid(t *testing.T) { + tests := []struct { + val string + exp Severity + } { + { "\"NONE\"", SeverityNone }, + { "\"LOW\"", SeverityLow }, + { "\"MEDIUM\"", SeverityMedium }, + { "\"HIGH\"", SeverityHigh }, + { "\"CRITICAL\"", SeverityCritical }, + } + + for _, test := range(tests) { + t.Run(test.val, func(t *testing.T) { + var got Severity + if err := json.Unmarshal([]byte(test.val), &got); err != nil { + t.Error(err) + return + } + + if got != test.exp { + t.Errorf("got \"%s\", exp \"%s\"", got, test.exp) + } + }) + } +} + +func TestSeverityString(t *testing.T) { + tests := []struct { + val Severity + exp string + } { + { SeverityNone, "NONE" }, + { SeverityLow, "LOW" }, + { SeverityMedium, "MEDIUM" }, + { SeverityHigh, "HIGH" }, + { SeverityCritical, "CRITICAL" }, + + { Severity(255), "Severity(255)" }, + } + + for _, test := range(tests) { + t.Run(test.exp, func(t *testing.T) { + got := test.val.String() + + if got != test.exp { + t.Errorf("got \"%s\", exp \"%s\"", got, test.exp) + } + }) + } +} |