From 2632e8a18ce586fa0d40ac71d4b23e8cfb6d508b Mon Sep 17 00:00:00 2001
From: Paul Duncan <pabs@pablotron.org>
Date: Tue, 1 Feb 2022 08:31:50 -0500
Subject: internal/feed: add v3privilegesrequired tests

---
 internal/feed/v3privilegesrequired.go | 45 +++++++++++++++++++++++++++++++++++
 1 file changed, 45 insertions(+)
 create mode 100644 internal/feed/v3privilegesrequired.go

(limited to 'internal/feed/v3privilegesrequired.go')

diff --git a/internal/feed/v3privilegesrequired.go b/internal/feed/v3privilegesrequired.go
new file mode 100644
index 0000000..3e69334
--- /dev/null
+++ b/internal/feed/v3privilegesrequired.go
@@ -0,0 +1,45 @@
+package feed
+
+//go:generate stringer -linecomment -type=V3PrivilegesRequired
+
+import (
+  "encoding/json"
+  "fmt"
+)
+
+// CVSS v3 privileges required.
+type V3PrivilegesRequired byte
+
+const (
+  V3PRNone V3PrivilegesRequired = iota // NONE
+  V3PRLow // LOW
+  V3PRMedium // MEDIUM
+  V3PRHigh // HIGH
+)
+
+// Unmarshal CVSS privileges required from JSON.
+func (me *V3PrivilegesRequired) UnmarshalJSON(b []byte) error {
+  // decode string, check for error
+  var s string
+  if err := json.Unmarshal(b, &s); err != nil {
+    return err
+  }
+
+  // check value
+  switch s {
+  case "NONE":
+    *me = V3PRNone
+  case "LOW":
+    *me = V3PRLow
+  case "MEDIUM":
+    *me = V3PRMedium
+  case "HIGH":
+    *me = V3PRHigh
+  default:
+    // return error
+    return fmt.Errorf("unknown CVSS v3 privileges required: %s", s)
+  }
+
+  // return success
+  return nil
+}
-- 
cgit v1.2.3