From 90aca6641d224ed2f4e4d798d374eceb9d9a9eb3 Mon Sep 17 00:00:00 2001
From: Karol Latecki <karol.latecki@intel.com>
Date: Wed, 20 Dec 2017 08:54:03 +0100
Subject: [PATCH] test/vhost: add trap to lvol test

Change-Id: Ibd3e485085793260ca4bb2bfa3da6025b2c997aa
Signed-off-by: Karol Latecki <karol.latecki@intel.com>
Reviewed-on: https://review.gerrithub.io/392450
Reviewed-by: Tomasz Zawadzki <tomasz.zawadzki@intel.com>
Reviewed-by: Jim Harris <james.r.harris@intel.com>
Reviewed-by: Daniel Verkamp <daniel.verkamp@intel.com>
Reviewed-by: Ben Walker <benjamin.walker@intel.com>
Tested-by: SPDK Automated Test System <sys_sgsw@intel.com>
---
 test/vhost/lvol/lvol_test.sh | 55 ++++++++++++++++++++----------------
 1 file changed, 31 insertions(+), 24 deletions(-)

diff --git a/test/vhost/lvol/lvol_test.sh b/test/vhost/lvol/lvol_test.sh
index e09adf673..e9bc480e2 100755
--- a/test/vhost/lvol/lvol_test.sh
+++ b/test/vhost/lvol/lvol_test.sh
@@ -42,6 +42,33 @@ function usage()
     exit 0
 }
 
+function clean_lvol_cfg()
+{
+    echo "INFO: Removing nested lvol bdevs"
+    for lvol_bdev in "${nest_lvol_bdevs[@]}"; do
+        $rpc_py delete_bdev $lvol_bdev
+        echo -e "\tINFO: nested lvol bdev $lvol_bdev removed"
+    done
+
+    echo "INFO: Removing nested lvol stores"
+    for lvol_store in "${nest_lvol_stores[@]}"; do
+        $rpc_py destroy_lvol_store -u $lvol_store
+        echo -e "\tINFO: nested lvol store $lvol_store removed"
+    done
+
+    echo "INFO: Removing lvol bdevs"
+    for lvol_bdev in "${lvol_bdevs[@]}"; do
+        $rpc_py delete_bdev $lvol_bdev
+        echo -e "\tINFO: lvol bdev $lvol_bdev removed"
+    done
+
+    echo "INFO: Removing lvol stores"
+    for lvol_store in "${lvol_stores[@]}"; do
+        $rpc_py destroy_lvol_store -u $lvol_store
+        echo -e "\tINFO: lvol store $lvol_store removed"
+    done
+}
+
 while getopts 'xh-:' optchar; do
     case "$optchar" in
         -)
@@ -81,7 +108,7 @@ if $distribute_cores; then
     source $BASE_DIR/autotest.config
 fi
 
-trap 'error_exit "${FUNCNAME}" "${LINENO}"' ERR
+trap 'error_exit "${FUNCNAME}" "${LINENO}"' SIGTERM SIGABRT ERR
 
 vm_kill_all
 
@@ -90,6 +117,8 @@ echo ""
 $COMMON_DIR/run_vhost.sh $x --work-dir=$TEST_DIR --conf-dir=$BASE_DIR
 echo ""
 
+trap 'clean_lvol_cfg; error_exit "${FUNCNAME}" "${LINENO}"' SIGTERM SIGABRT ERR
+
 lvol_stores=()
 lvol_bdevs=()
 nest_lvol_stores=()
@@ -240,29 +269,7 @@ elif [[ "$ctrl_type" == "vhost_blk" ]]; then
     done
 fi
 
-echo "INFO: Removing nested lvol bdevs"
-for lvol_bdev in "${nest_lvol_bdevs[@]}"; do
-    $rpc_py delete_bdev $lvol_bdev
-    echo -e "\tINFO: nested lvol bdev $lvol_bdev removed"
-done
-
-echo "INFO: Removing nested lvol stores"
-for lvol_store in "${nest_lvol_stores[@]}"; do
-    $rpc_py destroy_lvol_store -u $lvol_store
-    echo -e "\tINFO: nested lvol store $lvol_store removed"
-done
-
-echo "INFO: Removing lvol bdevs"
-for lvol_bdev in "${lvol_bdevs[@]}"; do
-    $rpc_py delete_bdev $lvol_bdev
-    echo -e "\tINFO: lvol bdev $lvol_bdev removed"
-done
-
-echo "INFO: Removing lvol stores"
-for lvol_store in "${lvol_stores[@]}"; do
-    $rpc_py destroy_lvol_store -u $lvol_store
-    echo -e "\tINFO: lvol store $lvol_store removed"
-done
+clean_lvol_cfg
 
 $rpc_py get_lvol_stores
 $rpc_py get_bdevs