X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=tests%2Flib%2Ftest_bt_notification_iterator.c;h=3b1b15279a3b0d1edce33c32ae9151805b4f65f9;hb=40f4ba76dd6f9508ca51b6220eaed57632281a07;hp=f076af51095e93dd808e873ce877e783bc24428d;hpb=ad8474550bfc1968366fe45818c046fa5bc15581;p=babeltrace.git diff --git a/tests/lib/test_bt_notification_iterator.c b/tests/lib/test_bt_notification_iterator.c index f076af51..3b1b1527 100644 --- a/tests/lib/test_bt_notification_iterator.c +++ b/tests/lib/test_bt_notification_iterator.c @@ -21,41 +21,8 @@ #include #include #include +#include #include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include #include #include "tap/tap.h" @@ -70,7 +37,6 @@ enum test { enum test_event_type { TEST_EV_TYPE_NOTIF_UNEXPECTED, TEST_EV_TYPE_NOTIF_EVENT, - TEST_EV_TYPE_NOTIF_INACTIVITY, TEST_EV_TYPE_NOTIF_STREAM_BEGIN, TEST_EV_TYPE_NOTIF_PACKET_BEGIN, TEST_EV_TYPE_NOTIF_PACKET_END, @@ -81,15 +47,14 @@ enum test_event_type { struct test_event { enum test_event_type type; - struct bt_stream *stream; - struct bt_packet *packet; + const struct bt_stream *stream; + const struct bt_packet *packet; }; static bool debug = false; static enum test current_test; static GArray *test_events; -static struct bt_graph *graph; -static struct bt_clock_class_priority_map *src_empty_cc_prio_map; +static struct bt_private_graph *graph; static struct bt_stream_class *src_stream_class; static struct bt_event_class *src_event_class; static struct bt_stream *src_stream1; @@ -98,6 +63,12 @@ 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 const struct bt_stream *pub_src_stream1; +static const struct bt_stream *pub_src_stream2; +static const struct bt_packet *pub_src_stream1_packet1; +static const struct bt_packet *pub_src_stream1_packet2; +static const struct bt_packet *pub_src_stream2_packet1; +static const struct bt_packet *pub_src_stream2_packet2; enum { SEQ_END = -1, @@ -117,7 +88,6 @@ enum { SEQ_EVENT_STREAM1_PACKET2 = -15, SEQ_EVENT_STREAM2_PACKET1 = -16, SEQ_EVENT_STREAM2_PACKET2 = -17, - SEQ_INACTIVITY = -18, }; struct src_iter_user_data { @@ -126,7 +96,7 @@ struct src_iter_user_data { }; struct sink_user_data { - struct bt_notification_iterator *notif_iter; + void *notif_iter; }; /* @@ -171,9 +141,6 @@ void print_test_event(FILE *fp, const struct test_event *event) case TEST_EV_TYPE_NOTIF_EVENT: fprintf(fp, "TEST_EV_TYPE_NOTIF_EVENT"); break; - case TEST_EV_TYPE_NOTIF_INACTIVITY: - fprintf(fp, "TEST_EV_TYPE_NOTIF_INACTIVITY"); - break; case TEST_EV_TYPE_NOTIF_STREAM_BEGIN: fprintf(fp, "TEST_EV_TYPE_NOTIF_STREAM_BEGIN"); break; @@ -280,58 +247,37 @@ bool compare_test_events(const struct test_event *expected_events) static void init_static_data(void) { - int ret; struct bt_trace *trace; - struct bt_field_type *empty_struct_ft; /* Test events */ test_events = g_array_new(FALSE, TRUE, sizeof(struct test_event)); BT_ASSERT(test_events); /* Metadata */ - empty_struct_ft = bt_field_type_structure_create(); - BT_ASSERT(empty_struct_ft); trace = bt_trace_create(); BT_ASSERT(trace); - ret = bt_trace_set_packet_header_field_type(trace, empty_struct_ft); - BT_ASSERT(ret == 0); - src_empty_cc_prio_map = bt_clock_class_priority_map_create(); - BT_ASSERT(src_empty_cc_prio_map); - src_stream_class = bt_stream_class_create("my-stream-class"); + src_stream_class = bt_stream_class_create(trace); BT_ASSERT(src_stream_class); - ret = bt_stream_class_set_packet_context_field_type(src_stream_class, - empty_struct_ft); - BT_ASSERT(ret == 0); - ret = bt_stream_class_set_event_header_field_type(src_stream_class, - empty_struct_ft); - BT_ASSERT(ret == 0); - ret = bt_stream_class_set_event_context_field_type(src_stream_class, - empty_struct_ft); - BT_ASSERT(ret == 0); - src_event_class = bt_event_class_create("my-event-class"); - ret = bt_event_class_set_context_field_type(src_event_class, - empty_struct_ft); - BT_ASSERT(ret == 0); - ret = bt_event_class_set_payload_field_type(src_event_class, - empty_struct_ft); - BT_ASSERT(ret == 0); - ret = bt_stream_class_add_event_class(src_stream_class, - src_event_class); - BT_ASSERT(ret == 0); - ret = bt_trace_add_stream_class(trace, src_stream_class); - BT_ASSERT(ret == 0); - src_stream1 = bt_stream_create(src_stream_class, "stream-1", 0); + src_event_class = bt_event_class_create(src_stream_class); + BT_ASSERT(src_event_class); + src_stream1 = bt_stream_create(src_stream_class); BT_ASSERT(src_stream1); - src_stream2 = bt_stream_create(src_stream_class, "stream-2", 1); + pub_src_stream1 = src_stream1; + src_stream2 = bt_stream_create(src_stream_class); BT_ASSERT(src_stream2); + pub_src_stream2 = src_stream2; src_stream1_packet1 = bt_packet_create(src_stream1); BT_ASSERT(src_stream1_packet1); + pub_src_stream1_packet1 = src_stream1_packet1; src_stream1_packet2 = bt_packet_create(src_stream1); BT_ASSERT(src_stream1_packet2); + pub_src_stream1_packet2 = src_stream1_packet2; src_stream2_packet1 = bt_packet_create(src_stream2); BT_ASSERT(src_stream2_packet1); + pub_src_stream2_packet1 = src_stream2_packet1; src_stream2_packet2 = bt_packet_create(src_stream2); BT_ASSERT(src_stream2_packet2); + pub_src_stream2_packet2 = src_stream2_packet2; if (debug) { fprintf(stderr, ":: stream 1: %p\n", src_stream1); @@ -342,8 +288,7 @@ void init_static_data(void) fprintf(stderr, ":: stream 2, packet 2: %p\n", src_stream2_packet2); } - bt_put(trace); - bt_put(empty_struct_ft); + bt_object_put_ref(trace); } static @@ -353,24 +298,22 @@ void fini_static_data(void) g_array_free(test_events, TRUE); /* Metadata */ - bt_put(src_empty_cc_prio_map); - bt_put(src_stream_class); - bt_put(src_event_class); - bt_put(src_stream1); - bt_put(src_stream2); - bt_put(src_stream1_packet1); - bt_put(src_stream1_packet2); - bt_put(src_stream2_packet1); - bt_put(src_stream2_packet2); + 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); } static -void src_iter_finalize( - struct bt_private_connection_private_notification_iterator *private_notification_iterator) +void src_iter_finalize(struct 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); @@ -378,18 +321,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( + struct bt_self_notification_iterator *self_notif_iter, + struct bt_self_component_source *self_comp, + struct 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: @@ -400,100 +341,76 @@ enum bt_notification_iterator_status src_iter_init( abort(); } - return BT_NOTIFICATION_ITERATOR_STATUS_OK; + return BT_SELF_NOTIFICATION_ITERATOR_STATUS_OK; } static -struct bt_notification_iterator_next_method_return src_iter_next_seq( - struct src_iter_user_data *user_data) +void src_iter_next_seq_one(struct bt_self_notification_iterator* notif_iter, + struct src_iter_user_data *user_data, + struct bt_notification **notif) { - struct bt_notification_iterator_next_method_return next_return = { - .status = BT_NOTIFICATION_ITERATOR_STATUS_OK, - }; - int64_t cur_ts_ns; struct bt_packet *event_packet = NULL; - BT_ASSERT(user_data->seq); - cur_ts_ns = user_data->seq[user_data->at]; - - switch (cur_ts_ns) { - case SEQ_END: - next_return.status = - BT_NOTIFICATION_ITERATOR_STATUS_END; - break; - case SEQ_INACTIVITY: - next_return.notification = - bt_notification_inactivity_create(graph, - src_empty_cc_prio_map); - BT_ASSERT(next_return.notification); - break; + switch (user_data->seq[user_data->at]) { case SEQ_STREAM1_BEGIN: - next_return.notification = - bt_notification_stream_begin_create(graph, src_stream1); - BT_ASSERT(next_return.notification); + *notif = bt_private_notification_as_notification( + bt_private_notification_stream_begin_create( + notif_iter, src_stream1)); break; case SEQ_STREAM2_BEGIN: - next_return.notification = - bt_notification_stream_begin_create(graph, src_stream2); - BT_ASSERT(next_return.notification); + *notif = bt_private_notification_as_notification( + bt_private_notification_stream_begin_create( + notif_iter, src_stream2)); break; case SEQ_STREAM1_END: - next_return.notification = - bt_notification_stream_end_create(graph, src_stream1); - BT_ASSERT(next_return.notification); + *notif = bt_private_notification_as_notification( + bt_private_notification_stream_end_create( + notif_iter, src_stream1)); break; case SEQ_STREAM2_END: - next_return.notification = - bt_notification_stream_end_create(graph, src_stream2); - BT_ASSERT(next_return.notification); + *notif = bt_private_notification_as_notification( + bt_private_notification_stream_end_create( + notif_iter, src_stream2)); break; case SEQ_STREAM1_PACKET1_BEGIN: - next_return.notification = - bt_notification_packet_begin_create(graph, - src_stream1_packet1); - BT_ASSERT(next_return.notification); + *notif = bt_private_notification_as_notification( + bt_private_notification_packet_begin_create( + notif_iter, src_stream1_packet1)); break; case SEQ_STREAM1_PACKET2_BEGIN: - next_return.notification = - bt_notification_packet_begin_create(graph, - src_stream1_packet2); - BT_ASSERT(next_return.notification); + *notif = bt_private_notification_as_notification( + bt_private_notification_packet_begin_create( + notif_iter, src_stream1_packet2)); break; case SEQ_STREAM2_PACKET1_BEGIN: - next_return.notification = - bt_notification_packet_begin_create(graph, - src_stream2_packet1); - BT_ASSERT(next_return.notification); + *notif = bt_private_notification_as_notification( + bt_private_notification_packet_begin_create( + notif_iter, src_stream2_packet1)); break; case SEQ_STREAM2_PACKET2_BEGIN: - next_return.notification = - bt_notification_packet_begin_create(graph, - src_stream2_packet2); - BT_ASSERT(next_return.notification); + *notif = bt_private_notification_as_notification( + bt_private_notification_packet_begin_create( + notif_iter, src_stream2_packet2)); break; case SEQ_STREAM1_PACKET1_END: - next_return.notification = - bt_notification_packet_end_create(graph, - src_stream1_packet1); - BT_ASSERT(next_return.notification); + *notif = bt_private_notification_as_notification( + bt_private_notification_packet_end_create( + notif_iter, src_stream1_packet1)); break; case SEQ_STREAM1_PACKET2_END: - next_return.notification = - bt_notification_packet_end_create(graph, - src_stream1_packet2); - BT_ASSERT(next_return.notification); + *notif = bt_private_notification_as_notification( + bt_private_notification_packet_end_create( + notif_iter, src_stream1_packet2)); break; case SEQ_STREAM2_PACKET1_END: - next_return.notification = - bt_notification_packet_end_create(graph, - src_stream2_packet1); - BT_ASSERT(next_return.notification); + *notif = bt_private_notification_as_notification( + bt_private_notification_packet_end_create( + notif_iter, src_stream2_packet1)); break; case SEQ_STREAM2_PACKET2_END: - next_return.notification = - bt_notification_packet_end_create(graph, - src_stream2_packet2); - BT_ASSERT(next_return.notification); + *notif = bt_private_notification_as_notification( + bt_private_notification_packet_end_create( + notif_iter, src_stream2_packet2)); break; case SEQ_EVENT_STREAM1_PACKET1: event_packet = src_stream1_packet1; @@ -512,108 +429,99 @@ struct bt_notification_iterator_next_method_return src_iter_next_seq( } if (event_packet) { - next_return.notification = - bt_notification_event_create(graph, src_event_class, - event_packet, src_empty_cc_prio_map); - BT_ASSERT(next_return.notification); + *notif = bt_private_notification_as_notification( + bt_private_notification_event_create( + notif_iter, src_event_class, event_packet)); + } + + BT_ASSERT(*notif); + user_data->at++; +} + +static +enum bt_self_notification_iterator_status src_iter_next_seq( + struct bt_self_notification_iterator *notif_iter, + struct src_iter_user_data *user_data, + bt_notification_array notifs, uint64_t capacity, + uint64_t *count) +{ + 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_SELF_NOTIFICATION_ITERATOR_STATUS_END; + goto end; } - if (next_return.status != BT_NOTIFICATION_ITERATOR_STATUS_END) { - user_data->at++; + while (i < capacity && user_data->seq[user_data->at] != SEQ_END) { + src_iter_next_seq_one(notif_iter, user_data, ¬ifs[i]); + i++; } - return next_return; + BT_ASSERT(i > 0 && i <= capacity); + *count = i; + +end: + return status; } static -struct bt_notification_iterator_next_method_return src_iter_next( - struct bt_private_connection_private_notification_iterator *priv_iterator) +enum bt_self_notification_iterator_status src_iter_next( + struct bt_self_notification_iterator *self_notif_iter, + bt_notification_array notifs, uint64_t capacity, + uint64_t *count) { - struct bt_notification_iterator_next_method_return next_return = { - .status = BT_NOTIFICATION_ITERATOR_STATUS_OK, - .notification = NULL, - }; 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); - next_return = src_iter_next_seq(user_data); - return next_return; + 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( + struct bt_self_component_source *self_comp, + const struct 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(struct bt_self_component_source *self_comp) { } static -enum bt_notification_iterator_status common_consume( - struct bt_notification_iterator *notif_iter, - bool is_output_port_notif_iter) +void append_test_events_from_notification(struct bt_notification *notification) { - enum bt_notification_iterator_status ret; - struct bt_notification *notification = NULL; struct test_event test_event = { 0 }; - bool do_append_test_event = true; - BT_ASSERT(notif_iter); - - if (is_output_port_notif_iter) { - ret = bt_output_port_notification_iterator_next(notif_iter, - ¬ification); - } else { - ret = bt_private_connection_notification_iterator_next( - notif_iter, ¬ification); - } - - if (ret < 0) { - do_append_test_event = false; - goto end; - } - - switch (ret) { - case BT_NOTIFICATION_ITERATOR_STATUS_END: - test_event.type = TEST_EV_TYPE_END; - goto end; - case BT_NOTIFICATION_ITERATOR_STATUS_AGAIN: - abort(); - default: - break; - } - - BT_ASSERT(notification); switch (bt_notification_get_type(notification)) { case BT_NOTIFICATION_TYPE_EVENT: { - struct bt_event *event; + const struct bt_event *event; test_event.type = TEST_EV_TYPE_NOTIF_EVENT; event = bt_notification_event_borrow_event(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_INACTIVITY: - test_event.type = TEST_EV_TYPE_NOTIF_INACTIVITY; - break; case BT_NOTIFICATION_TYPE_STREAM_BEGIN: test_event.type = TEST_EV_TYPE_NOTIF_STREAM_BEGIN; test_event.stream = - bt_notification_stream_begin_borrow_stream(notification); + bt_notification_stream_begin_borrow_stream( + notification); BT_ASSERT(test_event.stream); break; case BT_NOTIFICATION_TYPE_STREAM_END: @@ -625,7 +533,8 @@ enum bt_notification_iterator_status common_consume( case BT_NOTIFICATION_TYPE_PACKET_BEGIN: test_event.type = TEST_EV_TYPE_NOTIF_PACKET_BEGIN; test_event.packet = - bt_notification_packet_begin_borrow_packet(notification); + bt_notification_packet_begin_borrow_packet( + notification); BT_ASSERT(test_event.packet); break; case BT_NOTIFICATION_TYPE_PACKET_END: @@ -640,40 +549,84 @@ enum bt_notification_iterator_status common_consume( } 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); } -end: - if (do_append_test_event) { + append_test_event(&test_event); +} + +static +enum bt_notification_iterator_status common_consume( + void *notif_iter, bool is_output_port_notif_iter) +{ + enum bt_notification_iterator_status ret; + bt_notification_array notifications = NULL; + uint64_t count = 0; + struct test_event test_event = { 0 }; + uint64_t i; + + BT_ASSERT(notif_iter); + + if (is_output_port_notif_iter) { + ret = bt_port_output_notification_iterator_next(notif_iter, + ¬ifications, &count); + } else { + ret = bt_self_component_port_input_notification_iterator_next( + notif_iter, ¬ifications, &count); + } + + if (ret < 0) { + goto end; + } + + switch (ret) { + case BT_NOTIFICATION_ITERATOR_STATUS_END: + test_event.type = TEST_EV_TYPE_END; append_test_event(&test_event); + goto end; + case BT_NOTIFICATION_ITERATOR_STATUS_AGAIN: + abort(); + default: + break; } - bt_put(notification); + BT_ASSERT(notifications); + BT_ASSERT(count > 0); + + for (i = 0; i < count; i++) { + append_test_events_from_notification(notifications[i]); + bt_object_put_ref(notifications[i]); + } + +end: return ret; } static -enum bt_component_status sink_consume( - struct bt_private_component *priv_component) +enum bt_self_component_status sink_consume( + struct 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_PUT(user_data->notif_iter); + ret = BT_SELF_COMPONENT_STATUS_END; + BT_OBJECT_PUT_REF_AND_RESET(user_data->notif_iter); goto end; case BT_NOTIFICATION_ITERATOR_STATUS_AGAIN: abort(); @@ -686,102 +639,107 @@ end: } static -void 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( + struct bt_self_component_sink *self_comp, + struct bt_self_component_port_input *self_port, + struct 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_put(priv_conn); + 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( + struct bt_self_component_sink *self_comp, + const struct 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(struct 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_put(user_data->notif_iter); + bt_object_put_ref(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(struct bt_private_graph *graph, + struct bt_component_source **source, + struct bt_component_sink **sink) { - struct bt_component_class *src_comp_class; - struct bt_component_class *sink_comp_class; + struct bt_private_component_class_source *src_comp_class; + struct bt_private_component_class_sink *sink_comp_class; int ret; /* Create source component */ if (source) { - src_comp_class = bt_component_class_source_create("src", + src_comp_class = bt_private_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_private_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_private_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( + ret = bt_private_component_class_source_set_notification_iterator_init_method( src_comp_class, src_iter_init); BT_ASSERT(ret == 0); - ret = bt_component_class_source_set_notification_iterator_finalize_method( + ret = bt_private_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_private_graph_add_source_component(graph, + bt_private_component_class_source_as_component_class_source( + src_comp_class), "source", NULL, source); BT_ASSERT(ret == 0); - bt_put(src_comp_class); + bt_object_put_ref(src_comp_class); } /* Create sink component */ if (sink) { - sink_comp_class = bt_component_class_sink_create("sink", + sink_comp_class = bt_private_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_private_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_private_component_class_sink_set_finalize_method( + sink_comp_class, sink_finalize); + ret = bt_private_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_private_graph_add_sink_component(graph, + bt_private_component_class_sink_as_component_class_sink( + sink_comp_class), + "sink", NULL, sink); BT_ASSERT(ret == 0); - bt_put(sink_comp_class); + bt_object_put_ref(sink_comp_class); } } @@ -789,37 +747,38 @@ 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; + struct bt_component_source *src_comp; + struct bt_component_sink *sink_comp; + struct bt_port_output *upstream_port; + struct bt_port_input *downstream_port; enum bt_graph_status graph_status = BT_GRAPH_STATUS_OK; clear_test_events(); current_test = test; diag("test: %s", name); BT_ASSERT(!graph); - graph = bt_graph_create(); + graph = bt_private_graph_create(); BT_ASSERT(graph); 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( + 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( + sink_comp, "in"); BT_ASSERT(downstream_port); - graph_status = bt_graph_connect_ports(graph, upstream_port, + graph_status = bt_private_graph_connect_ports(graph, upstream_port, downstream_port, NULL); - bt_put(upstream_port); - bt_put(downstream_port); /* Run the graph until the end */ while (graph_status == BT_GRAPH_STATUS_OK || graph_status == BT_GRAPH_STATUS_AGAIN) { - graph_status = bt_graph_run(graph); + graph_status = bt_private_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) { @@ -827,31 +786,31 @@ void do_std_test(enum test test, const char *name, "the produced sequence of test events is the expected one"); } - bt_put(src_comp); - bt_put(sink_comp); - BT_PUT(graph); + bt_object_put_ref(src_comp); + bt_object_put_ref(sink_comp); + BT_OBJECT_PUT_REF_AND_RESET(graph); } static void test_no_auto_notifs(void) { const struct test_event expected_test_events[] = { - { .type = TEST_EV_TYPE_NOTIF_STREAM_BEGIN, .stream = src_stream1, .packet = NULL, }, - { .type = TEST_EV_TYPE_NOTIF_PACKET_BEGIN, .stream = src_stream1, .packet = src_stream1_packet1, }, - { .type = TEST_EV_TYPE_NOTIF_EVENT, .stream = src_stream1, .packet = src_stream1_packet1, }, - { .type = TEST_EV_TYPE_NOTIF_EVENT, .stream = src_stream1, .packet = src_stream1_packet1, }, - { .type = TEST_EV_TYPE_NOTIF_STREAM_BEGIN, .stream = src_stream2, .packet = NULL, }, - { .type = TEST_EV_TYPE_NOTIF_EVENT, .stream = src_stream1, .packet = src_stream1_packet1, }, - { .type = TEST_EV_TYPE_NOTIF_PACKET_BEGIN, .stream = src_stream2, .packet = src_stream2_packet2, }, - { .type = TEST_EV_TYPE_NOTIF_EVENT, .stream = src_stream2, .packet = src_stream2_packet2, }, - { .type = TEST_EV_TYPE_NOTIF_EVENT, .stream = src_stream1, .packet = src_stream1_packet1, }, - { .type = TEST_EV_TYPE_NOTIF_PACKET_END, .stream = src_stream1, .packet = src_stream1_packet1, }, - { .type = TEST_EV_TYPE_NOTIF_PACKET_END, .stream = src_stream2, .packet = src_stream2_packet2, }, - { .type = TEST_EV_TYPE_NOTIF_PACKET_BEGIN, .stream = src_stream1, .packet = src_stream1_packet2, }, - { .type = TEST_EV_TYPE_NOTIF_EVENT, .stream = src_stream1, .packet = src_stream1_packet2, }, - { .type = TEST_EV_TYPE_NOTIF_STREAM_END, .stream = src_stream2, .packet = NULL, }, - { .type = TEST_EV_TYPE_NOTIF_PACKET_END, .stream = src_stream1, .packet = src_stream1_packet2, }, - { .type = TEST_EV_TYPE_NOTIF_STREAM_END, .stream = src_stream1, .packet = NULL, }, + { .type = TEST_EV_TYPE_NOTIF_STREAM_BEGIN, .stream = pub_src_stream1, .packet = NULL, }, + { .type = TEST_EV_TYPE_NOTIF_PACKET_BEGIN, .stream = pub_src_stream1, .packet = pub_src_stream1_packet1, }, + { .type = TEST_EV_TYPE_NOTIF_EVENT, .stream = pub_src_stream1, .packet = pub_src_stream1_packet1, }, + { .type = TEST_EV_TYPE_NOTIF_EVENT, .stream = pub_src_stream1, .packet = pub_src_stream1_packet1, }, + { .type = TEST_EV_TYPE_NOTIF_STREAM_BEGIN, .stream = pub_src_stream2, .packet = NULL, }, + { .type = TEST_EV_TYPE_NOTIF_EVENT, .stream = pub_src_stream1, .packet = pub_src_stream1_packet1, }, + { .type = TEST_EV_TYPE_NOTIF_PACKET_BEGIN, .stream = pub_src_stream2, .packet = pub_src_stream2_packet2, }, + { .type = TEST_EV_TYPE_NOTIF_EVENT, .stream = pub_src_stream2, .packet = pub_src_stream2_packet2, }, + { .type = TEST_EV_TYPE_NOTIF_EVENT, .stream = pub_src_stream1, .packet = pub_src_stream1_packet1, }, + { .type = TEST_EV_TYPE_NOTIF_PACKET_END, .stream = pub_src_stream1, .packet = pub_src_stream1_packet1, }, + { .type = TEST_EV_TYPE_NOTIF_PACKET_END, .stream = pub_src_stream2, .packet = pub_src_stream2_packet2, }, + { .type = TEST_EV_TYPE_NOTIF_PACKET_BEGIN, .stream = pub_src_stream1, .packet = pub_src_stream1_packet2, }, + { .type = TEST_EV_TYPE_NOTIF_EVENT, .stream = pub_src_stream1, .packet = pub_src_stream1_packet2, }, + { .type = TEST_EV_TYPE_NOTIF_STREAM_END, .stream = pub_src_stream2, .packet = NULL, }, + { .type = TEST_EV_TYPE_NOTIF_PACKET_END, .stream = pub_src_stream1, .packet = pub_src_stream1_packet2, }, + { .type = TEST_EV_TYPE_NOTIF_STREAM_END, .stream = pub_src_stream1, .packet = NULL, }, { .type = TEST_EV_TYPE_END, }, { .type = TEST_EV_TYPE_SENTINEL, }, }; @@ -864,45 +823,45 @@ static void test_output_port_notification_iterator(void) { const struct test_event expected_test_events[] = { - { .type = TEST_EV_TYPE_NOTIF_STREAM_BEGIN, .stream = src_stream1, .packet = NULL, }, - { .type = TEST_EV_TYPE_NOTIF_PACKET_BEGIN, .stream = src_stream1, .packet = src_stream1_packet1, }, - { .type = TEST_EV_TYPE_NOTIF_EVENT, .stream = src_stream1, .packet = src_stream1_packet1, }, - { .type = TEST_EV_TYPE_NOTIF_EVENT, .stream = src_stream1, .packet = src_stream1_packet1, }, - { .type = TEST_EV_TYPE_NOTIF_STREAM_BEGIN, .stream = src_stream2, .packet = NULL, }, - { .type = TEST_EV_TYPE_NOTIF_EVENT, .stream = src_stream1, .packet = src_stream1_packet1, }, - { .type = TEST_EV_TYPE_NOTIF_PACKET_BEGIN, .stream = src_stream2, .packet = src_stream2_packet2, }, - { .type = TEST_EV_TYPE_NOTIF_EVENT, .stream = src_stream2, .packet = src_stream2_packet2, }, - { .type = TEST_EV_TYPE_NOTIF_EVENT, .stream = src_stream1, .packet = src_stream1_packet1, }, - { .type = TEST_EV_TYPE_NOTIF_PACKET_END, .stream = src_stream1, .packet = src_stream1_packet1, }, - { .type = TEST_EV_TYPE_NOTIF_PACKET_END, .stream = src_stream2, .packet = src_stream2_packet2, }, - { .type = TEST_EV_TYPE_NOTIF_PACKET_BEGIN, .stream = src_stream1, .packet = src_stream1_packet2, }, - { .type = TEST_EV_TYPE_NOTIF_EVENT, .stream = src_stream1, .packet = src_stream1_packet2, }, - { .type = TEST_EV_TYPE_NOTIF_STREAM_END, .stream = src_stream2, .packet = NULL, }, - { .type = TEST_EV_TYPE_NOTIF_PACKET_END, .stream = src_stream1, .packet = src_stream1_packet2, }, - { .type = TEST_EV_TYPE_NOTIF_STREAM_END, .stream = src_stream1, .packet = NULL, }, + { .type = TEST_EV_TYPE_NOTIF_STREAM_BEGIN, .stream = pub_src_stream1, .packet = NULL, }, + { .type = TEST_EV_TYPE_NOTIF_PACKET_BEGIN, .stream = pub_src_stream1, .packet = pub_src_stream1_packet1, }, + { .type = TEST_EV_TYPE_NOTIF_EVENT, .stream = pub_src_stream1, .packet = pub_src_stream1_packet1, }, + { .type = TEST_EV_TYPE_NOTIF_EVENT, .stream = pub_src_stream1, .packet = pub_src_stream1_packet1, }, + { .type = TEST_EV_TYPE_NOTIF_STREAM_BEGIN, .stream = pub_src_stream2, .packet = NULL, }, + { .type = TEST_EV_TYPE_NOTIF_EVENT, .stream = pub_src_stream1, .packet = pub_src_stream1_packet1, }, + { .type = TEST_EV_TYPE_NOTIF_PACKET_BEGIN, .stream = pub_src_stream2, .packet = pub_src_stream2_packet2, }, + { .type = TEST_EV_TYPE_NOTIF_EVENT, .stream = pub_src_stream2, .packet = pub_src_stream2_packet2, }, + { .type = TEST_EV_TYPE_NOTIF_EVENT, .stream = pub_src_stream1, .packet = pub_src_stream1_packet1, }, + { .type = TEST_EV_TYPE_NOTIF_PACKET_END, .stream = pub_src_stream1, .packet = pub_src_stream1_packet1, }, + { .type = TEST_EV_TYPE_NOTIF_PACKET_END, .stream = pub_src_stream2, .packet = pub_src_stream2_packet2, }, + { .type = TEST_EV_TYPE_NOTIF_PACKET_BEGIN, .stream = pub_src_stream1, .packet = pub_src_stream1_packet2, }, + { .type = TEST_EV_TYPE_NOTIF_EVENT, .stream = pub_src_stream1, .packet = pub_src_stream1_packet2, }, + { .type = TEST_EV_TYPE_NOTIF_STREAM_END, .stream = pub_src_stream2, .packet = NULL, }, + { .type = TEST_EV_TYPE_NOTIF_PACKET_END, .stream = pub_src_stream1, .packet = pub_src_stream1_packet2, }, + { .type = TEST_EV_TYPE_NOTIF_STREAM_END, .stream = pub_src_stream1, .packet = NULL, }, { .type = TEST_EV_TYPE_END, }, { .type = TEST_EV_TYPE_SENTINEL, }, }; - struct bt_component *src_comp; - struct bt_notification_iterator *notif_iter; + struct bt_component_source *src_comp; + struct bt_port_output_notification_iterator *notif_iter; enum bt_notification_iterator_status iter_status = BT_NOTIFICATION_ITERATOR_STATUS_OK; - struct bt_port *upstream_port; + struct bt_port_output *upstream_port; clear_test_events(); current_test = TEST_OUTPUT_PORT_NOTIFICATION_ITERATOR; diag("test: output port notification iterator"); BT_ASSERT(!graph); - graph = bt_graph_create(); + graph = bt_private_graph_create(); BT_ASSERT(graph); 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_put(upstream_port); + upstream_port = bt_component_source_borrow_output_port_by_name( + 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) { @@ -916,9 +875,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_put(src_comp); - BT_PUT(graph); - bt_put(notif_iter); + bt_object_put_ref(src_comp); + BT_OBJECT_PUT_REF_AND_RESET(graph); + bt_object_put_ref(notif_iter); } #define DEBUG_ENV_VAR "TEST_BT_NOTIFICATION_ITERATOR_DEBUG"