Fix: Tests that assume CPU 0 is present

[lttng-tools.git] / tests / regression / tools / snapshots / test_kernel

diff --git a/tests/regression/tools/snapshots/test_kernel b/tests/regression/tools/snapshots/test_kernel
index d4e0008f9be279c860a39303c3d225acea4ea8aa..870c437b217fc2ed0eb339628fff428e220312fd 100755 (executable)
--- a/tests/regression/tools/snapshots/test_kernel
+++ b/tests/regression/tools/snapshots/test_kernel
@@ -12,7 +12,7 @@ EVENT_NAME="sched_switch"
 SESSION_NAME=""
 CHANNEL_NAME="snapchan"
 
-TRACE_PATH=$(mktemp -d)
+TRACE_PATH=$(mktemp -d -t tmp.test_snapshots_kernel_trace_path.XXXXXX)
 
 NUM_TESTS=2061
 
@@ -56,8 +56,6 @@ function test_kernel_local_snapshot_after_stop ()
        if [ $? -eq 0 ]; then
                # Only delete if successful
                rm -rf $TRACE_PATH
-       else
-               break
        fi
 }
 
@@ -79,8 +77,6 @@ function test_kernel_local_snapshot_append_to_metadata ()
        if [ $? -eq 0 ]; then
                # Only delete if successful
                rm -rf $TRACE_PATH
-       else
-               break
        fi
 
        diag "Adding event $EVENT2"
@@ -92,33 +88,33 @@ function test_kernel_local_snapshot_append_to_metadata ()
        if [ $? -eq 0 ]; then
                # Only delete if successful
                rm -rf $TRACE_PATH
-       else
-               break
        fi
 
        stop_lttng_tracing_ok $SESSION_NAME
        destroy_lttng_session_ok $SESSION_NAME
 }
 
-function true_loop_cpu0()
+function generate_filter_events_cpu0()
 {
-       # Generate many system call events (fill buffers) on CPU 0
-       for a in $(seq 1 $1); do
-               taskset 0x00000001 /bin/true;
-       done
+       # Generate many events (fill buffers) on a single CPU
+       taskset --cpu-list "$(get_any_available_cpu)" /bin/echo -n "$1" > /proc/lttng-test-filter-event 2> /dev/null
 }
 
 function test_kernel_local_snapshot_discard ()
 {
        diag "Test local kernel snapshots with small discard buffers"
+
+       local event_name="lttng_test_filter_event"
+
+       modprobe lttng-test
        create_lttng_session_no_output $SESSION_NAME
        enable_lttng_mmap_discard_small_kernel_channel $SESSION_NAME \
                        $CHANNEL_NAME
-       lttng_enable_kernel_syscall_ok $SESSION_NAME -a $CHANNEL_NAME
+       enable_kernel_lttng_event_ok $SESSION_NAME $event_name $CHANNEL_NAME
        start_lttng_tracing_ok $SESSION_NAME
        lttng_snapshot_add_output_ok $SESSION_NAME file://$TRACE_PATH
 
-       true_loop_cpu0 10000
+       generate_filter_events_cpu0 10000
 
        # Take first snapshot, remember first line.
        lttng_snapshot_record $SESSION_NAME
@@ -127,7 +123,8 @@ function test_kernel_local_snapshot_discard ()
 
        rm -rf $TRACE_PATH
 
-       true_loop_cpu0 10000
+       # Fill CPU 0's buffer and overwrite previous contents
+       generate_filter_events_cpu0 10000
 
        # Take 2nd snapshot, compare first line. In discard mode, they
        # should still be the same.
@@ -144,19 +141,24 @@ function test_kernel_local_snapshot_discard ()
 
        stop_lttng_tracing_ok $SESSION_NAME
        destroy_lttng_session_ok $SESSION_NAME
+       modprobe --remove lttng-test
 }
 
 function test_kernel_local_snapshot_overwrite_small_buffers ()
 {
        diag "Test local kernel snapshot with small overwrite buffers"
+
+       local event_name="lttng_test_filter_event"
+
+       modprobe lttng-test
        create_lttng_session_no_output $SESSION_NAME
        enable_lttng_mmap_overwrite_small_kernel_channel $SESSION_NAME \
                        $CHANNEL_NAME
-       lttng_enable_kernel_syscall_ok $SESSION_NAME -a $CHANNEL_NAME
+       enable_kernel_lttng_event_ok $SESSION_NAME $event_name $CHANNEL_NAME
        start_lttng_tracing_ok $SESSION_NAME
        lttng_snapshot_add_output_ok $SESSION_NAME file://$TRACE_PATH
 
-       true_loop_cpu0 10000
+       generate_filter_events_cpu0 10000
 
        # Take first snapshot, remember first line.
        lttng_snapshot_record $SESSION_NAME
@@ -165,7 +167,8 @@ function test_kernel_local_snapshot_overwrite_small_buffers ()
 
        rm -rf $TRACE_PATH
 
-       true_loop_cpu0 10000
+       # Fill CPU 0's buffer and overwrite previous contents
+       generate_filter_events_cpu0 10000
 
        # Take 2nd snapshot, compare first line. In overwrite mode, they
        # WILL be different.
@@ -182,6 +185,7 @@ function test_kernel_local_snapshot_overwrite_small_buffers ()
 
        stop_lttng_tracing_ok $SESSION_NAME
        destroy_lttng_session_ok $SESSION_NAME
+       modprobe --remove lttng-test
 }
 
 function test_kernel_1000_local_snapshots ()
@@ -215,6 +219,9 @@ plan_tests $NUM_TESTS
 
 print_test_banner "$TEST_DESC"
 
+bail_out_if_no_babeltrace
+
+
 if [ "$(id -u)" == "0" ]; then
        isroot=1
 else