1 files changed, 44 insertions, 0 deletions
diff --git a/dbstore/util_test.go b/dbstore/util_test.go
new file mode 100644
index 0000000..a949878
--- /dev/null
+++ b/dbstore/util_test.go
@@ -0,0 +1,44 @@
+package dbstore
+
+import (
+  "github.com/pablotron/cvez/feed"
+  "testing"
+)
+
+func TestCheckNvdData(t *testing.T) {
+  passTests := []struct {
+    name string
+    dt feed.DataType
+    df feed.DataFormat
+    dv feed.DataVersion
+  } {
+    { "valid", feed.CveType, feed.MitreFormat, feed.V40 },
+  }
+
+  for _, test := range(passTests) {
+    t.Run(test.name, func(t *testing.T) {
+      if err := checkNvdData(test.dt, test.df, test.dv); err != nil {
+        t.Error(err)
+      }
+    })
+  }
+
+  failTests := []struct {
+    name string
+    dt feed.DataType
+    df feed.DataFormat
+    dv feed.DataVersion
+  } {
+    { "bad type", feed.DataType(255), feed.MitreFormat, feed.V40 },
+    { "bad format", feed.CveType, feed.DataFormat(255), feed.V40 },
+    { "bad version", feed.CveType, feed.MitreFormat, feed.DataVersion(255) },
+  }
+
+  for _, test := range(failTests) {
+    t.Run(test.name, func(t *testing.T) {
+      if checkNvdData(test.dt, test.df, test.dv) == nil {
+        t.Errorf("got success, exp error")
+      }
+    })
+  }
+}