cli: exit with status 2 when interrupted by SIGINT

[babeltrace.git] / tests / cli / test_exit_status

diff --git a/tests/cli/test_exit_status b/tests/cli/test_exit_status
new file mode 100755 (executable)
index 0000000..3e6f502
--- /dev/null
+++ b/tests/cli/test_exit_status
@@ -0,0 +1,93 @@
+#!/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=$(mktemp -t test_cli_exit_status_stdout_actual.XXXXXX)
+       local 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=$(mktemp -t test_cli_exit_status_stdout_actual.XXXXXX)
+       local 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=$(mktemp -t test_cli_exit_status_stdout_actual.XXXXXX)
+       local 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