X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=tests%2Fregression%2Ftools%2Frotation%2Ftest_ust;fp=tests%2Fregression%2Ftools%2Frotation%2Ftest_ust;h=aaeee4454c7a38255bb84af162b9b4450d04a031;hp=7c52f34bced2ac016ba31a43b3213299b5d5ffab;hb=34764ec838debdb5fdb59e178b31a44047ea7229;hpb=bf1e8411c30b21bc8dd6184c9890e9885bb7d503

diff --git a/tests/regression/tools/rotation/test_ust b/tests/regression/tools/rotation/test_ust
index 7c52f34bc..aaeee4454 100755
--- a/tests/regression/tools/rotation/test_ust
+++ b/tests/regression/tools/rotation/test_ust
@@ -255,6 +255,13 @@ function test_incompatible_sessions ()
 	destroy_lttng_session_ok $SESSION_NAME
 }
 
+function produce_n_events ()
+{
+	local event_count=$1
+
+	$TESTAPP_BIN -i "$event_count" -w 0 > /dev/null 2>&1
+}
+
 function test_ust_local_size_uid ()
 {
 	diag "Rotate uid local session every 2MiB"
@@ -267,7 +274,7 @@ function test_ust_local_size_uid ()
 	lttng_enable_rotation_size_ok $SESSION_NAME $size_threshold
 	start_lttng_tracing_ok $SESSION_NAME
 
-	wait_for_archives "$TRACE_PATH" 5
+	trace_until_n_archives produce_n_events "$TRACE_PATH" 5
 
 	destroy_lttng_session_ok $SESSION_NAME
 }
@@ -284,7 +291,7 @@ function test_ust_local_size_pid ()
 	lttng_enable_rotation_size_ok $SESSION_NAME $size_threshold
 	start_lttng_tracing_ok $SESSION_NAME
 
-	wait_for_archives "$TRACE_PATH" 3
+	trace_until_n_archives produce_n_events "$TRACE_PATH" 3
 
 	destroy_lttng_session_ok $SESSION_NAME
 }