stig tests option to lockdown permissions

src/freedombone-tests

@@ -32,6 +32,7 @@ PROJECT_NAME='freedombone'
 
 export TEXTDOMAIN=${PROJECT_NAME}-tests
 export TEXTDOMAINDIR="/usr/share/locale"
+source /usr/share/${PROJECT_NAME}/utils/${PROJECT_NAME}-utils-setup
 
 # Whether to run STIG tests
 RUN_STIG=
@@ -45,7 +46,7 @@ function show_help {
     echo ''
     echo $'Runs tests on the system'
     echo ''
-    echo $'     --stig [yes|no]      Run STIG tests'
+    echo $'     --stig [yes|no|fix]  Run STIG tests'
     echo $'     --help               Show help'
     echo ''
     exit 0
@@ -130,6 +131,13 @@ function stig_spinner {
     wait $1
 }
 
+function fix_stig {
+    if [[ $RUN_STIG != 'fix' ]]; then
+        return
+    fi
+    lockdown_permissions
+}
+
 function test_stig {
     if [ ! $RUN_STIG ]; then
         return
@@ -1125,7 +1133,7 @@ do
             if [[ "$1" == 'showall' ]]; then
                 SHOW_ALL_TESTS=1
             fi
-            RUN_STIG=1
+            RUN_STIG="$1"
             ;;
         *)
             # unknown option
@@ -1140,6 +1148,7 @@ fi
 
 test_app_functions
 test_unique_onion_ports
+fix_stig
 test_stig
 
 if [ ! $RUN_STIG ]; then