X-Git-Url: https://git.lttng.org/?a=blobdiff_plain;f=tests%2Fregression%2Ftools%2Fsnapshots%2Ftest_ust_streaming;h=2e04d667e76890a9dbf7141fef82b429d4f9dbad;hb=c1896e2d809e0d32d2d9b8a840755de6d5d70eaf;hp=38ffba09c71f448e2b13820ed364b724bad78948;hpb=03f66fb8271904c7bca9d3cfd5b3cf1d9c4e5221;p=lttng-tools.git

diff --git a/tests/regression/tools/snapshots/test_ust_streaming b/tests/regression/tools/snapshots/test_ust_streaming
index 38ffba09c..2e04d667e 100755
--- a/tests/regression/tools/snapshots/test_ust_streaming
+++ b/tests/regression/tools/snapshots/test_ust_streaming
@@ -2,38 +2,28 @@
 #
 # Copyright (C) - 2013 David Goulet <dgoulet@efficios.com>
 #
-# This library is free software; you can redistribute it and/or modify it under
-# the terms of the GNU Lesser General Public License as published by the Free
-# Software Foundation; version 2.1 of the License.
-#
-# This library 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 Lesser General Public License for more
-# details.
-#
-# You should have received a copy of the GNU Lesser General Public License
-# along with this library; if not, write to the Free Software Foundation, Inc.,
-# 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301 USA
+# SPDX-License-Identifier: LGPL-2.1-only
+
 TEST_DESC="Streaming - Snapshot UST tracing"
 
-CURDIR=$(dirname $0)/
-TESTDIR=$CURDIR/../../..
+CURDIR=$(dirname "$0")/
+TESTDIR="$CURDIR/../../.."
 EVENT_NAME="tp:tptest"
 SESSION_NAME=""
 CHANNEL_NAME="chan1"
-BIN_NAME="gen-nevents"
 TESTAPP_PATH="$TESTDIR/utils/testapp"
 TESTAPP_NAME="gen-ust-events"
 TESTAPP_BIN="$TESTAPP_PATH/$TESTAPP_NAME/$TESTAPP_NAME"
-NR_ITER=2000000
+NR_ITER=-1
 NR_USEC_WAIT=100
-APPS_PID=
+APP_PIDS=()
 
-TRACE_PATH=$(mktemp -d)
+TRACE_PATH=$(mktemp -d -t tmp.test_snapshots_ust_streaming_trace_path.XXXXXX)
 
-NUM_TESTS=49
+NUM_TESTS=85
 
-source $TESTDIR/utils/utils.sh
+# shellcheck source-path=SCRIPTDIR/../../../
+source "$TESTDIR/utils/utils.sh"
 
 if [ ! -x "$TESTAPP_BIN" ]; then
 	BAIL_OUT "No UST events binary detected."
@@ -44,25 +34,42 @@ function snapshot_add_output ()
 	local sess_name=$1
 	local trace_path=$2
 	local name=$3
-	local extra_opt=""
+	local options=()
+	local args=()
+
+	if [ -n "$sess_name" ] ; then
+		options+=(-s "$sess_name")
+	fi
 
-	if [ ! -z $name ]; then
-		extra_opt="-n $name"
+	if [ -n "$name" ]; then
+		options+=(-n "$name")
 	fi
 
