internal/feed: split out datatype, add tests

1 files changed, 2 insertions, 27 deletions

diff --git a/internal/feed/feed.go b/internal/feed/feed.go
index 1c15a0a..b4d514b 100644
--- a/internal/feed/feed.go
+++ b/internal/feed/feed.go
@@ -9,8 +9,7 @@ import (
 )
 
 const (
-  CveType = iota  // CVE data type
-  MitreFormat     // MITRE data format
+  MitreFormat = iota    // MITRE data format
   DataVersion40   // Version 4.0
 
   OrNodeOp        // OR operator
@@ -42,31 +41,7 @@ const (
   Single          // Single authentication
 )
 
-// TODO: parse cpe, cvss vectors (v3.x, v2)
-
-// Data type for NVD feeds and feed items.
-type DataType int
-
-// Unmarshal DataType from JSON.
-func (me *DataType) 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 "CVE":
-    *me = CveType
-  default:
-    // return error
-    return fmt.Errorf("unknown data type: %s", s)
-  }
-
-  // return success
-  return nil
-}
+// TODO: parse cpe
 
 // Data format for NVD feeds and feed items.
 type DataFormat int