aboutsummaryrefslogtreecommitdiff
path: root/internal/feed/v3impact_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'internal/feed/v3impact_test.go')
-rw-r--r--internal/feed/v3impact_test.go79
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)
+ }
+ })
+ }
+}