+ local sess_name=$3
+ local trace_path=$4
+ local opt=$5
+
+ if [ -z "$trace_path" ]; then
+ # Use lttng-sessiond default output.
+ trace_path=""
+ else
+ trace_path="-o $trace_path"
+ fi
+
+ _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
+ create "$sess_name" $trace_path $opt
+ ret=$?
+ if [ $expected_to_fail -eq "1" ]; then
+ test "$ret" -ne "0"
+ ret=$?
+ if [ $withtap -eq "1" ]; then
+ ok $ret "Create session $sess_name in $trace_path failed as expected"
+ fi
+ else
+ if [ $withtap -eq "1" ]; then
+ ok $ret "Create session $sess_name in $trace_path"
+ fi
+ fi
+ return $ret
+}
+
+function create_lttng_session_ok ()
+{
+ create_lttng_session 1 0 "$@"
+}
+
+function create_lttng_session_fail ()
+{
+ create_lttng_session 1 1 "$@"
+}
+
+function create_lttng_session_notap ()
+{
+ create_lttng_session 0 0 "$@"
+}
+
+
+function enable_ust_lttng_channel ()
+{
+ local withtap=$1
+ local expected_to_fail=$2
+ local sess_name=$3
+ local channel_name=$4
+ local opts="${@:5}"
+
+ _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
+ enable-channel -u $channel_name -s $sess_name $opts
+ ret=$?
+ if [[ $expected_to_fail -eq "1" ]]; then
+ test "$ret" -ne "0"
+ ret=$?
+ if [ $withtap -eq "1" ]; then
+ ok $ret "Enable channel $channel_name for session $sess_name failed as expected"
+ fi
+ else
+ if [ $withtap -eq "1" ]; then
+ ok $ret "Enable channel $channel_name for session $sess_name"
+ fi
+ fi
+ return $ret
+}
+
+function enable_ust_lttng_channel_ok ()
+{
+ enable_ust_lttng_channel 1 0 "$@"
+}
+
+function enable_ust_lttng_channel_fail ()
+{
+ enable_ust_lttng_channel 1 1 "$@"
+}
+
+function enable_ust_lttng_channel_notap ()
+{
+ enable_ust_lttng_channel 0 0 "$@"
+}
+
+function disable_ust_lttng_channel()
+{
+ local sess_name=$1
+ local channel_name=$2
+
+ _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
+ disable-channel -u $channel_name -s $sess_name
+ ok $? "Disable channel $channel_name for session $sess_name"
+}
+
+function enable_lttng_mmap_overwrite_kernel_channel()
+{
+ local sess_name=$1
+ local channel_name=$2
+
+ _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
+ enable-channel -s $sess_name $channel_name -k --output mmap --overwrite
+ ok $? "Enable channel $channel_name for session $sess_name"
+}
+
+function enable_lttng_mmap_discard_small_kernel_channel()
+{
+ local sess_name=$1
+ local channel_name=$2
+
+ _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
+ enable-channel -s $sess_name $channel_name -k --output mmap --discard --subbuf-size=$(getconf PAGE_SIZE) --num-subbuf=2
+ ok $? "Enable small discard channel $channel_name for session $sess_name"
+}
+
+function enable_lttng_mmap_overwrite_small_kernel_channel()
+{
+ local sess_name=$1
+ local channel_name=$2
+
+ _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
+ enable-channel -s $sess_name $channel_name -k --output mmap --overwrite --subbuf-size=$(getconf PAGE_SIZE) --num-subbuf=2
+ ok $? "Enable small discard channel $channel_name for session $sess_name"
+}
+
+function enable_lttng_mmap_overwrite_ust_channel()
+{
+ local sess_name=$1
+ local channel_name=$2
+
+ _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
+ enable-channel -s $sess_name $channel_name -u --output mmap --overwrite
+ ok $? "Enable channel $channel_name for session $sess_name"
+}
+
+function enable_ust_lttng_event ()
+{
+ local withtap=$1
+ local expected_to_fail=$2
+ local sess_name=$3
+ local event_name="$4"
+ local channel_name=$5
+
+ if [ -z $channel_name ]; then
+ # default channel if none specified
+ chan=""
+ else
+ chan="-c $channel_name"
+ fi
+
+ _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
+ enable-event "$event_name" $chan -s "$sess_name" -u
+ ret=$?
+ if [[ $expected_to_fail -eq "1" ]]; then
+ test $ret -ne "0"
+ ret=$?
+ if [[ $withtap -eq "1" ]]; then
+ ok $ret "Enable ust event $event_name for session $session_name failed as expected"
+ fi
+ else
+ if [[ $withtap -eq "1" ]]; then
+ ok $ret "Enable ust event $event_name for session $sess_name"
+ fi
+ fi
+ return $ret
+}
+
+function enable_ust_lttng_event_ok ()
+{
+ enable_ust_lttng_event 1 0 "$@"
+}
+
+function enable_ust_lttng_event_fail ()
+{
+ enable_ust_lttng_event 1 1 "$@"
+}
+
+function enable_ust_lttng_event_notap ()
+{
+ enable_ust_lttng_event 0 0 "$@"
+}
+
+function enable_jul_lttng_event()
+{
+ sess_name=$1
+ event_name="$2"
+ channel_name=$3
+
+ if [ -z $channel_name ]; then
+ # default channel if none specified
+ chan=""
+ else
+ chan="-c $channel_name"
+ fi
+
+ _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
+ enable-event "$event_name" $chan -s $sess_name -j
+ ok $? "Enable JUL event $event_name for session $sess_name"
+}
+
+function enable_jul_lttng_event_loglevel()
+{
+ local sess_name=$1
+ local event_name="$2"
+ local loglevel=$3
+ local channel_name=$4
+
+ if [ -z $channel_name ]; then
+ # default channel if none specified
+ chan=""
+ else
+ chan="-c $channel_name"
+ fi
+
+ _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
+ enable-event --loglevel $loglevel "$event_name" $chan -s $sess_name -j
+ ok $? "Enable JUL event $event_name for session $sess_name with loglevel $loglevel"
+}
+
+function enable_log4j_lttng_event()
+{
+ local sess_name=$1
+ local event_name=$2
+ local channel_name=$3
+
+ local chan_opt=()
+
+ # default channel if none specified
+ if [ -n "$channel_name" ]; then
+ chan_opt=("-c" "$channel_name")
+ fi
+
+ _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
+ enable-event "$event_name" "${chan_opt[@]}" -s "$sess_name" --log4j
+ ok $? "Enable LOG4J event '$event_name' for session '$sess_name'"
+}
+
+function enable_log4j_lttng_event_filter()
+{
+ local sess_name=$1
+ local event_name=$2
+ local filter=$3
+
+ _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
+ enable-event "$event_name" -s "$sess_name" --log4j --filter "$filter"
+ ok $? "Enable LOG4J event '$event_name' with filter '$filter' for session '$sess_name'"
+}
+
+function enable_log4j_lttng_event_filter_loglevel_only()
+{
+ local sess_name=$1
+ local event_name=$2
+ local filter=$3
+ local loglevel=$4
+
+ _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
+ enable-event --loglevel-only "$loglevel" "$event_name" -s "$sess_name" -l --filter "$filter"
+ ok $? "Enable LOG4J event '$event_name' with filter '$filter' and loglevel-only '$loglevel' for session '$sess_name'"
+}
+
+function enable_log4j_lttng_event_loglevel()
+{
+ local sess_name=$1
+ local event_name=$2
+ local loglevel=$3
+ local channel_name=$4
+
+
+ # default channel if none specified
+ if [ -n "$channel_name" ]; then
+ chan_opt=("-c" "$channel_name")
+ fi
+
+ _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
+ enable-event --loglevel "$loglevel" "$event_name" "${chan_opt[@]}" -s "$sess_name" --log4j
+ ok $? "Enable LOG4J event '$event_name' for session '$sess_name' with loglevel '$loglevel'"
+}
+
+function enable_log4j_lttng_event_loglevel_only()
+{
+ local sess_name=$1
+ local event_name=$2
+ local loglevel=$3
+ local channel_name=$4
+
+ local chan_opt=()
+
+ # default channel if none specified
+ if [ -n "$channel_name" ]; then
+ chan_opt=("-c" "$channel_name")
+ fi
+
+ _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
+ enable-event --loglevel-only "$loglevel" "$event_name" "${chan_opt[@]}" -s "$sess_name" --log4j
+ ok $? "Enable LOG4J event '$event_name' for session '$sess_name' with loglevel-only '$loglevel'"
+}
+
+function enable_python_lttng_event()
+{
+ sess_name=$1
+ event_name="$2"
+ channel_name=$3
+
+ if [ -z $channel_name ]; then
+ # default channel if none specified
+ chan=""
+ else
+ chan="-c $channel_name"
+ fi
+
+ _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
+ enable-event "$event_name" $chan -s $sess_name -p
+ ok $? "Enable Python event $event_name for session $sess_name"
+}
+
+function enable_python_lttng_event_loglevel()
+{
+ local sess_name=$1
+ local event_name="$2"
+ local loglevel=$3
+ local channel_name=$4
+
+ if [ -z $channel_name ]; then
+ # default channel if none specified
+ chan=""
+ else
+ chan="-c $channel_name"
+ fi
+
+ _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
+ enable-event --loglevel $loglevel "$event_name" $chan -s $sess_name -p
+ ok $? "Enable Python event $event_name for session $sess_name with loglevel $loglevel"
+}
+
+function enable_ust_lttng_event_filter()
+{
+ local sess_name="$1"
+ local event_name="$2"
+ local filter="$3"
+ local channel_name=$4
+
+ if [ -z $channel_name ]; then
+ # default channel if none specified
+ chan=""
+ else
+ chan="-c $channel_name"
+ fi
+
+ _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
+ enable-event $chan "$event_name" -s $sess_name -u --filter "$filter"
+ ok $? "Enable event $event_name with filtering for session $sess_name"
+}
+
+function enable_ust_lttng_event_loglevel()
+{
+ local sess_name="$1"
+ local event_name="$2"
+ local loglevel="$3"
+
+ _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
+ enable-event "$event_name" -s $sess_name -u --loglevel $loglevel
+ ok $? "Enable event $event_name with loglevel $loglevel"
+}
+
+function enable_ust_lttng_event_loglevel_only()
+{
+ local sess_name="$1"
+ local event_name="$2"
+ local loglevel="$3"
+
+ _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
+ enable-event "$event_name" -s $sess_name -u --loglevel-only $loglevel
+ ok $? "Enable event $event_name with loglevel-only $loglevel"
+}
+
+function disable_ust_lttng_event ()
+{
+ local sess_name="$1"
+ local event_name="$2"
+ local channel_name="$3"
+
+ if [ -z $channel_name ]; then
+ # default channel if none specified
+ chan=""
+ else
+ chan="-c $channel_name"
+ fi
+
+ _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
+ disable-event "$event_name" -s $sess_name $chan -u
+ ok $? "Disable event $event_name for session $sess_name"
+}
+
+function disable_jul_lttng_event ()
+{
+ local sess_name="$1"
+ local event_name="$2"
+
+ $TESTDIR/../src/bin/lttng/$LTTNG_BIN disable-event "$event_name" -s $sess_name -j >/dev/null 2>&1
+ ok $? "Disable JUL event $event_name for session $sess_name"
+}
+
+function disable_log4j_lttng_event ()
+{
+ local sess_name="$1"
+ local event_name="$2"
+
+ _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
+ disable-event "$event_name" -s "$sess_name" --log4j
+ ok $? "Disable LOG4J event '$event_name' for session '$sess_name'"
+}
+
+function disable_python_lttng_event ()
+{
+ local sess_name="$1"
+ local event_name="$2"
+
+ _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
+ disable-event "$event_name" -s $sess_name -p
+ ok $? "Disable Python event $event_name for session $sess_name"
+}
+
+function start_lttng_tracing_opt ()
+{
+ local withtap=$1
+ local expected_to_fail=$2
+ local sess_name=$3
+
+ _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
+ start $sess_name
+ ret=$?
+ if [[ $expected_to_fail -eq "1" ]]; then
+ test "$ret" -ne "0"
+ ret=$?
+ if [ $withtap -eq "1" ]; then
+ ok $? "Start tracing for session $sess_name failed as expected"
+ fi
+ else
+ if [ $withtap -eq "1" ]; then
+ ok $ret "Start tracing for session $sess_name"
+ fi
+ fi
+}
+
+function start_lttng_tracing_ok ()
+{
+ start_lttng_tracing_opt 1 0 "$@"
+}
+
+function start_lttng_tracing_fail ()
+{
+ start_lttng_tracing_opt 1 1 "$@"
+}
+
+function start_lttng_tracing_notap ()
+{
+ start_lttng_tracing_opt 0 1 "$@"
+}
+
+function stop_lttng_tracing_opt ()
+{
+ local withtap=$1
+ local expected_to_fail=$2
+ local sess_name=$3
+
+ _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
+ stop $sess_name
+ ret=$?
+ if [[ $expected_to_fail -eq "1" ]]; then
+ test "$ret" -ne "0"
+ ret=$?
+ if [ $withtap -eq "1" ]; then
+ ok $? "Stop lttng tracing for session $sess_name failed as expected"
+ fi
+ else
+ if [ $withtap -eq "1" ]; then
+ ok $ret "Stop lttng tracing for session $sess_name"
+ fi
+ fi
+}
+
+function stop_lttng_tracing_ok ()
+{
+ stop_lttng_tracing_opt 1 0 "$@"
+}
+
+function stop_lttng_tracing_fail ()
+{
+ stop_lttng_tracing_opt 1 1 "$@"
+}
+
+function stop_lttng_tracing_notap ()
+{
+ stop_lttng_tracing_opt 0 0 "$@"
+}
+
+function destroy_lttng_session ()
+{
+ local withtap=$1
+ local expected_to_fail=$2
+ local sess_name=$3
+
+ _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
+ destroy $sess_name
+ ret=$?
+ if [[ $expected_to_fail -eq "1" ]]; then
+ test "$ret" -ne "0"
+ ret=$?
+ if [ $withtap -eq "1" ]; then
+ ok $ret "Destroy session $sess_name failed as expected"
+ fi
+ else
+ if [ $withtap -eq "1" ]; then
+ ok $ret "Destroy session $sess_name"
+ fi
+ fi
+}
+
+function destroy_lttng_session_ok ()
+{
+ destroy_lttng_session 1 0 "$@"
+
+}
+
+function destroy_lttng_session_fail ()
+{
+ destroy_lttng_session 1 1 "$@"
+}
+
+function destroy_lttng_session_notap ()
+{
+ destroy_lttng_session 0 0 "$@"
+}
+
+function destroy_lttng_sessions ()
+{
+ _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
+ destroy --all
+ ok $? "Destroy all lttng sessions"
+}
+
+function lttng_snapshot_add_output ()
+{
+ local expected_to_fail=$1
+ local sess_name=$2
+ local trace_path=$3
+ local opts=$4
+
+ _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
+ snapshot add-output -s $sess_name $trace_path $opts
+ ret=$?
+ if [[ $expected_to_fail -eq 1 ]]; then
+ test "$ret" -ne "0"
+ ok $? "Added snapshot output $trace_path failed as expected"
+ else
+ ok $ret "Added snapshot output $trace_path"
+ fi
+}
+
+function lttng_snapshot_add_output_ok ()
+{
+ lttng_snapshot_add_output 0 "$@"
+}
+
+function lttng_snapshot_add_output_fail ()
+{
+ lttng_snapshot_add_output 1 "$@"
+}
+
+function lttng_snapshot_del_output ()
+{
+ local expected_to_fail=$1
+ local sess_name=$2
+ local id=$3
+
+ _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
+ snapshot del-output -s $sess_name $id
+ ret=$?
+ if [[ $expected_to_fail -eq "1" ]]; then
+ test "$ret" -ne "0"
+ ok $? "Deleted snapshot output id $id failed as expected"
+ else
+ ok $ret "Deleted snapshot output id $id"
+ fi
+}
+
+function lttng_snapshot_del_output_ok ()
+{
+ lttng_snapshot_del_output 0 "$@"
+}
+
+function lttng_snapshot_del_output_fail ()
+{
+ lttng_snapshot_del_output 1 "$@"
+}
+
+function lttng_snapshot_record ()
+{
+ local sess_name=$1
+ local trace_path=$2
+
+ _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
+ snapshot record -s "$sess_name" "$trace_path"
+ ok $? "Snapshot recorded"
+}
+
+function lttng_snapshot_list ()
+{
+ local sess_name=$1
+ _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
+ snapshot list-output -s $sess_name
+ ok $? "Snapshot list"
+}
+
+function lttng_save()
+{
+ local sess_name=$1
+ local opts=$2
+
+ _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
+ save $sess_name $opts
+ ok $? "Session saved"
+}
+
+function lttng_load()
+{
+ local expected_to_fail=$1
+ local opts=$2
+
+ _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
+ load $opts
+ ret=$?
+ if [[ $expected_to_fail -eq "1" ]]; then
+ test $ret -ne "0"
+ ok $? "Load command failed as expected with opts: $opts"
+ else
+ ok $ret "Load command with opts: $opts"
+ fi
+}
+
+function lttng_load_ok()
+{
+ lttng_load 0 "$@"
+}
+
+function lttng_load_fail()
+{
+ lttng_load 1 "$@"
+}
+
+function lttng_track()
+{
+ local expected_to_fail="$1"
+ shift 1
+ local opts="$@"
+ $TESTDIR/../src/bin/lttng/$LTTNG_BIN track $opts >$OUTPUT_DEST
+ ret=$?
+ if [[ $expected_to_fail -eq "1" ]]; then
+ test $ret -ne "0"
+ ok $? "Track command failed as expected with opts: $opts"
+ else
+ ok $ret "Track command with opts: $opts"
+ fi
+}
+
+function lttng_track_ok()
+{
+ lttng_track 0 "$@"
+}
+
+function lttng_track_fail()
+{
+ lttng_track 1 "$@"
+}
+
+function lttng_untrack()
+{
+ local expected_to_fail="$1"
+ shift 1
+ local opts="$@"
+ $TESTDIR/../src/bin/lttng/$LTTNG_BIN untrack $opts >$OUTPUT_DEST
+ ret=$?
+ if [[ $expected_to_fail -eq "1" ]]; then
+ test $ret -ne "0"
+ ok $? "Untrack command failed as expected with opts: $opts"
+ else
+ ok $ret "Untrack command with opts: $opts"
+ fi
+}
+
+function lttng_untrack_ok()
+{
+ lttng_untrack 0 "$@"
+}
+
+function lttng_untrack_fail()
+{
+ lttng_untrack 1 "$@"
+}
+
+function lttng_track_pid_ok()
+{
+ PID=$1
+ "$TESTDIR/../src/bin/lttng/$LTTNG_BIN" track --kernel --pid=$PID 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
+ ok $? "Lttng track pid on the kernel domain"
+}
+
+function lttng_untrack_kernel_all_ok()
+{
+ "$TESTDIR/../src/bin/lttng/$LTTNG_BIN" untrack --kernel --pid --all 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
+ ok $? "Lttng untrack all pid on the kernel domain"
+}
+
+function lttng_track_ust_ok()
+{
+ lttng_track_ok -u "$@"
+}
+
+function lttng_track_ust_fail()
+{
+ lttng_track_fail -u "$@"
+}
+
+function lttng_track_kernel_ok()
+{
+ lttng_track_ok -k "$@"
+}
+
+function lttng_track_kernel_fail()
+{
+ lttng_track_fail -k "$@"
+}
+
+function lttng_untrack_ust_ok()
+{
+ lttng_untrack_ok -u "$@"
+}
+
+function lttng_untrack_ust_fail()
+{
+ lttng_untrack_fail -u "$@"
+}
+
+function lttng_untrack_kernel_ok()
+{
+ lttng_untrack_ok -k "$@"
+}
+
+function lttng_untrack_kernel_fail()
+{
+ lttng_untrack_fail -k "$@"
+}
+
+function lttng_add_context_list()
+{
+ _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
+ add-context --list
+ ret=$?
+ ok $ret "Context listing"
+}
+
+function add_context_lttng()
+{
+ local expected_to_fail="$1"
+ local domain="$2"
+ local session_name="$3"
+ local channel_name="$4"
+ local type="$5"
+
+ _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
+ add-context -s $session_name -c $channel_name -t $type $domain
+ ret=$?
+ if [[ $expected_to_fail -eq "1" ]]; then
+ test $ret -ne "0"
+ ok $? "Add context command failed as expected for type: $type"
+ else
+ ok $ret "Add context command for type: $type"
+ fi
+}
+
+function add_context_ust_ok()
+{
+ add_context_lttng 0 -u "$@"
+}
+
+function add_context_ust_fail()
+{
+ add_context_lttng 1 -u "$@"
+}
+
+function add_context_kernel_ok()
+{
+ add_context_lttng 0 -k "$@"
+}
+
+function add_context_kernel_fail()
+{
+ add_context_lttng 1 -k "$@"
+}
+
+function wait_live_trace_ready ()
+{
+ local url=$1
+ local zero_client_match=0
+
+ diag "Waiting for live trace at url: $url"
+ while [ $zero_client_match -eq 0 ]; do
+ zero_client_match=$($BABELTRACE_BIN -i lttng-live $url | grep "0 client(s) connected" | wc -l)
+ sleep 0.5
+ done
+ pass "Waiting for live trace at url: $url"
+}
+
+function wait_live_viewer_connect ()
+{
+ local url=$1
+ local one_client_match=0
+
+ diag "Waiting for live viewers on url: $url"
+ while [ $one_client_match -eq 0 ]; do
+ one_client_match=$($BABELTRACE_BIN -i lttng-live $url | grep "1 client(s) connected" | wc -l)
+ sleep 0.5
+ done
+ pass "Waiting for live viewers on url: $url"
+}
+
+function bail_out_if_no_babeltrace()
+{
+ which "$BABELTRACE_BIN" >/dev/null
+ if [ $? -ne 0 ]; then
+ LTTNG_BAIL_OUT "\"$BABELTRACE_BIN\" binary not found. Skipping tests"
+ fi
+}
+
+# Check that the trace metadata contains '$expected' event ids matching '$event_name'.
+function validate_metadata_event()
+{
+ local event_name=$1
+ local expected=$2
+ local trace_path=$3
+
+ local metadata_file
+ local metadata_path
+ local count
+
+ metadata_file=$(find "$trace_path" -name "metadata")
+ metadata_path=$(dirname "$metadata_file")
+
+ bail_out_if_no_babeltrace
+
+ count=$($BABELTRACE_BIN --output-format=ctf-metadata "$metadata_path" | grep -c "$event_name")
+
+ test "$count" -eq "$expected"
+ ok $? "Found $count / $expected metadata event id matching '$event_name'"
+}
+
+# Check that the trace contains '$expected' events matching '$event_name', other
+# events not matching '$event_name' can be present.
+function trace_matches()
+{
+ local event_name=$1
+ local expected=$2
+ local trace_path=$3
+
+ local count
+ local total
+
+ bail_out_if_no_babeltrace
+
+ count=$($BABELTRACE_BIN "$trace_path" | grep -c "$event_name")
+ total=$($BABELTRACE_BIN "$trace_path" | wc -l)
+
+ test "$count" -eq "$expected"
+
+ ok $? "Found $count / $expected events matching '$event_name' out of $total events"
+}
+
+# Check that the trace contains '$expected' events matching '$event_name' and no
+# other events.
+function trace_match_only()
+{
+ local event_name=$1
+ local expected=$2
+ local trace_path=$3
+
+ local count
+ local total
+
+ bail_out_if_no_babeltrace
+
+ count=$($BABELTRACE_BIN "$trace_path" | grep -c "$event_name")
+ total=$($BABELTRACE_BIN "$trace_path" | wc -l)
+
+ test "$expected" -eq "$count" && test "$total" -eq "$expected"