diff options
author | Paul Duncan <pabs@pablotron.org> | 2022-02-01 12:48:15 -0500 |
---|---|---|
committer | Paul Duncan <pabs@pablotron.org> | 2022-02-01 12:48:15 -0500 |
commit | 465f427a04c2f10bc5163f6bceef13693f7e99e6 (patch) | |
tree | ad4c963ee1b918fee01436030ceda2e6c882b225 /internal/feed/v3impact_test.go | |
parent | f4f3bf7f03e790b5d98004de3498c102bdac50ad (diff) | |
download | cvez-465f427a04c2f10bc5163f6bceef13693f7e99e6.tar.bz2 cvez-465f427a04c2f10bc5163f6bceef13693f7e99e6.zip |
internal/feed: add v2impact, v3impact, and tests
Diffstat (limited to 'internal/feed/v3impact_test.go')
-rw-r--r-- | internal/feed/v3impact_test.go | 79 |
1 files changed, 79 insertions, 0 deletions
diff --git a/internal/feed/v3impact_test.go b/internal/feed/v3impact_test.go new file mode 100644 index 0000000..a369f44 --- /dev/null +++ b/internal/feed/v3impact_test.go @@ -0,0 +1,79 @@ +package feed + +import ( + "encoding/json" + "testing" +) + +func TestV3ImpactUnmarshalInvalidData(t *testing.T) { + test := []byte(`{}`) + var val V3Impact + + if err := json.Unmarshal(test, &val); err == nil { + t.Errorf("got \"%s\", exp error", val) + } +} + +func TestV3ImpactUnmarshalUnknown(t *testing.T) { + test := []byte(`"foo"`) + exp := "unknown CVSS v3 impact: foo" + var val V3Impact + + 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 TestV3ImpactUnmarshalValid(t *testing.T) { + tests := []struct { + val string + exp V3Impact + } { + { "\"NONE\"", V3ImpactNone }, + { "\"LOW\"", V3ImpactLow }, + { "\"HIGH\"", V3ImpactHigh }, + } + + for _, test := range(tests) { + t.Run(test.val, func(t *testing.T) { + var got V3Impact + 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 TestV3ImpactString(t *testing.T) { + tests := []struct { + val V3Impact + exp string + } { + { V3ImpactNone, "NONE" }, + { V3ImpactLow, "LOW" }, + { V3ImpactHigh, "HIGH" }, + + { V3Impact(255), "V3Impact(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) + } + }) + } +} |