lib: use object pool for event and packet notifications

[babeltrace.git] / tests / lib / test_graph_topo.c

diff --git a/tests/lib/test_graph_topo.c b/tests/lib/test_graph_topo.c
index 2efac144433798df9a5a646717925203935685ac..185c6c2259c601c0e255acc1de4bb37009a0ca4b 100644 (file)
--- a/tests/lib/test_graph_topo.c
+++ b/tests/lib/test_graph_topo.c
@@ -17,6 +17,7 @@
  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  */
 
+#include <babeltrace/ref.h>
 #include <babeltrace/graph/component-class.h>
 #include <babeltrace/graph/component-class-source.h>
 #include <babeltrace/graph/component-class-sink.h>
@@ -280,14 +281,14 @@ size_t event_pos(struct event *event)
                }
        }
 
-       return -1ULL;
+       return SIZE_MAX;
 }
 
 static
-struct bt_notification_iterator_next_return src_iter_next(
-               struct bt_private_notification_iterator *priv_iterator)
+struct bt_notification_iterator_next_method_return src_iter_next(
+               struct bt_private_connection_private_notification_iterator *priv_iterator)
 {
-       struct bt_notification_iterator_next_return ret = {
+       struct bt_notification_iterator_next_method_return ret = {
                .status = BT_NOTIFICATION_ITERATOR_STATUS_ERROR,
        };
 
@@ -303,14 +304,12 @@ enum bt_component_status accept_port_connection(
        struct event event = {
                .type = COMP_ACCEPT_PORT_CONNECTION,
                .data.comp_accept_port_connection = {
-                       .comp = bt_component_from_private_component(private_component),
-                       .self_port = bt_port_from_private_port(self_private_port),
+                       .comp = bt_component_borrow_from_private(private_component),
+                       .self_port = bt_port_borrow_from_private(self_private_port),
                        .other_port = other_port,
                },
        };
 
-       bt_put(event.data.comp_accept_port_connection.comp);
-       bt_put(event.data.comp_accept_port_connection.self_port);
        append_event(&event);
        return BT_COMPONENT_STATUS_OK;
 }
@@ -325,14 +324,12 @@ void src_port_connected(struct bt_private_component *private_component,
        struct event event = {
                .type = COMP_PORT_CONNECTED,
                .data.comp_port_connected = {
-                       .comp = bt_component_from_private_component(private_component),
-                       .self_port = bt_port_from_private_port(self_private_port),
+                       .comp = bt_component_borrow_from_private(private_component),
+                       .self_port = bt_port_borrow_from_private(self_private_port),
                        .other_port = other_port,
                },
        };
 
-       bt_put(event.data.comp_port_connected.comp);
-       bt_put(event.data.comp_port_connected.self_port);
        append_event(&event);
 
        switch (current_test) {
@@ -354,13 +351,11 @@ void src_port_disconnected(struct bt_private_component *private_component,
        struct event event = {
                .type = COMP_PORT_DISCONNECTED,
                .data.comp_port_disconnected = {
-                       .comp = bt_component_from_private_component(private_component),
-                       .port = bt_port_from_private_port(private_port),
+                       .comp = bt_component_borrow_from_private(private_component),
+                       .port = bt_port_borrow_from_private(private_port),
                },
        };
 
-       bt_put(event.data.comp_port_disconnected.comp);
-       bt_put(event.data.comp_port_disconnected.port);
        append_event(&event);
 
        switch (current_test) {
@@ -416,14 +411,12 @@ void sink_port_connected(struct bt_private_component *private_component,
        struct event event = {
                .type = COMP_PORT_CONNECTED,
                .data.comp_port_connected = {
-                       .comp = bt_component_from_private_component(private_component),
-                       .self_port = bt_port_from_private_port(self_private_port),
+                       .comp = bt_component_borrow_from_private(private_component),
+                       .self_port = bt_port_borrow_from_private(self_private_port),
                        .other_port = other_port,
                },
        };
 
-       bt_put(event.data.comp_port_connected.comp);
-       bt_put(event.data.comp_port_connected.self_port);
        append_event(&event);
 }
 
@@ -434,13 +427,11 @@ void sink_port_disconnected(struct bt_private_component *private_component,
        struct event event = {
                .type = COMP_PORT_DISCONNECTED,
                .data.comp_port_disconnected = {
-                       .comp = bt_component_from_private_component(private_component),
-                       .port = bt_port_from_private_port(private_port),
+                       .comp = bt_component_borrow_from_private(private_component),
+                       .port = bt_port_borrow_from_private(private_port),
                },
        };
 
-       bt_put(event.data.comp_port_disconnected.comp);
-       bt_put(event.data.comp_port_disconnected.port);
        append_event(&event);
 }
 
@@ -618,16 +609,17 @@ struct bt_graph *create_graph(void)
        int ret;
 
        assert(graph);
-       ret = bt_graph_add_port_added_listener(graph, graph_port_added, NULL);
+       ret = bt_graph_add_port_added_listener(graph, graph_port_added, NULL,
+               NULL);
        assert(ret >= 0);
        ret = bt_graph_add_port_removed_listener(graph, graph_port_removed,
-               NULL);
+               NULL, NULL);
        assert(ret >= 0);
-       ret = bt_graph_add_ports_connected_listener(graph, graph_ports_connected,
-               NULL);
+       ret = bt_graph_add_ports_connected_listener(graph,
+               graph_ports_connected, NULL, NULL);
        assert(ret >= 0);
        ret = bt_graph_add_ports_disconnected_listener(graph,
-               graph_ports_disconnected, NULL);
+               graph_ports_disconnected, NULL, NULL);
        assert(ret >= 0);
        return graph;
 }