#!/bin/bash
#
# Copyright (C) 2019 EfficiOS Inc.
#
# This program is free software; you can redistribute it and/or modify it
# under the terms of the GNU General Public License, version 2 only, as
# published by the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
# more details.
#
# You should have received a copy of the GNU General Public License along with
# this program; if not, write to the Free Software Foundation, Inc., 51
# Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

SH_TAP=1

if [ "x${BT_TESTS_SRCDIR:-}" != "x" ]; then
	UTILSSH="$BT_TESTS_SRCDIR/utils/utils.sh"
else
	UTILSSH="$(dirname "$0")/../utils/utils.sh"
fi

# shellcheck source=../utils/utils.sh
source "$UTILSSH"

data_dir="$BT_TESTS_DATADIR/cli/exit_status"
source_name="src.test_exit_status.StatusSrc"

test_interrupted_graph() {
	local cli_args=("--plugin-path=$data_dir" "-c" "$source_name" "-p" "case=\"INTERRUPTED\"")
	local actual_stdout
	local actual_stderr

	actual_stdout=$(mktemp -t test_cli_exit_status_stdout_actual.XXXXXX)
	actual_stderr=$(mktemp -t test_cli_exit_status_stderr_actual.XXXXXX)

	bt_cli "$actual_stdout" "$actual_stderr" "${cli_args[@]}"

	is $? 2 "Interrupted graph exits with status 2"

	bt_diff /dev/null "$actual_stdout"
	ok $? "Interrupted graph gives no stdout"

	bt_diff /dev/null "$actual_stderr"
	ok $? "Interrupted graph gives no stderr"

	rm -f "${actual_stdout}"
	rm -f "${actual_stderr}"
}

test_error_graph() {
	local cli_args=("--plugin-path=$data_dir" "-c" "$source_name" "-p" "case=\"ERROR\"")
	local actual_stdout
	local actual_stderr

	actual_stdout=$(mktemp -t test_cli_exit_status_stdout_actual.XXXXXX)
	actual_stderr=$(mktemp -t test_cli_exit_status_stderr_actual.XXXXXX)

	bt_cli "$actual_stdout" "$actual_stderr" "${cli_args[@]}"

	is $? 1 "Erroring graph exits with status 1"

	bt_diff /dev/null "$actual_stdout"
	ok $? "Erroring graph gives expected stdout"

	like "$(cat "${actual_stderr}")" "TypeError: Raising type error" \
		"Erroring graph gives expected error message"

	rm -f "${actual_stdout}"
	rm -f "${actual_stderr}"
}

test_stop_graph() {
	local cli_args=("--plugin-path=$data_dir" "-c" "$source_name" "-p" "case=\"STOP\"")
	local actual_stdout
	local actual_stderr

	actual_stdout=$(mktemp -t test_cli_exit_status_stdout_actual.XXXXXX)
	actual_stderr=$(mktemp -t test_cli_exit_status_stderr_actual.XXXXXX)

	bt_cli "$actual_stdout" "$actual_stderr" "${cli_args[@]}"

	is $? 0 "Successful graph exits with status 0"

	bt_diff /dev/null "$actual_stdout"
	ok $? "Successful graph gives no stdout"

	bt_diff /dev/null "$actual_stderr"
	ok $? "Successful graph gives no stderr"

	rm -f "${actual_stdout}"
	rm -f "${actual_stderr}"
}

plan_tests 9

test_interrupted_graph
test_error_graph
test_stop_graph