Tests: add tests for the new trace and trace clock accessors
[babeltrace.git] / tests / lib / test_ctf_writer.c
index c8942e946c85ec15c4d4b8cf2d20ce2b24e9ef06..42277026fcb9b2ac95e31dcdadd7305ae9ed0f22 100644 (file)
@@ -31,7 +31,7 @@
 #include <stdlib.h>
 #include <stdio.h>
 #include <sys/utsname.h>
-#include <limits.h>
+#include <babeltrace/compat/limits.h>
 #include <string.h>
 #include <assert.h>
 #include <unistd.h>
@@ -44,6 +44,7 @@
 
 #define METADATA_LINE_SIZE 512
 #define SEQUENCE_TEST_LENGTH 10
+#define ARRAY_TEST_LENGTH 5
 #define PACKET_RESIZE_TEST_LENGTH 100000
 
 #define DEFAULT_CLOCK_FREQ 1000000000
 
 static uint64_t current_time = 42;
 
+/* Return 1 if uuids match, zero if different. */
+int uuid_match(const unsigned char *uuid_a, const unsigned char *uuid_b)
+{
+       int ret = 0;
+       int i;
+
+       if (!uuid_a || !uuid_b) {
+               goto end;
+       }
+
+       for (i = 0; i < 16; i++) {
+               if (uuid_a[i] != uuid_b[i]) {
+                       goto end;
+               }
+       }
+
+       ret = 1;
+end:
+       return ret;
+}
+
 void validate_metadata(char *parser_path, char *metadata_path)
 {
        int ret = 0;
@@ -284,6 +306,10 @@ void append_simple_event(struct bt_ctf_stream_class *stream_class,
        size_t ret_size_t;
        int64_t ret_range_start_int64_t, ret_range_end_int64_t;
        uint64_t ret_range_start_uint64_t, ret_range_end_uint64_t;
+       struct bt_ctf_clock *ret_clock;
+       struct bt_ctf_event_class *ret_event_class;
+       struct bt_ctf_field *packet_context;
+       struct bt_ctf_field *packet_context_field;
 
        ok(uint_12_type, "Create an unsigned integer type");
 
@@ -429,11 +455,41 @@ void append_simple_event(struct bt_ctf_stream_class *stream_class,
        bt_ctf_stream_class_add_event_class(stream_class,
                simple_event_class);
 
-       simple_event = bt_ctf_event_create(simple_event_class);
+       ok(bt_ctf_stream_class_get_event_class_count(NULL) < 0,
+               "bt_ctf_stream_class_get_event_class_count handles NULL correctly");
+       ok(bt_ctf_stream_class_get_event_class_count(stream_class) == 1,
+               "bt_ctf_stream_class_get_event_class_count returns a correct number of event classes");
+       ok(bt_ctf_stream_class_get_event_class(NULL, 0) == NULL,
+               "bt_ctf_stream_class_get_event_class handles NULL correctly");
+       ok(bt_ctf_stream_class_get_event_class(stream_class, 8724) == NULL,
+               "bt_ctf_stream_class_get_event_class handles invalid indexes correctly");
+       ret_event_class = bt_ctf_stream_class_get_event_class(stream_class, 0);
+       ok(ret_event_class == simple_event_class,
+               "bt_ctf_stream_class_get_event_class returns the correct event class");
+       bt_ctf_event_class_put(ret_event_class);
+
+       ok(bt_ctf_stream_class_get_event_class_by_name(NULL, "some event name") == NULL,
+               "bt_ctf_stream_class_get_event_class_by_name handles a NULL stream class correctly");
+       ok(bt_ctf_stream_class_get_event_class_by_name(stream_class, NULL) == NULL,
+               "bt_ctf_stream_class_get_event_class_by_name handles a NULL event class name correctly");
+       ok(bt_ctf_stream_class_get_event_class_by_name(stream_class, "some event name") == NULL,
+               "bt_ctf_stream_class_get_event_class_by_name handles non-existing event class names correctly");
+        ret_event_class = bt_ctf_stream_class_get_event_class_by_name(stream_class, "Simple Event");
+       ok(ret_event_class == simple_event_class,
+               "bt_ctf_stream_class_get_event_class_by_name returns a correct event class");
+       bt_ctf_event_class_put(ret_event_class);
 
+       simple_event = bt_ctf_event_create(simple_event_class);
        ok(simple_event,
                "Instantiate an event containing a single integer field");
 
+       ok(bt_ctf_event_get_clock(NULL) == NULL,
+               "bt_ctf_event_get_clock handles NULL correctly");
+       ret_clock = bt_ctf_event_get_clock(simple_event);
+       ok(ret_clock == clock,
+               "bt_ctf_event_get_clock returns a correct clock");
+       bt_ctf_clock_put(clock);
+
        integer_field = bt_ctf_field_create(uint_12_type);
        bt_ctf_field_unsigned_integer_set_value(integer_field, 42);
        ok(bt_ctf_event_set_payload(simple_event, "integer_field",
@@ -484,6 +540,29 @@ void append_simple_event(struct bt_ctf_stream_class *stream_class,
        ok(bt_ctf_stream_append_event(stream, simple_event) == 0,
                "Append simple event to trace stream");
 
+       ok(bt_ctf_stream_get_packet_context(NULL) == NULL,
+               "bt_ctf_stream_get_packet_context handles NULL correctly");
+       packet_context = bt_ctf_stream_get_packet_context(stream);
+       ok(packet_context,
+               "bt_ctf_stream_get_packet_context returns a packet context");
+
+       packet_context_field = bt_ctf_field_structure_get_field(packet_context,
+               "packet_size");
+       ok(packet_context_field,
+               "Packet context contains the default packet_size field.");
+       bt_ctf_field_put(packet_context_field);
+       packet_context_field = bt_ctf_field_structure_get_field(packet_context,
+               "custom_field");
+       ok(bt_ctf_field_unsigned_integer_set_value(packet_context_field, 8) == 0,
+               "Custom packet context field value successfully set.");
+
+       ok(bt_ctf_stream_set_packet_context(NULL, packet_context_field) < 0,
+               "bt_ctf_stream_set_packet_context handles a NULL stream correctly");
+       ok(bt_ctf_stream_set_packet_context(stream, NULL) < 0,
+               "bt_ctf_stream_set_packet_context handles a NULL packet context correctly");
+       ok(bt_ctf_stream_set_packet_context(stream, packet_context) == 0,
+               "Successfully set a stream's packet context");
+
        ok(bt_ctf_stream_flush(stream) == 0,
                "Flush trace stream with one event");
 
@@ -501,12 +580,15 @@ void append_simple_event(struct bt_ctf_stream_class *stream_class,
        bt_ctf_field_put(enum_field_unsigned);
        bt_ctf_field_put(enum_container_field);
        bt_ctf_field_put(enum_container_field_unsigned);
+       bt_ctf_field_put(packet_context);
+       bt_ctf_field_put(packet_context_field);
 }
 
 void append_complex_event(struct bt_ctf_stream_class *stream_class,
                struct bt_ctf_stream *stream, struct bt_ctf_clock *clock)
 {
        int i;
+       const char *complex_test_event_string = "Complex Test Event";
        const char *test_string = "Test string";
        struct bt_ctf_field_type *uint_35_type =
                bt_ctf_field_type_integer_create(35);
@@ -522,6 +604,7 @@ void append_complex_event(struct bt_ctf_stream_class *stream_class,
        struct bt_ctf_field_type *string_type =
                bt_ctf_field_type_string_create();
        struct bt_ctf_field_type *sequence_type;
+       struct bt_ctf_field_type *array_type;
        struct bt_ctf_field_type *inner_structure_type =
                bt_ctf_field_type_structure_create();
        struct bt_ctf_field_type *complex_structure_type =
@@ -532,17 +615,21 @@ void append_complex_event(struct bt_ctf_stream_class *stream_class,
        struct bt_ctf_field *uint_35_field, *int_16_field, *a_string_field,
                *inner_structure_field, *complex_structure_field,
                *a_sequence_field, *enum_variant_field, *enum_container_field,
-               *variant_field, *ret_field;
-       int64_t ret_signed_int;
+               *variant_field, *an_array_field, *ret_field;
        uint64_t ret_unsigned_int;
+       int64_t ret_signed_int;
        const char *ret_string;
        size_t ret_size_t;
+       struct bt_ctf_stream_class *ret_stream_class;
+       struct bt_ctf_event_class *ret_event_class;
+       struct bt_ctf_field *packet_context, *packet_context_field;
 
        bt_ctf_field_type_set_alignment(int_16_type, 32);
        bt_ctf_field_type_integer_set_signed(int_16_type, 1);
        bt_ctf_field_type_integer_set_base(uint_35_type,
                BT_CTF_INTEGER_BASE_HEXADECIMAL);
 
+       array_type = bt_ctf_field_type_array_create(int_16_type, ARRAY_TEST_LENGTH);
        sequence_type = bt_ctf_field_type_sequence_create(int_16_type,
                "seq_len");
 
@@ -563,6 +650,8 @@ void append_complex_event(struct bt_ctf_stream_class *stream_class,
                uint_35_type, "seq_len");
        bt_ctf_field_type_structure_add_field(inner_structure_type,
                sequence_type, "a_sequence");
+       bt_ctf_field_type_structure_add_field(inner_structure_type,
+               array_type, "an_array");
 
        bt_ctf_field_type_enumeration_add_mapping(enum_variant_type,
                "UINT3_TYPE", 0, 0);
@@ -662,7 +751,7 @@ void append_complex_event(struct bt_ctf_stream_class *stream_class,
 
        ok(bt_ctf_event_class_create("clock") == NULL,
                "Reject creation of an event class with an illegal name");
-       event_class = bt_ctf_event_class_create("Complex Test Event");
+       event_class = bt_ctf_event_class_create(complex_test_event_string);
        ok(event_class, "Create an event class");
        ok(bt_ctf_event_class_add_field(event_class, uint_35_type, ""),
                "Reject addition of a field with an empty name to an event");
@@ -680,15 +769,82 @@ void append_complex_event(struct bt_ctf_stream_class *stream_class,
                "complex_structure") == 0,
                "Add composite structure to an event");
 
+       ok(bt_ctf_event_class_get_name(NULL) == NULL,
+               "bt_ctf_event_class_get_name handles NULL correctly");
+       ret_string = bt_ctf_event_class_get_name(event_class);
+       ok(!strcmp(ret_string, complex_test_event_string),
+               "bt_ctf_event_class_get_name returns a correct name");
+       ok(bt_ctf_event_class_get_id(event_class) < 0,
+               "bt_ctf_event_class_get_id returns a negative value when not set");
+       ok(bt_ctf_event_class_get_id(NULL) < 0,
+               "bt_ctf_event_class_get_id handles NULL correctly");
+       ok(bt_ctf_event_class_set_id(NULL, 42) < 0,
+               "bt_ctf_event_class_set_id handles NULL correctly");
+       ok(bt_ctf_event_class_set_id(event_class, 42) == 0,
+               "Set an event class' id");
+       ok(bt_ctf_event_class_get_id(event_class) == 42,
+               "bt_ctf_event_class_get_id returns the correct value");
+
        /* Add event class to the stream class */
        ok(bt_ctf_stream_class_add_event_class(stream_class, NULL),
                "Reject addition of NULL event class to a stream class");
        ok(bt_ctf_stream_class_add_event_class(stream_class,
                event_class) == 0, "Add an event class to stream class");
 
+       ok(bt_ctf_event_class_get_stream_class(NULL) == NULL,
+               "bt_ctf_event_class_get_stream_class handles NULL correctly");
+       ret_stream_class = bt_ctf_event_class_get_stream_class(event_class);
+       ok(ret_stream_class == stream_class,
+               "bt_ctf_event_class_get_stream_class returns the correct stream class");
+       bt_ctf_stream_class_put(ret_stream_class);
+
+       ok(bt_ctf_event_class_get_field_count(NULL) < 0,
+               "bt_ctf_event_class_get_field_count handles NULL correctly");
+       ok(bt_ctf_event_class_get_field_count(event_class) == 3,
+               "bt_ctf_event_class_get_field_count returns a correct value");
+
+       ok(bt_ctf_event_class_get_field(NULL, &ret_string,
+               &ret_field_type, 0) < 0,
+               "bt_ctf_event_class_get_field handles a NULL event class correctly");
+       ok(bt_ctf_event_class_get_field(event_class, NULL,
+               &ret_field_type, 0) < 0,
+               "bt_ctf_event_class_get_field handles a NULL field name correctly");
+       ok(bt_ctf_event_class_get_field(event_class, &ret_string,
+               NULL, 0) < 0,
+               "bt_ctf_event_class_get_field handles a NULL field type correctly");
+       ok(bt_ctf_event_class_get_field(event_class, &ret_string,
+               &ret_field_type, 42) < 0,
+               "bt_ctf_event_class_get_field handles an invalid index correctly");
+       ok(bt_ctf_event_class_get_field(event_class, &ret_string,
+               &ret_field_type, 0) == 0,
+               "bt_ctf_event_class_get_field returns a field");
+       ok(ret_field_type == uint_35_type,
+               "bt_ctf_event_class_get_field returns a correct field type");
+       bt_ctf_field_type_put(ret_field_type);
+       ok(!strcmp(ret_string, "uint_35"),
+               "bt_ctf_event_class_get_field returns a correct field name");
+       ok(bt_ctf_event_class_get_field_by_name(NULL, "") == NULL,
+               "bt_ctf_event_class_get_field_by_name handles a NULL event class correctly");
+       ok(bt_ctf_event_class_get_field_by_name(event_class, NULL) == NULL,
+               "bt_ctf_event_class_get_field_by_name handles a NULL field name correctly");
+       ok(bt_ctf_event_class_get_field_by_name(event_class, "truie") == NULL,
+               "bt_ctf_event_class_get_field_by_name handles an invalid field name correctly");
+       ret_field_type = bt_ctf_event_class_get_field_by_name(event_class,
+               "complex_structure");
+       ok(ret_field_type == complex_structure_type,
+               "bt_ctf_event_class_get_field_by_name returns a correct field type");
+       bt_ctf_field_type_put(ret_field_type);
+
        event = bt_ctf_event_create(event_class);
        ok(event, "Instanciate a complex event");
 
+       ok(bt_ctf_event_get_class(NULL) == NULL,
+               "bt_ctf_event_get_class handles NULL correctly");
+       ret_event_class = bt_ctf_event_get_class(event);
+       ok(ret_event_class == event_class,
+               "bt_ctf_event_get_class returns the correct event class");
+       bt_ctf_event_class_put(ret_event_class);
+
        uint_35_field = bt_ctf_event_get_payload(event, "uint_35");
        if (!uint_35_field) {
                printf("uint_35_field is NULL\n");
@@ -696,9 +852,9 @@ void append_complex_event(struct bt_ctf_stream_class *stream_class,
 
        ok(uint_35_field, "Use bt_ctf_event_get_payload to get a field instance ");
        bt_ctf_field_unsigned_integer_set_value(uint_35_field, 0x0DDF00D);
-       ok(bt_ctf_field_unsigned_integer_get_value(NULL, &ret_unsigned_int) == -1,
+       ok(bt_ctf_field_unsigned_integer_get_value(NULL, &ret_unsigned_int) < 0,
                "bt_ctf_field_unsigned_integer_get_value properly properly handles a NULL field.");
-       ok(bt_ctf_field_unsigned_integer_get_value(uint_35_field, NULL) == -1,
+       ok(bt_ctf_field_unsigned_integer_get_value(uint_35_field, NULL) < 0,
                "bt_ctf_field_unsigned_integer_get_value properly handles a NULL return value");
        ok(bt_ctf_field_unsigned_integer_get_value(uint_35_field,
                &ret_unsigned_int) == 0,
@@ -706,15 +862,15 @@ void append_complex_event(struct bt_ctf_stream_class *stream_class,
        ok(ret_unsigned_int == 0x0DDF00D,
                "bt_ctf_field_unsigned_integer_get_value returns the correct value");
        ok(bt_ctf_field_signed_integer_get_value(uint_35_field,
-               &ret_signed_int) == -1,
+               &ret_signed_int) < 0,
                "bt_ctf_field_signed_integer_get_value fails on an unsigned field");
        bt_ctf_field_put(uint_35_field);
 
        int_16_field = bt_ctf_event_get_payload(event, "int_16");
        bt_ctf_field_signed_integer_set_value(int_16_field, -12345);
-       ok(bt_ctf_field_signed_integer_get_value(NULL, &ret_signed_int) == -1,
+       ok(bt_ctf_field_signed_integer_get_value(NULL, &ret_signed_int) < 0,
                "bt_ctf_field_signed_integer_get_value properly handles a NULL field");
-       ok(bt_ctf_field_signed_integer_get_value(int_16_field, NULL) == -1,
+       ok(bt_ctf_field_signed_integer_get_value(int_16_field, NULL) < 0,
                "bt_ctf_field_signed_integer_get_value properly handles a NULL return value");
        ok(bt_ctf_field_signed_integer_get_value(int_16_field,
                &ret_signed_int) == 0,
@@ -722,7 +878,7 @@ void append_complex_event(struct bt_ctf_stream_class *stream_class,
        ok(ret_signed_int == -12345,
                "bt_ctf_field_signed_integer_get_value returns the correct value");
        ok(bt_ctf_field_unsigned_integer_get_value(int_16_field,
-               &ret_unsigned_int) == -1,
+               &ret_unsigned_int) < 0,
                "bt_ctf_field_unsigned_integer_get_value fails on a signed field");
        bt_ctf_field_put(int_16_field);
 
@@ -753,6 +909,8 @@ void append_complex_event(struct bt_ctf_stream_class *stream_class,
                inner_structure_field, "seq_len");
        a_sequence_field = bt_ctf_field_structure_get_field(
                inner_structure_field, "a_sequence");
+       an_array_field = bt_ctf_field_structure_get_field(
+               inner_structure_field, "an_array");
 
        enum_container_field = bt_ctf_field_enumeration_get_container(
                enum_variant_field);
@@ -769,7 +927,7 @@ void append_complex_event(struct bt_ctf_stream_class *stream_class,
                "bt_ctf_field_string_get_value correctly handles NULL");
        ret_string = bt_ctf_field_string_get_value(a_string_field);
        ok(ret_string, "bt_ctf_field_string_get_value returns a string");
-       ok(!strcmp(ret_string, test_string),
+       ok(ret_string ? !strcmp(ret_string, test_string) : 0,
                "bt_ctf_field_string_get_value returns a correct value");
        bt_ctf_field_unsigned_integer_set_value(uint_35_field,
                SEQUENCE_TEST_LENGTH);
@@ -802,9 +960,26 @@ void append_complex_event(struct bt_ctf_stream_class *stream_class,
                bt_ctf_field_put(int_16_field);
        }
 
+       for (i = 0; i < ARRAY_TEST_LENGTH; i++) {
+               int_16_field = bt_ctf_field_array_get_field(
+                       an_array_field, i);
+               bt_ctf_field_signed_integer_set_value(int_16_field, i);
+               bt_ctf_field_put(int_16_field);
+       }
+
        bt_ctf_clock_set_time(clock, ++current_time);
        ok(bt_ctf_stream_append_event(stream, event) == 0,
                "Append a complex event to a stream");
+
+       /*
+        * Populate the custom packet context field with a dummy value
+        * otherwise flush will fail.
+        */
+       packet_context = bt_ctf_stream_get_packet_context(stream);
+       packet_context_field = bt_ctf_field_structure_get_field(packet_context,
+               "custom_field");
+       bt_ctf_field_unsigned_integer_set_value(packet_context_field, 1);
+
        ok(bt_ctf_stream_flush(stream) == 0,
                "Flush a stream containing a complex event");
 
@@ -813,14 +988,18 @@ void append_complex_event(struct bt_ctf_stream_class *stream_class,
        bt_ctf_field_put(inner_structure_field);
        bt_ctf_field_put(complex_structure_field);
        bt_ctf_field_put(a_sequence_field);
+       bt_ctf_field_put(an_array_field);
        bt_ctf_field_put(enum_variant_field);
        bt_ctf_field_put(enum_container_field);
        bt_ctf_field_put(variant_field);
        bt_ctf_field_put(ret_field);
+       bt_ctf_field_put(packet_context_field);
+       bt_ctf_field_put(packet_context);
        bt_ctf_field_type_put(uint_35_type);
        bt_ctf_field_type_put(int_16_type);
        bt_ctf_field_type_put(string_type);
        bt_ctf_field_type_put(sequence_type);
+       bt_ctf_field_type_put(array_type);
        bt_ctf_field_type_put(inner_structure_type);
        bt_ctf_field_type_put(complex_structure_type);
        bt_ctf_field_type_put(uint_3_type);
@@ -1101,7 +1280,7 @@ void type_field_tests()
        ok(!enumeration_array_type,
                "Check enumeration types are validated when creating an array");
        ok(bt_ctf_field_type_structure_add_field(composite_structure_type,
-               enumeration_type, "enumeration") == 0,
+               enumeration_type, "enumeration"),
                "Check enumeration types are validated when adding them as structure members");
        enumeration = bt_ctf_field_create(enumeration_type);
        ok(!enumeration,
@@ -1140,6 +1319,12 @@ void packet_resize_test(struct bt_ctf_stream_class *stream_class,
                bt_ctf_field_type_integer_create(17);
        struct bt_ctf_field_type *string_type =
                bt_ctf_field_type_string_create();
+       struct bt_ctf_event *event;
+       struct bt_ctf_field *ret_field;
+       struct bt_ctf_field_type *ret_field_type;
+       uint64_t ret_uint64;
+       int events_appended = 0;
+       struct bt_ctf_field *packet_context, *packet_context_field;
 
        ret |= bt_ctf_event_class_add_field(event_class, integer_type,
                "field_1");
@@ -1151,8 +1336,22 @@ void packet_resize_test(struct bt_ctf_stream_class *stream_class,
                goto end;
        }
 
+       event = bt_ctf_event_create(event_class);
+       ret_field = bt_ctf_event_get_payload_by_index(event, 0);
+       ret_field_type = bt_ctf_field_get_type(ret_field);
+       ok(ret_field_type == integer_type,
+               "bt_ctf_event_get_payload_by_index returns a correct field");
+       bt_ctf_field_type_put(ret_field_type);
+       bt_ctf_field_put(ret_field);
+
+       ok(bt_ctf_event_get_payload_by_index(NULL, 0) == NULL,
+               "bt_ctf_event_get_payload_by_index handles NULL correctly");
+       ok(bt_ctf_event_get_payload_by_index(event, 4) == NULL,
+               "bt_ctf_event_get_payload_by_index handles an invalid index correctly");
+       bt_ctf_event_put(event);
+
        for (i = 0; i < PACKET_RESIZE_TEST_LENGTH; i++) {
-               struct bt_ctf_event *event = bt_ctf_event_create(event_class);
+               event = bt_ctf_event_create(event_class);
                struct bt_ctf_field *integer =
                        bt_ctf_field_create(integer_type);
                struct bt_ctf_field *string =
@@ -1174,12 +1373,41 @@ void packet_resize_test(struct bt_ctf_stream_class *stream_class,
                        break;
                }
        }
+
+       events_appended = 1;
+       ok(bt_ctf_stream_get_discarded_events_count(NULL, &ret_uint64) < 0,
+               "bt_ctf_stream_get_discarded_events_count handles a NULL stream correctly");
+       ok(bt_ctf_stream_get_discarded_events_count(stream, NULL) < 0,
+               "bt_ctf_stream_get_discarded_events_count handles a NULL return pointer correctly");
+       ret = bt_ctf_stream_get_discarded_events_count(stream, &ret_uint64);
+       ok(ret == 0 && ret_uint64 == 0,
+               "bt_ctf_stream_get_discarded_events_count returns a correct number of discarded events when none were discarded");
+       bt_ctf_stream_append_discarded_events(stream, 1000);
+       ret = bt_ctf_stream_get_discarded_events_count(stream, &ret_uint64);
+       ok(ret == 0 && ret_uint64 == 1000,
+               "bt_ctf_stream_get_discarded_events_count returns a correct number of discarded events when some were discarded");
+
 end:
-       ok(ret == 0, "Append 100 000 events to a stream");
+       ok(events_appended, "Append 100 000 events to a stream");
+
+       /*
+        * Populate the custom packet context field with a dummy value
+        * otherwise flush will fail.
+        */
+       packet_context = bt_ctf_stream_get_packet_context(stream);
+       packet_context_field = bt_ctf_field_structure_get_field(packet_context,
+               "custom_field");
+       bt_ctf_field_unsigned_integer_set_value(packet_context_field, 2);
+
        ok(bt_ctf_stream_flush(stream) == 0,
                "Flush a stream that forces a packet resize");
+       ret = bt_ctf_stream_get_discarded_events_count(stream, &ret_uint64);
+       ok(ret == 0 && ret_uint64 == 1000,
+               "bt_ctf_stream_get_discarded_events_count returns a correct number of discarded events after a flush");
        bt_ctf_field_type_put(integer_type);
        bt_ctf_field_type_put(string_type);
+       bt_ctf_field_put(packet_context);
+       bt_ctf_field_put(packet_context_field);
        bt_ctf_event_class_put(event_class);
 }
 
@@ -1199,14 +1427,20 @@ int main(int argc, char **argv)
        char *metadata_string;
        struct bt_ctf_writer *writer;
        struct utsname name;
-       char hostname[HOST_NAME_MAX];
-       struct bt_ctf_clock *clock;
+       char hostname[BABELTRACE_HOST_NAME_MAX];
+       struct bt_ctf_clock *clock, *ret_clock;
        struct bt_ctf_stream_class *stream_class;
        struct bt_ctf_stream *stream1;
+       const char *ret_string;
+       const unsigned char *ret_uuid;
+       unsigned char tmp_uuid[16] = { 0 };
+       struct bt_ctf_field_type *packet_context_type, *packet_context_field_type;
+       struct bt_ctf_trace *trace;
+       int ret;
 
        if (argc < 3) {
                printf("Usage: tests-ctf-writer path_to_ctf_parser_test path_to_babeltrace\n");
-               exit(-1);
+               return -1;
        }
 
        plan_no_plan();
@@ -1222,7 +1456,10 @@ int main(int argc, char **argv)
        ok(writer, "bt_ctf_create succeeds in creating trace with path");
 
        /* Add environment context to the trace */
-       gethostname(hostname, HOST_NAME_MAX);
+       ret = gethostname(hostname, sizeof(hostname));
+       if (ret < 0) {
+               return ret;
+       }
        ok(bt_ctf_writer_add_environment_field(writer, "host", hostname) == 0,
                "Add host (%s) environment field to writer instance",
                hostname);
@@ -1266,7 +1503,7 @@ int main(int argc, char **argv)
        ok(clock, "Clock created sucessfully");
        returned_clock_name = bt_ctf_clock_get_name(clock);
        ok(returned_clock_name, "bt_ctf_clock_get_name returns a clock name");
-       ok(strcmp(returned_clock_name, clock_name) == 0,
+       ok(returned_clock_name ? !strcmp(returned_clock_name, clock_name) : 0,
                "Returned clock name is valid");
 
        returned_clock_description = bt_ctf_clock_get_description(clock);
@@ -1277,7 +1514,8 @@ int main(int argc, char **argv)
        returned_clock_description = bt_ctf_clock_get_description(clock);
        ok(returned_clock_description,
                "bt_ctf_clock_get_description returns a description.");
-       ok(strcmp(returned_clock_description, clock_description) == 0,
+       ok(returned_clock_description ?
+               !strcmp(returned_clock_description, clock_description) : 0,
                "Returned clock description is valid");
 
        ok(bt_ctf_clock_get_frequency(clock) == DEFAULT_CLOCK_FREQ,
@@ -1326,6 +1564,26 @@ int main(int argc, char **argv)
        ok(bt_ctf_writer_add_clock(writer, clock),
                "Verify a clock can't be added twice to a writer instance");
 
+       ok(!bt_ctf_writer_get_trace(NULL),
+               "bt_ctf_writer_get_trace correctly handles NULL");
+       trace = bt_ctf_writer_get_trace(writer);
+       ok(trace,
+               "bt_ctf_writer_get_trace returns a bt_ctf_trace object");
+       ok(bt_ctf_trace_get_clock_count(NULL) < 0,
+               "bt_ctf_trace_get_clock_count correctly handles NULL");
+       ok(bt_ctf_trace_get_clock_count(trace) == 1,
+               "bt_ctf_trace_get_clock_count returns the correct number of clocks");
+       ok(!bt_ctf_trace_get_clock(NULL, 0),
+               "bt_ctf_trace_get_clock correctly handles NULL");
+       ok(!bt_ctf_trace_get_clock(trace, -1),
+               "bt_ctf_trace_get_clock correctly handles negative indexes");
+       ok(!bt_ctf_trace_get_clock(trace, 1),
+               "bt_ctf_trace_get_clock correctly handles out of bound accesses");
+       ret_clock = bt_ctf_trace_get_clock(trace, 0);
+       ok(ret_clock == clock,
+               "bt_ctf_trace_get_clock returns the right clock instance");
+       bt_ctf_clock_put(ret_clock);
+
        ok(!bt_ctf_clock_get_name(NULL),
                "bt_ctf_clock_get_name correctly handles NULL");
        ok(!bt_ctf_clock_get_description(NULL),
@@ -1338,39 +1596,119 @@ int main(int argc, char **argv)
                "bt_ctf_clock_get_offset_s correctly handles NULL");
        ok(bt_ctf_clock_get_offset(NULL) == -1ULL,
                "bt_ctf_clock_get_offset correctly handles NULL");
-       ok(bt_ctf_clock_get_is_absolute(NULL) == -1,
+       ok(bt_ctf_clock_get_is_absolute(NULL) < 0,
                "bt_ctf_clock_get_is_absolute correctly handles NULL");
        ok(bt_ctf_clock_get_time(NULL) == -1ULL,
                "bt_ctf_clock_get_time correctly handles NULL");
 
-       ok(bt_ctf_clock_set_description(NULL, NULL) == -1,
+       ok(bt_ctf_clock_set_description(NULL, NULL) < 0,
                "bt_ctf_clock_set_description correctly handles NULL clock");
-       ok(bt_ctf_clock_set_frequency(NULL, frequency) == -1,
+       ok(bt_ctf_clock_set_frequency(NULL, frequency) < 0,
                "bt_ctf_clock_set_frequency correctly handles NULL clock");
-       ok(bt_ctf_clock_set_precision(NULL, precision) == -1,
+       ok(bt_ctf_clock_set_precision(NULL, precision) < 0,
                "bt_ctf_clock_get_precision correctly handles NULL clock");
-       ok(bt_ctf_clock_set_offset_s(NULL, offset_s) == -1,
+       ok(bt_ctf_clock_set_offset_s(NULL, offset_s) < 0,
                "bt_ctf_clock_set_offset_s correctly handles NULL clock");
-       ok(bt_ctf_clock_set_offset(NULL, offset) == -1,
+       ok(bt_ctf_clock_set_offset(NULL, offset) < 0,
                "bt_ctf_clock_set_offset correctly handles NULL clock");
-       ok(bt_ctf_clock_set_is_absolute(NULL, is_absolute) == -1,
+       ok(bt_ctf_clock_set_is_absolute(NULL, is_absolute) < 0,
                "bt_ctf_clock_set_is_absolute correctly handles NULL clock");
-       ok(bt_ctf_clock_set_time(NULL, current_time) == -1,
+       ok(bt_ctf_clock_set_time(NULL, current_time) < 0,
                "bt_ctf_clock_set_time correctly handles NULL clock");
+       ok(bt_ctf_clock_get_uuid(NULL) == NULL,
+               "bt_ctf_clock_get_uuid correctly handles NULL clock");
+       ret_uuid = bt_ctf_clock_get_uuid(clock);
+       ok(ret_uuid,
+               "bt_ctf_clock_get_uuid returns a UUID");
+       if (ret_uuid) {
+               memcpy(tmp_uuid, ret_uuid, sizeof(tmp_uuid));
+               /* Slightly modify UUID */
+               tmp_uuid[sizeof(tmp_uuid) - 1]++;
+       }
+
+       ok(bt_ctf_clock_set_uuid(NULL, tmp_uuid) < 0,
+               "bt_ctf_clock_set_uuid correctly handles a NULL clock");
+       ok(bt_ctf_clock_set_uuid(clock, NULL) < 0,
+               "bt_ctf_clock_set_uuid correctly handles a NULL UUID");
+       ok(bt_ctf_clock_set_uuid(clock, tmp_uuid) == 0,
+               "bt_ctf_clock_set_uuid sets a new uuid succesfully");
+       ret_uuid = bt_ctf_clock_get_uuid(clock);
+       ok(ret_uuid,
+               "bt_ctf_clock_get_uuid returns a UUID after setting a new one");
+       ok(uuid_match(ret_uuid, tmp_uuid),
+               "bt_ctf_clock_get_uuid returns the correct UUID after setting a new one");
 
        /* Define a stream class */
        stream_class = bt_ctf_stream_class_create("test_stream");
+
+       ok(bt_ctf_stream_class_get_name(NULL) == NULL,
+               "bt_ctf_stream_class_get_name handles NULL correctly");
+       ret_string = bt_ctf_stream_class_get_name(stream_class);
+       ok(!strcmp(ret_string, "test_stream"),
+               "bt_ctf_stream_class_get_name returns a correct stream class name");
+
+       ok(bt_ctf_stream_class_get_clock(stream_class) == NULL,
+               "bt_ctf_stream_class_get_clock returns NULL when a clock was not set");
+       ok(bt_ctf_stream_class_get_clock(NULL) == NULL,
+               "bt_ctf_stream_class_get_clock handles NULL correctly");
+
        ok(stream_class, "Create stream class");
        ok(bt_ctf_stream_class_set_clock(stream_class, clock) == 0,
                "Set a stream class' clock");
+       ret_clock = bt_ctf_stream_class_get_clock(stream_class);
+       ok(ret_clock == clock,
+               "bt_ctf_stream_class_get_clock returns a correct clock");
+       bt_ctf_clock_put(ret_clock);
 
        /* Test the event fields and event types APIs */
        type_field_tests();
 
+       ok(bt_ctf_stream_class_get_id(stream_class) < 0,
+               "bt_ctf_stream_class_get_id returns an error when no id is set");
+       ok(bt_ctf_stream_class_get_id(NULL) < 0,
+               "bt_ctf_stream_class_get_id handles NULL correctly");
+       ok(bt_ctf_stream_class_set_id(NULL, 123) < 0,
+               "bt_ctf_stream_class_set_id handles NULL correctly");
+       ok(bt_ctf_stream_class_set_id(stream_class, 123) == 0,
+               "Set an stream class' id");
+       ok(bt_ctf_stream_class_get_id(stream_class) == 123,
+               "bt_ctf_stream_class_get_id returns the correct value");
+
+       /* Create a "uint5_t" equivalent custom packet context field */
+       packet_context_field_type = bt_ctf_field_type_integer_create(5);
+
+       ok(bt_ctf_stream_class_get_packet_context_type(NULL) == NULL,
+               "bt_ctf_stream_class_get_packet_context_type handles NULL correctly");
+
+       /* Add a custom field to the stream class' packet context */
+       packet_context_type = bt_ctf_stream_class_get_packet_context_type(stream_class);
+       ok(packet_context_type,
+               "bt_ctf_stream_class_get_packet_context_type returns a packet context type.");
+       ok(bt_ctf_field_type_get_type_id(packet_context_type) == CTF_TYPE_STRUCT,
+               "Packet context is a structure");
+
+       ok(bt_ctf_stream_class_set_packet_context_type(NULL, packet_context_type),
+               "bt_ctf_stream_class_set_packet_context_type handles a NULL stream class correctly");
+       ok(bt_ctf_stream_class_set_packet_context_type(stream_class, NULL),
+               "bt_ctf_stream_class_set_packet_context_type handles a NULL packet context type correctly");
+       ret = bt_ctf_field_type_structure_add_field(packet_context_type,
+               packet_context_field_type, "custom_field");
+       ok(ret == 0, "Packet context field added successfully");
+
+
        /* Instantiate a stream and append events */
        stream1 = bt_ctf_writer_create_stream(writer, stream_class);
        ok(stream1, "Instanciate a stream class from writer");
 
+       /*
+        * Try to modify the packet context type after a stream has been
+        * created.
+        */
+       ret = bt_ctf_field_type_structure_add_field(packet_context_type,
+               packet_context_field_type, "should_fail");
+       ok(ret < 0,
+               "Packet context type can't be modified once a stream class has been instanciated");
+
        /* Should fail after instanciating a stream (locked)*/
        ok(bt_ctf_stream_class_set_clock(stream_class, clock),
                "Changes to a stream class that was already instantiated fail");
@@ -1392,6 +1730,9 @@ int main(int argc, char **argv)
        bt_ctf_stream_class_put(stream_class);
        bt_ctf_writer_put(writer);
        bt_ctf_stream_put(stream1);
+       bt_ctf_field_type_put(packet_context_type);
+       bt_ctf_field_type_put(packet_context_field_type);
+       bt_ctf_trace_put(trace);
        free(metadata_string);
 
        /* Remove all trace files and delete temporary trace directory */
This page took 0.032164 seconds and 4 git commands to generate.