X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=tests%2Flib%2Ftest_bt_notification_iterator.c;h=cbce6b4c965a00acef12aa469933640ae6307e5f;hb=73d5c1adb1411e16c9c613c38a4c74a29ee608ae;hp=8dff7d0edbf9212313191002395a017b42404fd9;hpb=bd7cc15bddddb727ea02fb44bde2d9f15cd82847;p=babeltrace.git

diff --git a/tests/lib/test_bt_notification_iterator.c b/tests/lib/test_bt_notification_iterator.c
index 8dff7d0e..cbce6b4c 100644
--- a/tests/lib/test_bt_notification_iterator.c
+++ b/tests/lib/test_bt_notification_iterator.c
@@ -38,6 +38,7 @@
 #include <babeltrace/graph/component-sink.h>
 #include <babeltrace/graph/component-source.h>
 #include <babeltrace/graph/component.h>
+#include <babeltrace/graph/connection.h>
 #include <babeltrace/graph/graph.h>
 #include <babeltrace/graph/notification-event.h>
 #include <babeltrace/graph/notification-inactivity.h>
@@ -824,13 +825,13 @@ void sink_port_connected(struct bt_private_component *private_component,
 		bt_private_port_get_private_connection(self_private_port);
 	struct sink_user_data *user_data = bt_private_component_get_user_data(
 		private_component);
+	enum bt_connection_status conn_status;
 
 	assert(user_data);
 	assert(priv_conn);
-	user_data->notif_iter =
-		bt_private_connection_create_notification_iterator(priv_conn,
-			NULL);
-	assert(user_data->notif_iter);
+	conn_status = bt_private_connection_create_notification_iterator(
+		priv_conn, NULL, &user_data->notif_iter);
+	assert(conn_status == 0);
 	bt_put(priv_conn);
 }