#!/bin/bash

topdir="${CURTIN_VMTEST_TOPDIR:-${WORKSPACE:-$PWD}/output}"
pkeep=${CURTIN_VMTEST_KEEP_DATA_PASS:-logs,collect}
fkeep=${CURTIN_VMTEST_KEEP_DATA_FAIL:-logs,collect}
reuse=${CURTIN_VMTEST_REUSE_TOPDIR:-0}
export CURTIN_VMTEST_REUSE_TOPDIR=$reuse
export CURTIN_VMTEST_IMAGE_SYNC=${CURTIN_VMTEST_IMAGE_SYNC:-0}
export CURTIN_VMTEST_KEEP_DATA_PASS=$pkeep
export CURTIN_VMTEST_KEEP_DATA_FAIL=$fkeep
export CURTIN_VMTEST_TOPDIR="$topdir"
export CURTIN_VMTEST_LOG="${CURTIN_VMTEST_LOG:-$topdir/debug.log}"
export CURTIN_VMTEST_PARALLEL=${CURTIN_VMTEST_PARALLEL:-0}
export IMAGE_DIR=${IMAGE_DIR:-/srv/images}

fail() { echo "$@" 1>&2; exit 1; }

if [ "$reuse" != "1" ]; then
    if [ -d "$topdir" ]; then
        fail "topdir '$topdir' existed."
    fi
    mkdir -p "$topdir" || fail "failed mkdir $topdir"
fi

start_s=$(date +%s)
parallel=${CURTIN_VMTEST_PARALLEL}
ntargs=( )
while [ $# -ne 0 ]; do
    case "$1" in
        -p|--parallel) parallel="$2"; shift;;
        --parallel=*) parallel=${1#*=};;
        -p[0-9]|-p-1|-p[0-9][0-9]) parallel=${1#-p};;
        --)
            shift
            break
            ;;
        *) ntargs[${#ntargs[@]}]="$1";;
    esac
    shift;
done

CURTIN_VMTEST_PARALLEL=$parallel

if [ ${#ntargs[@]} -eq 0 ]; then
   set -- -vv --nologcapture tests/vmtests/
fi

# dump CURTIN* variables just for info
for v in ${!CURTIN_*}; do
   echo "$v=${!v}"
done

ntargs=( "${ntargs[@]}" "$@" )

pargs=( )
if [ -n "$parallel" -a "$parallel" != "0" -a "$parallel" != "1" ]; then
    pargs=( --process-timeout=86400 "--processes=$parallel" )
fi

if [ -z "$TGT_IPC_SOCKET" ] && command -v "tgtd" >/dev/null 2>&1; then
    tgtdir="$topdir/tgt.d"
    ./tools/find-tgt "$tgtdir" ||
        fail "could not start a tgt service"
    . "$tgtdir/info"
fi

# avoid LOG info by running python3 tests/vmtests/image_sync.py
# rather than python3 -m tests.vmtests.image_sync (LP: #1594465)
echo "Working with images in $IMAGE_DIR"
fmt="  %(release)-7s %(arch)s/%(subarch)s %(version_name)-10s"
PYTHONPATH="$PWD" python3 tests/vmtests/image_sync.py query \
    --output-format="$fmt" "$IMAGE_DIR" ftype=root-image.gz ||
    { ret=$?; echo "FATAL: error querying images in $IMAGE_DIR" 1>&2;
      exit $ret; }

echo "$(date -R): vmtest start: nosetests3 ${pargs[*]} ${ntargs[*]}"
nosetests3 "${pargs[@]}" "${ntargs[@]}"
ret=$?
end_s=$(date +%s)
echo "$(date -R): vmtest end [$ret] in $(($end_s-$start_s))s"
[ $ret -eq 0 ] && [[ $pkeep == *"all"* ]] || [ -n "${TGT_PID}" ] && kill -9 ${TGT_PID}
[ $ret -ne 0 ] && [[ $pfail == *"all"* ]] || [ -n "${TGT_PID}" ] && kill -9 ${TGT_PID}
exit $ret

# vi: ts=4 expandtab syntax=sh