-	$TESTDIR/../src/bin/lttng/$LTTNG_BIN snapshot add-output -s $sess_name $extra_opt $trace_path >/dev/null 2>&1
+	# Checking for any other optional arguments that should be passed to the lttng command
+	args=("$@")
+	if [ -n "${args[*]:3}" ] ; then
+		options+=("${args[@]:3}")
+	fi
+
+	# The trace_path is a positional argument and must always come last if it is defined
+	if [ -n "$trace_path" ] ; then
+		options+=("$trace_path")
+	fi
+
+	"$TESTDIR/../src/bin/lttng/$LTTNG_BIN" snapshot add-output "${options[@]}" >/dev/null 2>&1
 	ok $? "Added snapshot output $trace_path"
 }
 
 # Start trace application and return once one event has been hit.
 function start_test_app()
 {
-	local tmp_file=$(mktemp -u)
+	local tmp_file
+	tmp_file=$(mktemp -u -t tmp.test_"${FUNCNAME[0]}"_tmp_file.XXXXXX)
 
 	# Start application with a temporary file.
-	$TESTAPP_BIN $NR_ITER $NR_USEC_WAIT $tmp_file &
+	$TESTAPP_BIN -i $NR_ITER -w $NR_USEC_WAIT --sync-after-first-event "$tmp_file" &
 	ret=$?
-	APPS_PID="${APPS_PID} ${!}"
+	APP_PIDS+=(${!})
 	ok $ret "Start application to trace"
 
 	# Wait for the application file to appear indicating that at least one
@@ -71,54 +78,51 @@ function start_test_app()
 		sleep 0.5
 	done
 	diag "Removing test app temporary file $tmp_file"
-	rm -f $tmp_file
+	rm -f "$tmp_file"
 }
 
 function stop_test_apps()
 {
 	diag "Stopping $TESTAPP_NAME"
-	for p in ${APPS_PID}; do
-		kill ${p}
-		wait ${p} 2>/dev/null
-	done
-	APPS_PID=
+	kill "${APP_PIDS[@]}"
+	wait "${APP_PIDS[@]}" 2>/dev/null
+	APP_PIDS=()
 }
 
 # Test a snapshot using a default name for the output destination.
 function test_ust_default_name_with_del()
 {
 	diag "Test UST snapshot streaming with default name with delete output"
-	create_lttng_session_no_output $SESSION_NAME
-	enable_lttng_mmap_overwrite_ust_channel $SESSION_NAME $CHANNEL_NAME
-	enable_ust_lttng_event_ok $SESSION_NAME $EVENT_NAME $CHANNEL_NAME
-	start_lttng_tracing_ok $SESSION_NAME
+	create_lttng_session_no_output "$SESSION_NAME"
+	enable_lttng_mmap_overwrite_ust_channel "$SESSION_NAME" $CHANNEL_NAME
+	enable_ust_lttng_event_ok "$SESSION_NAME" $EVENT_NAME $CHANNEL_NAME
+	start_lttng_tracing_ok "$SESSION_NAME"
 
 	start_test_app
 
-	snapshot_add_output $SESSION_NAME "net://localhost"
-	lttng_snapshot_record $SESSION_NAME
+	snapshot_add_output "$SESSION_NAME" "net://localhost"
+	lttng_snapshot_record "$SESSION_NAME"
 
 	# Validate test
-	echo $TRACE_PATH/$HOSTNAME/snapshot-1
-	validate_trace $EVENT_NAME $TRACE_PATH/$HOSTNAME/snapshot-1*
-	if [ $? -ne 0 ]; then
+	validate_trace_path_ust_uid_snapshot_network "$TRACE_PATH" "$SESSION_NAME" "snapshot-1" 0
+	if ! validate_trace $EVENT_NAME "$TRACE_PATH/$HOSTNAME/$SESSION_NAME*/snapshot-1*" ; then
 		stop_test_apps
 		return $?
 	fi
 
-	lttng_snapshot_del_output_ok $SESSION_NAME 1
-	snapshot_add_output $SESSION_NAME "net://localhost"
-	lttng_snapshot_record $SESSION_NAME
+	lttng_snapshot_del_output_ok "$SESSION_NAME" 1
+	snapshot_add_output "$SESSION_NAME" "net://localhost"
+	lttng_snapshot_record "$SESSION_NAME"
 
 	# Validate test with the next ID since a del output was done prior.
-	validate_trace $EVENT_NAME $TRACE_PATH/$HOSTNAME/snapshot-2*
-	if [ $? -ne 0 ]; then
+	validate_trace_path_ust_uid_snapshot_network "$TRACE_PATH" "$SESSION_NAME" "snapshot-2" 1
+	if ! validate_trace $EVENT_NAME "$TRACE_PATH/$HOSTNAME/$SESSION_NAME*/snapshot-2*" ; then
 		stop_test_apps
 		return $?
 	fi
 
-	stop_lttng_tracing_ok $SESSION_NAME
-	destroy_lttng_session_ok $SESSION_NAME
+	stop_lttng_tracing_ok "$SESSION_NAME"
+	destroy_lttng_session_ok "$SESSION_NAME"
 
 	stop_test_apps
 
@@ -129,19 +133,20 @@ function test_ust_default_name_with_del()
 function test_ust_default_name()
 {
 	diag "Test UST snapshot streaming with default name"
-	create_lttng_session_no_output $SESSION_NAME
-	enable_lttng_mmap_overwrite_ust_channel $SESSION_NAME $CHANNEL_NAME
-	enable_ust_lttng_event_ok $SESSION_NAME $EVENT_NAME $CHANNEL_NAME
-	start_lttng_tracing_ok $SESSION_NAME
+	create_lttng_session_no_output "$SESSION_NAME"
+	enable_lttng_mmap_overwrite_ust_channel "$SESSION_NAME" $CHANNEL_NAME
+	enable_ust_lttng_event_ok "$SESSION_NAME" $EVENT_NAME $CHANNEL_NAME
+	start_lttng_tracing_ok "$SESSION_NAME"
 
 	start_test_app
 
-	snapshot_add_output $SESSION_NAME "net://localhost"
-	lttng_snapshot_record $SESSION_NAME
-	stop_lttng_tracing_ok $SESSION_NAME
-	destroy_lttng_session_ok $SESSION_NAME
+	snapshot_add_output "$SESSION_NAME" "net://localhost"
+	lttng_snapshot_record "$SESSION_NAME"
+	stop_lttng_tracing_ok "$SESSION_NAME"
+	destroy_lttng_session_ok "$SESSION_NAME"
 	# Validate test
-	validate_trace $EVENT_NAME $TRACE_PATH/$HOSTNAME/snapshot-1*
+	validate_trace_path_ust_uid_snapshot_network "$TRACE_PATH" "$SESSION_NAME" "snapshot-1" 0
+	validate_trace $EVENT_NAME "$TRACE_PATH/$HOSTNAME/$SESSION_NAME*/snapshot-1*"
 	out=$?
 
 	stop_test_apps
@@ -152,19 +157,20 @@ function test_ust_default_name()
 function test_ust_default_name_custom_uri()
 {
 	diag "Test UST snapshot streaming with default name with custom URL"
-	create_lttng_session_no_output $SESSION_NAME
-	enable_lttng_mmap_overwrite_ust_channel $SESSION_NAME $CHANNEL_NAME
-	enable_ust_lttng_event_ok $SESSION_NAME $EVENT_NAME $CHANNEL_NAME
-	start_lttng_tracing_ok $SESSION_NAME
+	create_lttng_session_no_output "$SESSION_NAME"
+	enable_lttng_mmap_overwrite_ust_channel "$SESSION_NAME" $CHANNEL_NAME
+	enable_ust_lttng_event_ok "$SESSION_NAME" $EVENT_NAME $CHANNEL_NAME
+	start_lttng_tracing_ok "$SESSION_NAME"
 
 	start_test_app
 
-	snapshot_add_output $SESSION_NAME "-C tcp://localhost:5342 -D tcp://localhost:5343"
-	lttng_snapshot_record $SESSION_NAME
-	stop_lttng_tracing_ok $SESSION_NAME
-	destroy_lttng_session_ok $SESSION_NAME
+	snapshot_add_output "$SESSION_NAME" '' '' -C tcp://localhost:5342 -D tcp://localhost:5343
+	lttng_snapshot_record "$SESSION_NAME"
+	stop_lttng_tracing_ok "$SESSION_NAME"
+	destroy_lttng_session_ok "$SESSION_NAME"
 	# Validate test
-	validate_trace $EVENT_NAME $TRACE_PATH/$HOSTNAME/snapshot-1*
+	validate_trace_path_ust_uid_snapshot_network "$TRACE_PATH" "$SESSION_NAME" "snapshot-1" 0
+	validate_trace $EVENT_NAME "$TRACE_PATH/$HOSTNAME/$SESSION_NAME*/snapshot-1*"
 	out=$?
 
 	stop_test_apps
@@ -179,22 +185,23 @@ function test_ust_custom_name()
 	local name="asnapshotname"
 
 	diag "Test UST snapshot streaming with custom name"
-	create_lttng_session_no_output $SESSION_NAME
-	enable_lttng_mmap_overwrite_ust_channel $SESSION_NAME $CHANNEL_NAME
-	enable_ust_lttng_event_ok $SESSION_NAME $EVENT_NAME $CHANNEL_NAME
-	start_lttng_tracing_ok $SESSION_NAME
+	create_lttng_session_no_output "$SESSION_NAME"
+	enable_lttng_mmap_overwrite_ust_channel "$SESSION_NAME" $CHANNEL_NAME
+	enable_ust_lttng_event_ok "$SESSION_NAME" $EVENT_NAME $CHANNEL_NAME
+	start_lttng_tracing_ok "$SESSION_NAME"
 
 	start_test_app
 
-	snapshot_add_output $SESSION_NAME "net://localhost" $name
-	lttng_snapshot_record $SESSION_NAME
-	stop_lttng_tracing_ok $SESSION_NAME
-	destroy_lttng_session_ok $SESSION_NAME
+	snapshot_add_output "$SESSION_NAME" "net://localhost" $name
+	lttng_snapshot_record "$SESSION_NAME"
+	stop_lttng_tracing_ok "$SESSION_NAME"
+	destroy_lttng_session_ok "$SESSION_NAME"
 
-	if ls $TRACE_PATH/$HOSTNAME/$name* &> /dev/null; then
+	if ls "$TRACE_PATH/$HOSTNAME/$SESSION_NAME"*/"$name"* &> /dev/null; then
 		ok 0 "Custom name snapshot exists"
 		# Validate test
-		validate_trace $EVENT_NAME $TRACE_PATH/$HOSTNAME/$name-*
+		validate_trace_path_ust_uid_snapshot_network "$TRACE_PATH" "$SESSION_NAME" "$name" 0
+		validate_trace $EVENT_NAME "$TRACE_PATH/$HOSTNAME/$SESSION_NAME*/$name-*"
 		out=$?
 	else
 		fail "No custom name snapshot found"
@@ -206,32 +213,62 @@ function test_ust_custom_name()
 	return $out
 }
 
+function test_ust_n_snapshot()
+{
+	diag "Test ust snapshot streaming multiple consecutive snapshot"
+	create_lttng_session_no_output "$SESSION_NAME"
+	enable_lttng_mmap_overwrite_ust_channel "$SESSION_NAME" $CHANNEL_NAME
+	enable_ust_lttng_event_ok "$SESSION_NAME" $EVENT_NAME $CHANNEL_NAME
+	snapshot_add_output "$SESSION_NAME" "net://localhost"
+	start_test_app
+	for i in {0..4};
+	do
+		start_lttng_tracing_ok "$SESSION_NAME"
+		lttng_snapshot_record "$SESSION_NAME"
+		stop_lttng_tracing_ok "$SESSION_NAME"
+		validate_trace_path_ust_uid_snapshot_network "$TRACE_PATH" "$SESSION_NAME" "snapshot-1" "$i"
+		if ! validate_trace $EVENT_NAME "$TRACE_PATH/$HOSTNAME/$SESSION_NAME*/snapshot-1*" ; then
+			return 1
+		fi
+		set -u
+		rm -rf "${TRACE_PATH:?}/${HOSTNAME}"
+		set +u
+	done
+
+	destroy_lttng_session_ok "$SESSION_NAME"
+	stop_test_apps
+	return 0
+}
+
 plan_tests $NUM_TESTS
 
 print_test_banner "$TEST_DESC"
 
-if [ "$(id -u)" == "0" ]; then
-	isroot=1
-else
-	isroot=0
-fi
+bail_out_if_no_babeltrace
 
 start_lttng_relayd "-o $TRACE_PATH"
+# shellcheck disable=SC2119
 start_lttng_sessiond
 
-tests=( test_ust_default_name_with_del test_ust_default_name test_ust_custom_name test_ust_default_name_custom_uri )
+tests=( test_ust_default_name_with_del
+	test_ust_default_name
+	test_ust_custom_name
+	test_ust_default_name_custom_uri
+	test_ust_n_snapshot
+)
 
-for fct_test in ${tests[@]};
+for fct_test in "${tests[@]}";
 do
 	SESSION_NAME=$(randstring 16 0)
-	${fct_test}
-	if [ $? -eq 0 ]; then
+	if ${fct_test} ; then
 		# Only delete if successful
-		rm -rf $TRACE_PATH
+		rm -rf "$TRACE_PATH"
 	else
 		break
 	fi
 done
 
+# shellcheck disable=SC2119
 stop_lttng_sessiond
+# shellcheck disable=SC2119
 stop_lttng_relayd