Tests: per-UID UST snapshot

Signed-off-by: Julien Desfossez <jdesfossez@efficios.com>
Signed-off-by: David Goulet <dgoulet@efficios.com>

diff --git a/tests/regression/tools/snapshots/test_ust b/tests/regression/tools/snapshots/test_ust
index 1e24411aa22eb0ff070e8f6f1f847a8de35ad8ee..d4fe165ee153be8a95a44353fe87b985fa56a974 100755 (executable)
--- a/tests/regression/tools/snapshots/test_ust
+++ b/tests/regression/tools/snapshots/test_ust
@@ -31,7 +31,7 @@ NR_USEC_WAIT=100
 
 TRACE_PATH=$(mktemp -d)
 
-NUM_TESTS=2019
+NUM_TESTS=2029
 
 source $TESTDIR/utils/utils.sh
 
@@ -66,6 +66,42 @@ function test_ust_local_snapshot ()
        kill $PID_APP >/dev/null 2>&1
 }
 
+function enable_channel_per_uid_mmap_overwrite()
+{
+       sess_name=$1
+       channel_name=$2
+
+       $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-channel --buffers-uid -u $channel_name -s $sess_name --output mmap --overwrite >/dev/null 2>&1
+       ok $? "Enable channel $channel_name per UID for session $sess_name"
+}
+
+function test_ust_per_uid_local_snapshot ()
+{
+       diag "Test local UST snapshots"
+       create_lttng_session_no_output $SESSION_NAME
+       enable_channel_per_uid_mmap_overwrite $SESSION_NAME $CHANNEL_NAME
+       enable_ust_lttng_event $SESSION_NAME $EVENT_NAME $CHANNEL_NAME
+       start_lttng_tracing $SESSION_NAME
+       lttng_snapshot_add_output $SESSION_NAME $TRACE_PATH
+       $TESTAPP_BIN $NR_ITER $NR_USEC_WAIT &
+       ok $? "Start application to trace"
+       lttng_snapshot_record $SESSION_NAME
+       stop_lttng_tracing $SESSION_NAME
+       destroy_lttng_session $SESSION_NAME
+
+       # Validate test
+       validate_trace $EVENT_NAME $TRACE_PATH/
+       if [ $? -eq 0 ]; then
+               # Only delete if successful
+               rm -rf $TRACE_PATH
+       else
+               break
+       fi
+       diag "Killing $TESTAPP_NAME"
+       PID_APP=`pidof $TESTAPP_NAME`
+       kill $PID_APP >/dev/null 2>&1
+}
+
 function test_ust_1000_local_snapshots ()
 {
        NB_SNAP=1000
@@ -109,7 +145,7 @@ fi
 
 start_lttng_sessiond
 
-tests=( test_ust_local_snapshot test_ust_1000_local_snapshots )
+tests=( test_ust_local_snapshot test_ust_per_uid_local_snapshot test_ust_1000_local_snapshots )
 
 for fct_test in ${tests[@]};
 do