btrfs-progs: tests: check if kernel has btrfs support

[platform/upstream/btrfs-progs.git] / tests / common

diff --git a/tests/common b/tests/common
index 67d1558..c20fec8 100644 (file)
--- a/tests/common
+++ b/tests/common
@@ -24,7 +24,12 @@ _not_run()
 run_check()
 {
        echo "############### $@" >> $RESULTS 2>&1
-       "$@" >> $RESULTS 2>&1 || _fail "failed: $@"
+       if [ "$TEST_LOG" = 'tty' ]; then echo "CMD: $@" > /dev/tty; fi
+       if [ "$1" = 'root_helper' ]; then
+               "$@" >> $RESULTS 2>&1 || _fail "failed: $@"
+       else
+               $INSTRUMENT "$@" >> $RESULTS 2>&1 || _fail "failed: $@"
+       fi
 }
 
 # same as run_check but the stderr+stdout output is duplicated on stdout and
@@ -32,20 +37,76 @@ run_check()
 run_check_stdout()
 {
        echo "############### $@" >> $RESULTS 2>&1
-       "$@" 2>&1 | tee -a $RESULTS || _fail "failed: $@"
+       if [ "$TEST_LOG" = 'tty' ]; then echo "CMD(stdout): $@" > /dev/tty; fi
+       if [ "$1" = 'root_helper' ]; then
+               "$@" 2>&1 | tee -a $RESULTS || _fail "failed: $@"
+       else
+               $INSTRUMENT "$@" 2>&1 | tee -a $RESULTS || _fail "failed: $@"
+       fi
 }
 
 # same as run_check but does not fail the test, output is logged
 run_mayfail()
 {
+       local ret
+
        echo "############### $@" >> $RESULTS 2>&1
-       "$@" >> $RESULTS 2>&1 || _log "failed (ignored): $@"
+       if [ "$TEST_LOG" = 'tty' ]; then echo "CMD(mayfail): $@" > /dev/tty; fi
+       if [ "$1" = 'root_helper' ]; then
+               "$@" >> $RESULTS 2>&1
+       else
+               $INSTRUMENT "$@" >> $RESULTS 2>&1
+       fi
+       ret=$?
+       if [ $ret != 0 ]; then
+               echo "failed (ignored, ret=$ret): $@" >> $RESULTS
+               if [ $ret == 139 ]; then
+                       _fail "mayfail: returned code 139 (SEGFAULT), not ignored"
+               elif [ $ret == 134 ]; then
+                       _fail "mayfail: returned code 134 (SIGABRT), not ignored"
+               fi
+               return $ret
+       fi
+}
+
+# first argument is error message to print if it fails, otherwise
+# same as run_check but expects the command to fail, output is logged
+run_mustfail()
+{
+       local msg
+
+       msg="$1"
+       shift
+
+       echo "############### $@" >> $RESULTS 2>&1
+       if [ "$TEST_LOG" = 'tty' ]; then echo "CMD(mustfail): $@" > /dev/tty; fi
+       if [ "$1" = 'root_helper' ]; then
+               "$@" >> $RESULTS 2>&1
+       else
+               $INSTRUMENT "$@" >> $RESULTS 2>&1
+       fi
+       if [ $? != 0 ]; then
+               echo "failed (expected): $@" >> $RESULTS
+               return 0
+       else
+               echo "succeeded (unexpected!): $@" >> $RESULTS
+               _fail "unexpected success: $msg"
+               return 1
+       fi
 }
 
 check_prereq()
 {
        if ! [ -f $TOP/$1 ]; then
-               _fail "Failed prerequisities: $1";
+               _fail "Failed prerequisites: $1";
+       fi
+}
+
+check_global_prereq()
+{
+       which $1 &> /dev/null
+       if [ $? -ne 0 ]; then
+               _fail "Failed system wide prerequisities: $1";
        fi
 }
 
@@ -67,6 +128,7 @@ check_image()
 # - dtto compressed by XZ, suffix .raw.xz
 # - meta-dump images with suffix .img
 # - dtto compressed by XZ, suffix .img.xz
+# - compressed send stream, .stream.xz
 extract_image()
 {
        local image
@@ -93,12 +155,19 @@ extract_image()
                image=${image%%.xz}
                mv "$image" "$image".restored
                ;;
+       *.stream.xz)
+               xz --decompress --keep "$image" || \
+                       _fail "failed to decompress file $image" >&2
+               image=${image%%.xz}
+               mv "$image" "$image".restored
+               ;;
        esac
 
        if ! [ -f $image.restored ]; then
                echo "restoring image $(basename $image)" >> $RESULTS
-               $TOP/btrfs-image -r $image $image.restored || \
-                       _fail "failed to restore image $image" >&2
+               $TOP/btrfs-image -r $image $image.restored \
+                       &>> $RESULTS \
+                       || _fail "failed to restore image $image" >&2
        fi
 
        [ -f "$cleanme" ] && rm -f "$cleanme"
@@ -153,7 +222,7 @@ root_helper()
 
 setup_root_helper()
 {
-       if [ $UID -eq 0 ]; then
+       if [ $UID -eq 0 -o -n "$SUDO_HELPER" ]; then
                return
        fi
 
@@ -176,7 +245,7 @@ prepare_test_dev()
        local size="$1"
 
        [[ "$TEST_DEV" ]] && return
-       [[ "$size" ]] || size='1G'
+       [[ "$size" ]] || size='2G'
 
        echo "\$TEST_DEV not given, use $TOP/test/test.img as fallback" >> \
                $RESULTS
@@ -211,10 +280,20 @@ run_check_umount_test_dev()
        run_check $SUDO_HELPER umount "$@" "$TEST_DEV"
 }
 
+check_kernel_support()
+{
+       if ! grep -iq 'btrfs' /proc/filesystems; then
+               echo "WARNING: btrfs filesystem not listed in /proc/filesystems, some tests might fail"
+               return 1
+       fi
+       return 0
+}
+
 init_env()
 {
        TEST_MNT="${TEST_MNT:-$TOP/tests/mnt}"
        export TEST_MNT
        mkdir -p "$TEST_MNT" || { echo "Failed mkdir -p $TEST_MNT"; exit 1; }
+
 }
 init_env