internal/feed: add v3attackcomplexity tests

1 files changed, 1 insertions, 29 deletions

diff --git a/internal/feed/feed.go b/internal/feed/feed.go
index a04c3ce..38a7767 100644
--- a/internal/feed/feed.go
+++ b/internal/feed/feed.go
@@ -29,34 +29,6 @@ const (
 
 // TODO: parse cpe
 
-// CVSS attack complexity
-type AttackComplexity int
-
-// Unmarshal CVSS attack complexity from JSON.
-func (me *AttackComplexity) 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 "LOW":
-    *me = Low
-  case "MEDIUM":
-    *me = Medium
-  case "HIGH":
-    *me = High
-  default:
-    // return error
-    return fmt.Errorf("unknown attack complexity: %s", s)
-  }
-
-  // return success
-  return nil
-}
-
 // CVSS privileges required
 type PrivilegesRequired int
 
@@ -484,7 +456,7 @@ type CvssV3 struct {
   AttackVector V3AttackVector `json:"attackVector"`
 
   // attack complexity
-  AttackComplexity AttackComplexity `json:"attackComplexity"`
+  AttackComplexity V3AttackComplexity `json:"attackComplexity"`
 
   // privileges required
   PrivilegesRequired PrivilegesRequired `json:"privilegesRequired"`