Update test case and fix exit in install script

ra1nb0rn · Jul 8, 2024 · 374fed0 · 374fed0
1 parent 8e68865
commit 374fed0
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 3 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,6 +1,12 @@
 # Changelog
 This file keeps track of all notable changes between the different versions of search_vulns.
 
+## v0.5.6 - 2024-07-08
+### Fixed
+- Update test case.
+- Fix install script to terminate with error code if DB build fails.
+
+
 ## v0.5.5 - 2024-06-28
 ### Changed
 - Exclude endoflife.date tests from workflow that updates the release assets.

diff --git a/install.sh b/install.sh
@@ -106,7 +106,7 @@ create_vuln_and_software_db() {
 
     if [ $? != 0 ]; then
         echo -e "${RED}Could not create vulnerability database"
-        return 1
+        exit 1
     fi
 }
 

diff --git a/tests/test_cve_completeness.py b/tests/test_cve_completeness.py
@@ -126,7 +126,7 @@ def test_search_openstack_glance(self):
         self.maxDiff = None
         query = 'cpe:2.3:a:openstack:glance:*:*:*:*:*:*:*:*'
         result = search_vulns.search_vulns(query=query, add_other_exploit_refs=True, is_good_cpe=False)
-        expected_cves = ['CVE-2022-47951', 'CVE-2013-1840', 'CVE-2015-8234', 'CVE-2015-3289', 'CVE-2015-5163', 'CVE-2015-5162', 'CVE-2013-4428', 'CVE-2016-8611', 'CVE-2017-7200', 'CVE-2022-4134', 'CVE-2015-5251', 'CVE-2014-0162', 'CVE-2014-5356', 'CVE-2016-0757', 'CVE-2015-5286', 'CVE-2014-9623', 'CVE-2015-1195', 'CVE-2014-1948', 'CVE-2012-5482', 'CVE-2013-0212', 'CVE-2013-4354', 'CVE-2015-1881', 'CVE-2014-9493', 'CVE-2014-9684', 'CVE-2012-4573']
+        expected_cves = ['CVE-2022-47951', 'CVE-2013-1840', 'CVE-2015-8234', 'CVE-2015-3289', 'CVE-2015-5163', 'CVE-2015-5162', 'CVE-2013-4428', 'CVE-2016-8611', 'CVE-2017-7200', 'CVE-2022-4134', 'CVE-2015-5251', 'CVE-2014-0162', 'CVE-2014-5356', 'CVE-2016-0757', 'CVE-2015-5286', 'CVE-2014-9623', 'CVE-2015-1195', 'CVE-2014-1948', 'CVE-2012-5482', 'CVE-2013-0212', 'CVE-2013-4354', 'CVE-2015-1881', 'CVE-2014-9493', 'CVE-2014-9684', 'CVE-2012-4573', 'CVE-2024-32498']
         self.assertEqual(set(expected_cves), set(list(result[query]['vulns'].keys())))
 
     def test_search_simplednsplus(self):

diff --git a/version.txt b/version.txt
@@ -1 +1 @@
-0.5.5
+0.5.6
-Original file line number
+Diff line change
@@ Expand Up / @@ -106,7 +106,7 @@ create_vuln_and_software_db() { @@
         if [ $? != 0 ]; then
             echo -e "${RED}Could not create vulnerability database"
-            return 1
+            exit 1
         fi
     }
@@ Expand Down @@