.gitignore: ignore local vscode workspace settings file
[lttng-tools.git] / tests / regression / tools / snapshots / test_ust_streaming
index 3aa38581eac519e56d368d568318d7c3a2760f6a..2e04d667e76890a9dbf7141fef82b429d4f9dbad 100755 (executable)
@@ -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"
-PID_RELAYD=0
 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
+APP_PIDS=()
 
-TRACE_PATH=$(mktemp -d)
+TRACE_PATH=$(mktemp -d -t tmp.test_snapshots_ust_streaming_trace_path.XXXXXX)
 
-NUM_TESTS=39
+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,52 +34,97 @@ 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
+       tmp_file=$(mktemp -u -t tmp.test_"${FUNCNAME[0]}"_tmp_file.XXXXXX)
+
+       # Start application with a temporary file.
+       $TESTAPP_BIN -i $NR_ITER -w $NR_USEC_WAIT --sync-after-first-event "$tmp_file" &
+       ret=$?
+       APP_PIDS+=(${!})
+       ok $ret "Start application to trace"
+
+       # Wait for the application file to appear indicating that at least one
+       # tracepoint has been fired.
+       while [ ! -f "$tmp_file" ]; do
+               sleep 0.5
+       done
+       diag "Removing test app temporary file $tmp_file"
+       rm -f "$tmp_file"
+}
+
+function stop_test_apps()
+{
+       diag "Stopping $TESTAPP_NAME"
+       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 $SESSION_NAME $EVENT_NAME $CHANNEL_NAME
-       start_lttng_tracing $SESSION_NAME
-       $TESTAPP_BIN $NR_ITER $NR_USEC_WAIT &
-       ok $? "Start application to trace"
-       snapshot_add_output $SESSION_NAME "net://localhost"
-       lttng_snapshot_record $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"
 
        # 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 $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 $SESSION_NAME
-       destroy_lttng_session $SESSION_NAME
+       stop_lttng_tracing_ok "$SESSION_NAME"
+       destroy_lttng_session_ok "$SESSION_NAME"
 
-       diag "Killing $TESTAPP_NAME"
-       PID_APP=`pidof $TESTAPP_NAME`
-       kill $PID_APP >/dev/null 2>&1
+       stop_test_apps
 
        return 0
 }
@@ -98,23 +133,47 @@ 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 $SESSION_NAME $EVENT_NAME $CHANNEL_NAME
-       start_lttng_tracing $SESSION_NAME
-       $TESTAPP_BIN $NR_ITER $NR_USEC_WAIT &
-       ok $? "Start application to trace"
-       snapshot_add_output $SESSION_NAME "net://localhost"
-       lttng_snapshot_record $SESSION_NAME
-       stop_lttng_tracing $SESSION_NAME
-       destroy_lttng_session $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"
+       # Validate test
+       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
+
+       return $out
+}
+
+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"
+
+       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"
        # 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=$?
 
-       diag "Killing $TESTAPP_NAME"
-       PID_APP=`pidof $TESTAPP_NAME`
-       kill $PID_APP >/dev/null 2>&1
+       stop_test_apps
 
        return $out
 }
@@ -126,60 +185,90 @@ 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 $SESSION_NAME $EVENT_NAME $CHANNEL_NAME
-       start_lttng_tracing $SESSION_NAME
-       $TESTAPP_BIN $NR_ITER $NR_USEC_WAIT &
-       ok $? "Start application to trace"
-       snapshot_add_output $SESSION_NAME "net://localhost" $name
-       lttng_snapshot_record $SESSION_NAME
-       stop_lttng_tracing $SESSION_NAME
-       destroy_lttng_session $SESSION_NAME
-
-       if ls $TRACE_PATH/$HOSTNAME/$name* &> /dev/null; then
+       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"
+
+       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"
                out=1
        fi
 
-       diag "Killing $TESTAPP_NAME"
-       PID_APP=`pidof $TESTAPP_NAME`
-       kill $PID_APP >/dev/null 2>&1
+       stop_test_apps
 
        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 )
+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
This page took 0.026977 seconds and 4 git commands to generate.