#!/bin/bash

_kill_all_jobs() {
	local _jobs=$(jobs -r -p)

	[ -n "$_jobs" ] && kill $_jobs
}

trap '
ret=$?;
_kill_all_jobs
exit $ret;
' EXIT

trap 'exit 1;' SIGINT

BASEDIR=$(realpath $(dirname "$0"))
. $BASEDIR/test-lib.sh
TESTCASEDIR="$BASEDIR/testcases"

console=0
testcases=""

while [ $# -gt 0 ]; do
	case $1 in
		'')
			break
			;;
		--console )
			console=1
			;;
		-*)
			echo "Invalid option $1"
			;;
		*)
			testcases+=" $1"
			;;
	esac
	shift;
done

if [ -z "$testcases" ]; then
	echo "==== Starting all tests: ===="
	testcases=$(ls -1 $TESTCASEDIR)
else
	echo "==== Starting specified tests: ===="
fi
echo ${testcases##*/}
echo

declare -A results
ret=0

for test_case in $testcases; do
	echo "======== Running Test Case $test_case ========"
	results[$test_case]="<Test Skipped>"

	testdir=$TESTCASEDIR/$test_case
	script_num=$(ls -1 $testdir | wc -l)
	scripts=$(ls -r -1 $testdir | tr '\n' ' ')
	test_outputs=""
	read main_script aux_script <<< "$scripts"

	if [ -z "$main_script" ]; then
		echo "ERROR: Empty testcase dir $testdir"
		continue
	fi

	for script in $scripts; do
		echo "---- Building image for: $script ----"
		echo "-------- Output image is: $(get_test_image $testdir/$script)"
		echo "-------- Building log is: $(get_test_image $testdir/$script).log"

		mkdir -p $(dirname $(get_test_image $testdir/$script))
		build_test_image $testdir/$script &> $(get_test_image $testdir/$script).log

		if [ $? -ne 0 ]; then
			echo "Failing building image!"
			continue 2
		fi
	done

	for script in $aux_script; do
		echo "---- Starting VM: $script ----"

		script="$testdir/$script"
		echo "-------- Qemu cmdline: $(get_test_qemu_cmd_file $script)"
		echo "-------- Console log: $(get_test_console_file $script)"
		echo "-------- Test log: $(get_test_output_file $script)"
		test_outputs+="$(get_test_output_file $script) "

		rm -f $(get_test_console_file $script)
		rm -f $(get_test_output_file $script)

		$(run_test_sync $script > $(get_test_console_file $script)) &

		sleep 5
	done

	script="$main_script"
	echo "---- Starting test VM: $(basename $script) ----"
	script="$testdir/$script"

	echo "-------- Qemu cmdline: $(get_test_qemu_cmd_file $script)"
	echo "-------- Console log: $(get_test_console_file $script)"
	echo "-------- Test log: $(get_test_output_file $script)"
	test_outputs+="$(get_test_output_file $script) "

	rm -f $(get_test_console_file $script)
	rm -f $(get_test_output_file $script)

	if [ $console -eq 1 ]; then
		run_test_sync $script | tee $(get_test_console_file $script)
		[ -n "$(jobs -p)" ] && kill $(jobs -p)
	else
		$(run_test_sync $script > $(get_test_console_file $script)) &
		watch_test_outputs $test_outputs
	fi

	res="$(gather_test_result $test_outputs)"

	[ $? -ne 0 ] && ret=$(expr $ret + 1)
	results[$test_case]="$res"

	echo -e "-------- Test finished: $test_case $res --------"
	for script in $scripts; do
		script="$testdir/$script"
		output="$(get_test_output_file $script) "
		image="$(get_test_image $script)"
		vmcore="$(sed -n 's/^VMCORE: \(\S*\).*/\1/p' $output)"
		kernel="$(sed -n 's/^KERNEL VERSION: \(\S*\).*/\1/p' $output)"
		if [ -n "$vmcore" ]; then
			echo "You can retrive the verify the vmcore file using following command:"
			echo "./scripts/copy-from-image.sh \\"
			echo "    $image \\"
			echo "    $vmcore ./"
			echo "Kernel package verion is: $kernel"
		fi
	done
done

echo "======== Test results ========"
for i in ${!results[@]}; do
	echo "----------------"
	echo -e "$i:\t\t${results[$i]}"
done

exit $ret