aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--dbstore/dbstore_test.go85
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)