dbstore: update tests to exclude excrutiating ones in short test mode

1 files changed, 20 insertions, 0 deletions

diff --git a/dbstore/dbstore_test.go b/dbstore/dbstore_test.go
index 69327d9..6394035 100644
--- a/dbstore/dbstore_test.go
+++ b/dbstore/dbstore_test.go
@@ -306,6 +306,11 @@ func TestGetQueries(t *testing.T) {
 }
 
 func TestAddCpeDictionaryPass(t *testing.T) {
+  if testing.Short() {
+    t.Skip("skipping TestAddCveFeeds() in short mode")
+    return
+  }
+
   path := "./testdata/test-addcpedict.db"
   ctx := context.Background()
 
@@ -331,6 +336,11 @@ func TestAddCpeDictionaryPass(t *testing.T) {
 }
 
 func TestAddCpeDictionaryFail(t *testing.T) {
+  if testing.Short() {
+    t.Skip("skipping TestAddCveFeeds() in short mode")
+    return
+  }
+
   // load test CPEs
   dict, err := getTestDictionary("testdata/test-0.xml.gz")
   if err != nil {
@@ -553,6 +563,11 @@ func TestCpeSearch(t *testing.T) {
 }
 
 func TestAddCpeMatches(t *testing.T) {
+  if testing.Short() {
+    t.Skip("skipping TestAddCveFeeds() in short mode")
+    return
+  }
+
   // cache context, create temp dir
   ctx := context.Background()
   dir, err := os.MkdirTemp("", "")
@@ -973,6 +988,11 @@ func TestCpeMatchSearch(t *testing.T) {
 func TestAddCveFeeds(t *testing.T) {
   ctx := context.Background()
 
+  if testing.Short() {
+    t.Skip("skipping TestAddCveFeeds() in short mode")
+    return
+  }
+
   tests := []string {
     "nvdcve-1.1-2002",
     "nvdcve-1.1-2003",