From 30e3efe2cb2a3532b79a1cc1863c268352f0462e Mon Sep 17 00:00:00 2001
From: pemontto <pemontto@gmail.com>
Date: Wed, 17 Apr 2019 10:12:29 +1000
Subject: [PATCH] set default path and fix restore

---
 tests/test-vuln_whisperer.sh | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/tests/test-vuln_whisperer.sh b/tests/test-vuln_whisperer.sh
index 050a56b..7739e8b 100755
--- a/tests/test-vuln_whisperer.sh
+++ b/tests/test-vuln_whisperer.sh
@@ -19,6 +19,8 @@ function yellow() {
 
 return_code=0
 
+TEST_PATH=${TEST_PATH:-"tests/data"}
+
 yellow "\n*********************************************"
 yellow "* Test successful scan download and parsing *"
 yellow "*********************************************"
@@ -89,7 +91,7 @@ else
 fi
 
 # Restore the removed files
-mv" ${TEST_PATH}/qualys_vuln/scan_1553941061.87241.bak" "${TEST_PATH}/qualys_vuln/scan_1553941061.87241"
-mv "${TEST_PATH}/nessus/GET_scans_exports_164_download.bak" "${TEST_PATH}/nessus/GET_scans_exports_164_download.bak"
+mv "${TEST_PATH}/qualys_vuln/scan_1553941061.87241.bak" "${TEST_PATH}/qualys_vuln/scan_1553941061.87241"
+mv "${TEST_PATH}/nessus/GET_scans_exports_164_download.bak" "${TEST_PATH}/nessus/GET_scans_exports_164_download"
 
-exit $return_code
\ No newline at end of file
+exit $return_code