lib: rename bt_plugin_create_all_*() -> bt_plugin_find_all_*()
[babeltrace.git] / tests / lib / test_bt_notification_iterator.c
index 609bb4e3244756b0e7134227cc66d47c091b8627..387eaa81f22b2135f1c280bc410a8aad41b3ecd4 100644 (file)
 #include <stdbool.h>
 #include <inttypes.h>
 #include <string.h>
+#include <babeltrace/babeltrace.h>
 #include <babeltrace/assert-internal.h>
-#include <babeltrace/trace-ir/event-class.h>
-#include <babeltrace/trace-ir/event.h>
-#include <babeltrace/trace-ir/field-classes.h>
-#include <babeltrace/trace-ir/fields.h>
-#include <babeltrace/trace-ir/packet.h>
-#include <babeltrace/trace-ir/stream-class.h>
-#include <babeltrace/trace-ir/stream.h>
-#include <babeltrace/trace-ir/trace.h>
-#include <babeltrace/graph/component-class-filter.h>
-#include <babeltrace/graph/component-class-sink.h>
-#include <babeltrace/graph/component-class-source.h>
-#include <babeltrace/graph/component-class.h>
-#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-iterator.h>
-#include <babeltrace/graph/notification-packet.h>
-#include <babeltrace/graph/notification-stream.h>
-#include <babeltrace/graph/output-port-notification-iterator.h>
-#include <babeltrace/graph/port.h>
-#include <babeltrace/graph/private-component-source.h>
-#include <babeltrace/graph/private-component-sink.h>
-#include <babeltrace/graph/private-component.h>
-#include <babeltrace/graph/private-connection.h>
-#include <babeltrace/graph/private-connection-notification-iterator.h>
-#include <babeltrace/graph/private-connection-private-notification-iterator.h>
-#include <babeltrace/graph/private-port.h>
-#include <babeltrace/plugin/plugin.h>
-#include <babeltrace/object.h>
 #include <glib.h>
 
 #include "tap/tap.h"
@@ -78,23 +47,22 @@ enum test_event_type {
 
 struct test_event {
        enum test_event_type type;
-       struct bt_stream *stream;
-       struct bt_packet *packet;
+       const bt_stream *stream;
+       const bt_packet *packet;
 };
 
 static bool debug = false;
 static enum test current_test;
 static GArray *test_events;
-static struct bt_graph *graph;
-static struct bt_private_connection_private_notification_iterator *cur_notif_iter;
-static struct bt_stream_class *src_stream_class;
-static struct bt_event_class *src_event_class;
-static struct bt_stream *src_stream1;
-static struct bt_stream *src_stream2;
-static struct bt_packet *src_stream1_packet1;
-static struct bt_packet *src_stream1_packet2;
-static struct bt_packet *src_stream2_packet1;
-static struct bt_packet *src_stream2_packet2;
+static bt_graph *graph;
+static bt_stream_class *src_stream_class;
+static bt_event_class *src_event_class;
+static bt_stream *src_stream1;
+static bt_stream *src_stream2;
+static bt_packet *src_stream1_packet1;
+static bt_packet *src_stream1_packet2;
+static bt_packet *src_stream2_packet1;
+static bt_packet *src_stream2_packet2;
 
 enum {
        SEQ_END = -1,
@@ -122,7 +90,7 @@ struct src_iter_user_data {
 };
 
 struct sink_user_data {
-       struct bt_notification_iterator *notif_iter;
+       bt_self_component_port_input_notification_iterator *notif_iter;
 };
 
 /*
@@ -273,22 +241,25 @@ bool compare_test_events(const struct test_event *expected_events)
 static
 void init_static_data(void)
 {
-       struct bt_trace *trace;
+       bt_trace_class *trace_class;
+       bt_trace *trace;
 
        /* Test events */
        test_events = g_array_new(FALSE, TRUE, sizeof(struct test_event));
        BT_ASSERT(test_events);
 
-       /* Metadata */
-       trace = bt_trace_create();
+       /* Metadata, streams, and packets*/
+       trace_class = bt_trace_class_create();
        BT_ASSERT(trace);
-       src_stream_class = bt_stream_class_create(trace);
+       src_stream_class = bt_stream_class_create(trace_class);
        BT_ASSERT(src_stream_class);
        src_event_class = bt_event_class_create(src_stream_class);
        BT_ASSERT(src_event_class);
-       src_stream1 = bt_stream_create(src_stream_class);
+       trace = bt_trace_create(trace_class);
+       BT_ASSERT(trace);
+       src_stream1 = bt_stream_create(src_stream_class, trace);
        BT_ASSERT(src_stream1);
-       src_stream2 = bt_stream_create(src_stream_class);
+       src_stream2 = bt_stream_create(src_stream_class, trace);
        BT_ASSERT(src_stream2);
        src_stream1_packet1 = bt_packet_create(src_stream1);
        BT_ASSERT(src_stream1_packet1);
@@ -308,7 +279,8 @@ void init_static_data(void)
                fprintf(stderr, ":: stream 2, packet 2: %p\n", src_stream2_packet2);
        }
 
-       bt_object_put_ref(trace);
+       bt_trace_put_ref(trace);
+       bt_trace_class_put_ref(trace_class);
 }
 
 static
