dbstore/dbstore_test.go: add TestAddCisaCatalogs() and stub TestCisaSearch()

1 files changed, 85 insertions, 0 deletions

diff --git a/dbstore/dbstore_test.go b/dbstore/dbstore_test.go
index df7dd62..524fecb 100644
--- a/dbstore/dbstore_test.go
+++ b/dbstore/dbstore_test.go
@@ -10,6 +10,7 @@ import (
   "errors"
   "fmt"
   _ "github.com/mattn/go-sqlite3"
+  "github.com/pablotron/cvez/cisa"
   "github.com/pablotron/cvez/cpedict"
   "github.com/pablotron/cvez/cpematch"
   nvd_feed "github.com/pablotron/cvez/feed"
@@ -42,6 +43,27 @@ func getFeed(path string) (nvd_feed.Feed, error) {
   return feed, d.Decode(&feed)
 }
 
+// Load gzip-compressed test CISA catalog.
+func getCisaCatalog(path string) (cisa.Catalog, error) {
+  var cat cisa.Catalog
+
+  // open file for reading
+  file, err := os.Open(path)
+  if err != nil {
+    return cat, err
+  }
+
+  // wrap in reader, return success
+  src, err := gzip.NewReader(file)
+  if err != nil {
+    return cat, err
+  }
+
+  // create decoder, decode catalog, return result
+  d := json.NewDecoder(src)
+  return cat, d.Decode(&cat)
+}
+
 func getTestDictionary(path string) (cpedict.Dictionary, error) {
   var dict cpedict.Dictionary
 
@@ -1376,3 +1398,66 @@ func TestCveSearch(t *testing.T) {
     }
   })
 }
+
+func TestAddCisaCatalogs(t *testing.T) {
+  ctx := context.Background()
+
+  tests := []struct {
+    name  string // test name
+    files []string // test files
+    fast  bool // is this test fast?
+  } {{
+    name:   "fast",
+    files:  []string { "cisa-kevc-20220313-tiny", },
+    fast:   true,
+  }, {
+    name:   "slow",
+    files:  []string { "cisa-kevc-20220313-tiny", "cisa-kevc-20220313" },
+    fast:   false,
+  }}
+
+  for _, test := range(tests) {
+    t.Run(test.name, func(t *testing.T) {
+      // skip slow tests in short mode
+      if !test.fast && testing.Short() {
+        t.Skip("skipping in short mode")
+        return
+      }
+
+      // build db path
+      dbPath := fmt.Sprintf("testdata/test-addcisacatalogs-%s.db", test.name)
+
+      // create test db
+      db, err := createTestDb(ctx, dbPath)
+      if err != nil {
+        t.Error(err)
+        return
+      }
+
+      // load test catalogs
+      cats := make([]cisa.Catalog, len(test.files))
+      for i, name := range(test.files) {
+        // build catalog path
+        path := fmt.Sprintf("testdata/%s.json.gz", name)
+
+        // get catalog
+        if cat, err := getCisaCatalog(path); err != nil {
+          t.Error(err)
+          return
+        } else {
+          cats[i] = cat
+        }
+      }
+
+      // add catalogs
+      if _, err = db.AddCisaCatalogs(ctx, cats); err != nil {
+        t.Error(err)
+        return
+      }
+    })
+  }
+}
+
+// TODO: TestCisaSearch()
+// test db: dbstore/testdata/mock-testcisasearch.db
+// test cases: "wordpad" (1 result), "microsoft excel" (4 results)