+#!/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