X-Git-Url: https://git.liburcu.org/?a=blobdiff_plain;f=tests%2Fregression%2Ftools%2Fnotification%2Frotation.c;h=f32a0d7737efb95b3b778c3e14784ec19dc40395;hb=a5c2d2a71919b8d1542b62f6d32579125cc2c8f8;hp=11768fd81998ca96afc620bcf910a28395261baf;hpb=9d16b343fb9e781fc8d8fa3c448a3f382306dd33;p=lttng-tools.git

diff --git a/tests/regression/tools/notification/rotation.c b/tests/regression/tools/notification/rotation.c
index 11768fd81..f32a0d773 100644
--- a/tests/regression/tools/notification/rotation.c
+++ b/tests/regression/tools/notification/rotation.c
@@ -14,17 +14,8 @@
 #include <assert.h>
 #include <tap/tap.h>
 #include <stdint.h>
-#include <lttng/rotation.h>
-#include <lttng/notification/channel.h>
-#include <lttng/notification/notification.h>
-#include <lttng/condition/evaluation.h>
-#include <lttng/condition/condition.h>
-#include <lttng/endpoint.h>
-#include <lttng/action/notify.h>
-#include <lttng/action/action.h>
-#include <lttng/trigger/trigger.h>
-#include <lttng/condition/session-rotation.h>
 #include <string.h>
+#include <lttng/lttng.h>
 
 #define TEST_COUNT 36
 
@@ -79,6 +70,7 @@ int setup_rotation_trigger(const struct session *session,
 	struct lttng_trigger *rotation_completed_trigger = NULL;
 	enum lttng_condition_status condition_status;
 	enum lttng_notification_channel_status notification_channel_status;
+	enum lttng_error_code ret_code;
 
 	notify = lttng_action_notify_create();
 	if (!notify) {
@@ -162,17 +154,23 @@ int setup_rotation_trigger(const struct session *session,
 	}
 
 	/* Register rotation ongoing and completed triggers. */
-	ret = lttng_register_trigger(rotation_ongoing_trigger);
-	ok(ret == 0, "Registered session rotation ongoing trigger");
-	if (ret) {
+	ret_code = lttng_register_trigger_with_automatic_name(
+			rotation_ongoing_trigger);
+	ok(ret_code == LTTNG_OK, "Registered session rotation ongoing trigger");
+	if (ret_code != LTTNG_OK) {
+		ret = -ret_code;
 		goto end;
 	}
 
-	ret = lttng_register_trigger(rotation_completed_trigger);
-	ok(ret == 0, "Registered session rotation completed trigger");
-	if (ret) {
+	ret_code = lttng_register_trigger_with_automatic_name(
+			rotation_completed_trigger);
+	ok(ret_code == LTTNG_OK,
+			"Registered session rotation completed trigger");
+	if (ret_code != LTTNG_OK) {
+		ret = -ret_code;
 		goto end;
 	}
+
 end:
 	lttng_trigger_destroy(rotation_ongoing_trigger);
 	lttng_trigger_destroy(rotation_completed_trigger);
@@ -409,5 +407,5 @@ int main(int argc, const char *argv[])
 error:
 	lttng_notification_channel_destroy(notification_channel);
 	lttng_rotation_handle_destroy(rotation_handle);
-	return ret;
+	return exit_status();
 }