@@ -318,23 +290,22 @@ void fini_static_data(void)
        g_array_free(test_events, TRUE);
 
        /* Metadata */
-       bt_object_put_ref(src_stream_class);
-       bt_object_put_ref(src_event_class);
-       bt_object_put_ref(src_stream1);
-       bt_object_put_ref(src_stream2);
-       bt_object_put_ref(src_stream1_packet1);
-       bt_object_put_ref(src_stream1_packet2);
-       bt_object_put_ref(src_stream2_packet1);
-       bt_object_put_ref(src_stream2_packet2);
+       bt_stream_class_put_ref(src_stream_class);
+       bt_event_class_put_ref(src_event_class);
+       bt_stream_put_ref(src_stream1);
+       bt_stream_put_ref(src_stream2);
+       bt_packet_put_ref(src_stream1_packet1);
+       bt_packet_put_ref(src_stream1_packet2);
+       bt_packet_put_ref(src_stream2_packet1);
+       bt_packet_put_ref(src_stream2_packet2);
 }
 
 static
-void src_iter_finalize(
-               struct bt_private_connection_private_notification_iterator *private_notification_iterator)
+void src_iter_finalize(bt_self_notification_iterator *self_notif_iter)
 {
        struct src_iter_user_data *user_data =
-               bt_private_connection_private_notification_iterator_get_user_data(
-                       private_notification_iterator);
+               bt_self_notification_iterator_get_data(
+                       self_notif_iter);
 
        if (user_data) {
                g_free(user_data);
@@ -342,18 +313,16 @@ void src_iter_finalize(
 }
 
 static
-enum bt_notification_iterator_status src_iter_init(
-               struct bt_private_connection_private_notification_iterator *priv_notif_iter,
-               struct bt_private_port *private_port)
+enum bt_self_notification_iterator_status src_iter_init(
+               bt_self_notification_iterator *self_notif_iter,
+               bt_self_component_source *self_comp,
+               bt_self_component_port_output *self_port)
 {
        struct src_iter_user_data *user_data =
                g_new0(struct src_iter_user_data, 1);
-       int ret;
 
        BT_ASSERT(user_data);
-       ret = bt_private_connection_private_notification_iterator_set_user_data(
-               priv_notif_iter, user_data);
-       BT_ASSERT(ret == 0);
+       bt_self_notification_iterator_set_data(self_notif_iter, user_data);
 
        switch (current_test) {
        case TEST_NO_AUTO_NOTIFS:
@@ -364,63 +333,64 @@ enum bt_notification_iterator_status src_iter_init(
                abort();
        }
 
-       return BT_NOTIFICATION_ITERATOR_STATUS_OK;
+       return BT_SELF_NOTIFICATION_ITERATOR_STATUS_OK;
 }
 
 static
-void src_iter_next_seq_one(struct src_iter_user_data *user_data,
-               struct bt_notification **notif)
+void src_iter_next_seq_one(bt_self_notification_iterator* notif_iter,
+               struct src_iter_user_data *user_data,
+               const bt_notification **notif)
 {
-       struct bt_packet *event_packet = NULL;
+       bt_packet *event_packet = NULL;
 
        switch (user_data->seq[user_data->at]) {
        case SEQ_STREAM1_BEGIN:
-               *notif = bt_notification_stream_begin_create(cur_notif_iter,
-                       src_stream1);
+               *notif = bt_notification_stream_beginning_create(notif_iter,
+                                                                    src_stream1);
                break;
        case SEQ_STREAM2_BEGIN:
-               *notif = bt_notification_stream_begin_create(cur_notif_iter,
-                       src_stream2);
+               *notif = bt_notification_stream_beginning_create(notif_iter,
+                                                            src_stream2);
                break;
        case SEQ_STREAM1_END:
-               *notif = bt_notification_stream_end_create(cur_notif_iter,
-                       src_stream1);
+               *notif = bt_notification_stream_end_create(notif_iter,
+                                                          src_stream1);
                break;
        case SEQ_STREAM2_END:
-               *notif = bt_notification_stream_end_create(cur_notif_iter,
-                       src_stream2);
+               *notif = bt_notification_stream_end_create(notif_iter,
+                                                          src_stream2);
                break;
        case SEQ_STREAM1_PACKET1_BEGIN:
-               *notif = bt_notification_packet_begin_create(cur_notif_iter,
-                       src_stream1_packet1);
+               *notif = bt_notification_packet_beginning_create(notif_iter,
+                                                            src_stream1_packet1);
                break;
        case SEQ_STREAM1_PACKET2_BEGIN:
-               *notif = bt_notification_packet_begin_create(cur_notif_iter,
-                       src_stream1_packet2);
+               *notif = bt_notification_packet_beginning_create(notif_iter,
+                                                            src_stream1_packet2);
                break;
        case SEQ_STREAM2_PACKET1_BEGIN:
-               *notif = bt_notification_packet_begin_create(cur_notif_iter,
-                       src_stream2_packet1);
+               *notif = bt_notification_packet_beginning_create(notif_iter,
+                                                            src_stream2_packet1);
                break;
        case SEQ_STREAM2_PACKET2_BEGIN:
-               *notif = bt_notification_packet_begin_create(cur_notif_iter,
-                       src_stream2_packet2);
+               *notif = bt_notification_packet_beginning_create(notif_iter,
+                                                            src_stream2_packet2);
                break;
        case SEQ_STREAM1_PACKET1_END:
-               *notif = bt_notification_packet_end_create(cur_notif_iter,
-                       src_stream1_packet1);
+               *notif = bt_notification_packet_end_create(notif_iter,
+                                                          src_stream1_packet1);
                break;
        case SEQ_STREAM1_PACKET2_END:
-               *notif = bt_notification_packet_end_create(cur_notif_iter,
-                       src_stream1_packet2);
+               *notif = bt_notification_packet_end_create(notif_iter,
+                                                          src_stream1_packet2);
                break;
        case SEQ_STREAM2_PACKET1_END:
-               *notif = bt_notification_packet_end_create(cur_notif_iter,
-                       src_stream2_packet1);
+               *notif = bt_notification_packet_end_create(notif_iter,
+                                                          src_stream2_packet1);
                break;
        case SEQ_STREAM2_PACKET2_END:
-               *notif = bt_notification_packet_end_create(cur_notif_iter,
-                       src_stream2_packet2);
+               *notif = bt_notification_packet_end_create(notif_iter,
+                                                          src_stream2_packet2);
                break;
        case SEQ_EVENT_STREAM1_PACKET1:
                event_packet = src_stream1_packet1;
@@ -439,8 +409,9 @@ void src_iter_next_seq_one(struct src_iter_user_data *user_data,
        }
 
        if (event_packet) {
-               *notif = bt_notification_event_create(cur_notif_iter,
-                       src_event_class, event_packet);
+               *notif = bt_notification_event_create(notif_iter,
+                                                     src_event_class,
+                                                     event_packet);
        }
 
        BT_ASSERT(*notif);
@@ -448,24 +419,25 @@ void src_iter_next_seq_one(struct src_iter_user_data *user_data,
 }
 
 static
-enum bt_notification_iterator_status src_iter_next_seq(
+enum bt_self_notification_iterator_status src_iter_next_seq(
+               bt_self_notification_iterator *notif_iter,
                struct src_iter_user_data *user_data,
-               bt_notification_array notifs, uint64_t capacity,
+               bt_notification_array_const notifs, uint64_t capacity,
                uint64_t *count)
 {
-       enum bt_notification_iterator_status status =
-               BT_NOTIFICATION_ITERATOR_STATUS_OK;
+       enum bt_self_notification_iterator_status status =
+               BT_SELF_NOTIFICATION_ITERATOR_STATUS_OK;
        uint64_t i = 0;
 
        BT_ASSERT(user_data->seq);
 
        if (user_data->seq[user_data->at] == SEQ_END) {
-               status = BT_NOTIFICATION_ITERATOR_STATUS_END;
+               status = BT_SELF_NOTIFICATION_ITERATOR_STATUS_END;
                goto end;
        }
 
        while (i < capacity && user_data->seq[user_data->at] != SEQ_END) {
-               src_iter_next_seq_one(user_data, &notifs[i]);
+               src_iter_next_seq_one(notif_iter, user_data, &notifs[i]);
                i++;
        }
 
@@ -477,76 +449,76 @@ end:
 }
 
 static
-enum bt_notification_iterator_status src_iter_next(
-               struct bt_private_connection_private_notification_iterator *priv_iterator,
-               bt_notification_array notifs, uint64_t capacity,
+enum bt_self_notification_iterator_status src_iter_next(
+               bt_self_notification_iterator *self_notif_iter,
+               bt_notification_array_const notifs, uint64_t capacity,
                uint64_t *count)
 {
        struct src_iter_user_data *user_data =
-               bt_private_connection_private_notification_iterator_get_user_data(priv_iterator);
+               bt_self_notification_iterator_get_data(self_notif_iter);
 
        BT_ASSERT(user_data);
-       cur_notif_iter = priv_iterator;
-       return src_iter_next_seq(user_data, notifs, capacity, count);
+       return src_iter_next_seq(self_notif_iter, user_data, notifs,
+               capacity, count);
 }
 
 static
-enum bt_component_status src_init(
-               struct bt_private_component *private_component,
-               struct bt_value *params, void *init_method_data)
+enum bt_self_component_status src_init(
+               bt_self_component_source *self_comp,
+               const bt_value *params, void *init_method_data)
 {
        int ret;
 
-       ret = bt_private_component_source_add_output_private_port(
-               private_component, "out", NULL, NULL);
+       ret = bt_self_component_source_add_output_port(
+               self_comp, "out", NULL, NULL);
        BT_ASSERT(ret == 0);
-       return BT_COMPONENT_STATUS_OK;
+       return BT_SELF_COMPONENT_STATUS_OK;
 }
 
 static
-void src_finalize(struct bt_private_component *private_component)
+void src_finalize(bt_self_component_source *self_comp)
 {
 }
 
 static
-void append_test_events_from_notification(struct bt_notification *notification)
+void append_test_events_from_notification(const bt_notification *notification)
 {
        struct test_event test_event = { 0 };
 
        switch (bt_notification_get_type(notification)) {
        case BT_NOTIFICATION_TYPE_EVENT:
        {
-               struct bt_event *event;
+               const bt_event *event;
 
                test_event.type = TEST_EV_TYPE_NOTIF_EVENT;
-               event = bt_notification_event_borrow_event(notification);
+               event = bt_notification_event_borrow_event_const(notification);
                BT_ASSERT(event);
-               test_event.packet = bt_event_borrow_packet(event);
+               test_event.packet = bt_event_borrow_packet_const(event);
                BT_ASSERT(test_event.packet);
                break;
        }
-       case BT_NOTIFICATION_TYPE_STREAM_BEGIN:
+       case BT_NOTIFICATION_TYPE_STREAM_BEGINNING:
                test_event.type = TEST_EV_TYPE_NOTIF_STREAM_BEGIN;
                test_event.stream =
-                       bt_notification_stream_begin_borrow_stream(notification);
+                       bt_notification_stream_beginning_borrow_stream_const(notification);
                BT_ASSERT(test_event.stream);
                break;
        case BT_NOTIFICATION_TYPE_STREAM_END:
                test_event.type = TEST_EV_TYPE_NOTIF_STREAM_END;
                test_event.stream =
-                       bt_notification_stream_end_borrow_stream(notification);
+                       bt_notification_stream_end_borrow_stream_const(notification);
                BT_ASSERT(test_event.stream);
                break;
-       case BT_NOTIFICATION_TYPE_PACKET_BEGIN:
+       case BT_NOTIFICATION_TYPE_PACKET_BEGINNING:
                test_event.type = TEST_EV_TYPE_NOTIF_PACKET_BEGIN;
                test_event.packet =
-                       bt_notification_packet_begin_borrow_packet(notification);
+                       bt_notification_packet_beginning_borrow_packet_const(notification);
                BT_ASSERT(test_event.packet);
                break;
        case BT_NOTIFICATION_TYPE_PACKET_END:
                test_event.type = TEST_EV_TYPE_NOTIF_PACKET_END;
                test_event.packet =
-                       bt_notification_packet_end_borrow_packet(notification);
+                       bt_notification_packet_end_borrow_packet_const(notification);
                BT_ASSERT(test_event.packet);
                break;
        default:
@@ -555,7 +527,8 @@ void append_test_events_from_notification(struct bt_notification *notification)
        }
 
        if (test_event.packet) {
-               test_event.stream = bt_packet_borrow_stream(test_event.packet);
+               test_event.stream = bt_packet_borrow_stream_const(
+                       test_event.packet);
                BT_ASSERT(test_event.stream);
        }
 
@@ -564,11 +537,10 @@ void append_test_events_from_notification(struct bt_notification *notification)
 
 static
 enum bt_notification_iterator_status common_consume(
-               struct bt_notification_iterator *notif_iter,
-               bool is_output_port_notif_iter)
+               void *notif_iter, bool is_output_port_notif_iter)
 {
        enum bt_notification_iterator_status ret;
-       bt_notification_array notifications = NULL;
+       bt_notification_array_const notifications = NULL;
        uint64_t count = 0;
        struct test_event test_event = { 0 };
        uint64_t i;
@@ -576,10 +548,10 @@ enum bt_notification_iterator_status common_consume(
        BT_ASSERT(notif_iter);
 
        if (is_output_port_notif_iter) {
-               ret = bt_output_port_notification_iterator_next(notif_iter,
+               ret = bt_port_output_notification_iterator_next(notif_iter,
                        &notifications, &count);
        } else {
-               ret = bt_private_connection_notification_iterator_next(
+               ret = bt_self_component_port_input_notification_iterator_next(
                        notif_iter, &notifications, &count);
        }
 
@@ -603,7 +575,7 @@ enum bt_notification_iterator_status common_consume(
 
        for (i = 0; i < count; i++) {
                append_test_events_from_notification(notifications[i]);
-               bt_object_put_ref(notifications[i]);
+               bt_notification_put_ref(notifications[i]);
        }
 
 end:
@@ -611,26 +583,29 @@ end:
 }
 
 static
-enum bt_component_status sink_consume(
-               struct bt_private_component *priv_component)
+enum bt_self_component_status sink_consume(
+               bt_self_component_sink *self_comp)
 {
-       enum bt_component_status ret = BT_COMPONENT_STATUS_OK;
+       enum bt_self_component_status ret = BT_SELF_COMPONENT_STATUS_OK;
        struct sink_user_data *user_data =
-               bt_private_component_get_user_data(priv_component);
+               bt_self_component_get_data(
+                       bt_self_component_sink_as_self_component(
+                               self_comp));
        enum bt_notification_iterator_status it_ret;
 
        BT_ASSERT(user_data && user_data->notif_iter);
        it_ret = common_consume(user_data->notif_iter, false);
 
        if (it_ret < 0) {
-               ret = BT_COMPONENT_STATUS_ERROR;
+               ret = BT_SELF_COMPONENT_STATUS_ERROR;
                goto end;
        }
 
        switch (it_ret) {
        case BT_NOTIFICATION_ITERATOR_STATUS_END:
-               ret = BT_COMPONENT_STATUS_END;
-               BT_OBJECT_PUT_REF_AND_RESET(user_data->notif_iter);
+               ret = BT_SELF_COMPONENT_STATUS_END;
+               BT_SELF_COMPONENT_PORT_INPUT_NOTIFICATION_ITERATOR_PUT_REF_AND_RESET(
+                       user_data->notif_iter);
                goto end;
        case BT_NOTIFICATION_ITERATOR_STATUS_AGAIN:
                abort();
@@ -643,62 +618,63 @@ end:
 }
 
 static
-enum bt_component_status sink_port_connected(
-               struct bt_private_component *private_component,
-               struct bt_private_port *self_private_port,
-               struct bt_port *other_port)
+enum bt_self_component_status sink_port_connected(
+               bt_self_component_sink *self_comp,
+               bt_self_component_port_input *self_port,
+               const bt_port_output *other_port)
 {
-       struct bt_private_connection *priv_conn =
-               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;
+       struct sink_user_data *user_data =
+               bt_self_component_get_data(
+                       bt_self_component_sink_as_self_component(
+                               self_comp));
 
        BT_ASSERT(user_data);
-       BT_ASSERT(priv_conn);
-       conn_status = bt_private_connection_create_notification_iterator(
-               priv_conn, &user_data->notif_iter);
-       BT_ASSERT(conn_status == 0);
-       bt_object_put_ref(priv_conn);
-       return BT_COMPONENT_STATUS_OK;
+       user_data->notif_iter =
+               bt_self_component_port_input_notification_iterator_create(
+                       self_port);
+       return BT_SELF_COMPONENT_STATUS_OK;
 }
 
 static
-enum bt_component_status sink_init(
-               struct bt_private_component *private_component,
-               struct bt_value *params, void *init_method_data)
+enum bt_self_component_status sink_init(
+               bt_self_component_sink *self_comp,
+               const bt_value *params, void *init_method_data)
 {
        struct sink_user_data *user_data = g_new0(struct sink_user_data, 1);
        int ret;
 
        BT_ASSERT(user_data);
-       ret = bt_private_component_set_user_data(private_component,
+       bt_self_component_set_data(
+               bt_self_component_sink_as_self_component(self_comp),
                user_data);
+       ret = bt_self_component_sink_add_input_port(
+               self_comp, "in", NULL, NULL);
        BT_ASSERT(ret == 0);
-       ret = bt_private_component_sink_add_input_private_port(
-               private_component, "in", NULL, NULL);
-       BT_ASSERT(ret == 0);
-       return BT_COMPONENT_STATUS_OK;
+       return BT_SELF_COMPONENT_STATUS_OK;
 }
 
 static
-void sink_finalize(struct bt_private_component *private_component)
+void sink_finalize(bt_self_component_sink *self_comp)
 {
-       struct sink_user_data *user_data = bt_private_component_get_user_data(
-               private_component);
+       struct sink_user_data *user_data =
+               bt_self_component_get_data(
+                       bt_self_component_sink_as_self_component(
+                               self_comp));
 
        if (user_data) {
-               bt_object_put_ref(user_data->notif_iter);
+               BT_SELF_COMPONENT_PORT_INPUT_NOTIFICATION_ITERATOR_PUT_REF_AND_RESET(
+                       user_data->notif_iter);
                g_free(user_data);
        }
 }
 
 static
-void create_source_sink(struct bt_graph *graph, struct bt_component **source,
-               struct bt_component **sink)
+void create_source_sink(bt_graph *graph,
+               const bt_component_source **source,
+               const bt_component_sink **sink)
 {
-       struct bt_component_class *src_comp_class;
-       struct bt_component_class *sink_comp_class;
+       bt_component_class_source *src_comp_class;
+       bt_component_class_sink *sink_comp_class;
        int ret;
 
        /* Create source component */
@@ -706,11 +682,11 @@ void create_source_sink(struct bt_graph *graph, struct bt_component **source,
                src_comp_class = bt_component_class_source_create("src",
                        src_iter_next);
                BT_ASSERT(src_comp_class);
-               ret = bt_component_class_set_init_method(src_comp_class,
-                       src_init);
+               ret = bt_component_class_source_set_init_method(
+                       src_comp_class, src_init);
                BT_ASSERT(ret == 0);
-               ret = bt_component_class_set_finalize_method(src_comp_class,
-                       src_finalize);
+               ret = bt_component_class_source_set_finalize_method(
+                       src_comp_class, src_finalize);
                BT_ASSERT(ret == 0);
                ret = bt_component_class_source_set_notification_iterator_init_method(
                        src_comp_class, src_iter_init);
@@ -718,10 +694,10 @@ void create_source_sink(struct bt_graph *graph, struct bt_component **source,
                ret = bt_component_class_source_set_notification_iterator_finalize_method(
                        src_comp_class, src_iter_finalize);
                BT_ASSERT(ret == 0);
-               ret = bt_graph_add_component(graph, src_comp_class, "source",
-                       NULL, source);
+               ret = bt_graph_add_source_component(graph,
+                       src_comp_class, "source", NULL, source);
                BT_ASSERT(ret == 0);
-               bt_object_put_ref(src_comp_class);
+               bt_component_class_source_put_ref(src_comp_class);
        }
 
        /* Create sink component */
@@ -729,18 +705,19 @@ void create_source_sink(struct bt_graph *graph, struct bt_component **source,
                sink_comp_class = bt_component_class_sink_create("sink",
                        sink_consume);
                BT_ASSERT(sink_comp_class);
-               ret = bt_component_class_set_init_method(sink_comp_class,
-                       sink_init);
+               ret = bt_component_class_sink_set_init_method(
+                       sink_comp_class, sink_init);
                BT_ASSERT(ret == 0);
-               ret = bt_component_class_set_finalize_method(sink_comp_class,
-                       sink_finalize);
-               ret = bt_component_class_set_port_connected_method(
+               ret = bt_component_class_sink_set_finalize_method(
+                       sink_comp_class, sink_finalize);
+               ret = bt_component_class_sink_set_input_port_connected_method(
                        sink_comp_class, sink_port_connected);
                BT_ASSERT(ret == 0);
-               ret = bt_graph_add_component(graph, sink_comp_class, "sink",
-                       NULL, sink);
+               ret = bt_graph_add_sink_component(graph,
+                       sink_comp_class,
+                       "sink", NULL, sink);
                BT_ASSERT(ret == 0);
-               bt_object_put_ref(sink_comp_class);
+               bt_component_class_sink_put_ref(sink_comp_class);
        }
 }
 
@@ -748,10 +725,10 @@ static
 void do_std_test(enum test test, const char *name,
                const struct test_event *expected_test_events)
 {
-       struct bt_component *src_comp;
-       struct bt_component *sink_comp;
-       struct bt_port *upstream_port;
-       struct bt_port *downstream_port;
+       const bt_component_source *src_comp;
+       const bt_component_sink *sink_comp;
+       const bt_port_output *upstream_port;
+       const bt_port_input *downstream_port;
        enum bt_graph_status graph_status = BT_GRAPH_STATUS_OK;
 
        clear_test_events();
@@ -763,14 +740,15 @@ void do_std_test(enum test test, const char *name,
        create_source_sink(graph, &src_comp, &sink_comp);
 
        /* Connect source to sink */
-       upstream_port = bt_component_source_get_output_port_by_name(src_comp, "out");
+       upstream_port =
+               bt_component_source_borrow_output_port_by_name_const(
+                       src_comp, "out");
        BT_ASSERT(upstream_port);
-       downstream_port = bt_component_sink_get_input_port_by_name(sink_comp, "in");
+       downstream_port = bt_component_sink_borrow_input_port_by_name_const(
+               sink_comp, "in");
        BT_ASSERT(downstream_port);
        graph_status = bt_graph_connect_ports(graph, upstream_port,
                downstream_port, NULL);
-       bt_object_put_ref(upstream_port);
-       bt_object_put_ref(downstream_port);
 
        /* Run the graph until the end */
        while (graph_status == BT_GRAPH_STATUS_OK ||
@@ -778,7 +756,8 @@ void do_std_test(enum test test, const char *name,
                graph_status = bt_graph_run(graph);
        }
 
-       ok(graph_status == BT_GRAPH_STATUS_END, "graph finishes without any error");
+       ok(graph_status == BT_GRAPH_STATUS_END,
+               "graph finishes without any error");
 
        /* Compare the resulting test events */
        if (expected_test_events) {
@@ -786,9 +765,9 @@ void do_std_test(enum test test, const char *name,
                        "the produced sequence of test events is the expected one");
        }
 
-       bt_object_put_ref(src_comp);
-       bt_object_put_ref(sink_comp);
-       BT_OBJECT_PUT_REF_AND_RESET(graph);
+       bt_component_source_put_ref(src_comp);
+       bt_component_sink_put_ref(sink_comp);
+       BT_GRAPH_PUT_REF_AND_RESET(graph);
 }
 
 static
@@ -842,11 +821,11 @@ void test_output_port_notification_iterator(void)
                { .type = TEST_EV_TYPE_END, },
                { .type = TEST_EV_TYPE_SENTINEL, },
        };
-       struct bt_component *src_comp;
-       struct bt_notification_iterator *notif_iter;
+       const bt_component_source *src_comp;
+       bt_port_output_notification_iterator *notif_iter;
        enum bt_notification_iterator_status iter_status =
                BT_NOTIFICATION_ITERATOR_STATUS_OK;
-       struct bt_port *upstream_port;
+       const bt_port_output *upstream_port;
 
        clear_test_events();
        current_test = TEST_OUTPUT_PORT_NOTIFICATION_ITERATOR;
@@ -857,11 +836,11 @@ void test_output_port_notification_iterator(void)
        create_source_sink(graph, &src_comp, NULL);
 
        /* Create notification iterator on source's output port */
-       upstream_port = bt_component_source_get_output_port_by_name(src_comp, "out");
-       notif_iter = bt_output_port_notification_iterator_create(upstream_port,
-               NULL);
-       ok(notif_iter, "bt_output_port_notification_iterator_create() succeeds");
-       bt_object_put_ref(upstream_port);
+       upstream_port = bt_component_source_borrow_output_port_by_name_const(src_comp,
+                                                                            "out");
+       notif_iter = bt_port_output_notification_iterator_create(graph,
+               upstream_port);
+       ok(notif_iter, "bt_private_output_port_notification_iterator_create() succeeds");
 
        /* Consume the notification iterator */
        while (iter_status == BT_NOTIFICATION_ITERATOR_STATUS_OK) {
@@ -875,9 +854,9 @@ void test_output_port_notification_iterator(void)
        ok(compare_test_events(expected_test_events),
                "the produced sequence of test events is the expected one");
 
-       bt_object_put_ref(src_comp);
-       BT_OBJECT_PUT_REF_AND_RESET(graph);
-       bt_object_put_ref(notif_iter);
+       bt_component_source_put_ref(src_comp);
+       BT_GRAPH_PUT_REF_AND_RESET(graph);
+       bt_port_output_notification_iterator_put_ref(notif_iter);
 }
 
 #define DEBUG_ENV_VAR  "TEST_BT_NOTIFICATION_ITERATOR_DEBUG"
This page took 0.051597 seconds and 4 git commands to generate.