Tests: add environment field can be added after stream creation
[babeltrace.git] / tests / lib / test_ctf_writer.c
index bfa300d4b7f95ea0d7cfa69d1cc165572a9ddb47..756d168055246b276221d9b25b52f0b20d764678 100644 (file)
@@ -566,7 +566,7 @@ void append_simple_event(struct bt_ctf_stream_class *stream_class,
        bt_ctf_event_set_payload(simple_event, "enum_field_unsigned",
                enum_field_unsigned);
        ret_char = bt_ctf_field_enumeration_get_mapping_name(enum_field_unsigned);
-       ok(!strcmp(ret_char, mapping_name_test),
+       ok(ret_char && !strcmp(ret_char, mapping_name_test),
                "bt_ctf_field_enumeration_get_mapping_name returns the correct mapping name with an unsigned container");
 
        ok(bt_ctf_clock_set_time(clock, current_time) == 0, "Set clock time");
@@ -654,7 +654,18 @@ void append_simple_event(struct bt_ctf_stream_class *stream_class,
 void append_complex_event(struct bt_ctf_stream_class *stream_class,
                struct bt_ctf_stream *stream, struct bt_ctf_clock *clock)
 {
+       struct event_class_attrs_counts {
+               int id;
+               int name;
+               int loglevel;
+               int modelemfuri;
+               int unknown;
+       } attrs_count;
+
        int i;
+       int ret;
+       int64_t int64_value;
+       struct event_class_attrs_counts ;
        const char *complex_test_event_string = "Complex Test Event";
        const char *test_string = "Test string";
        struct bt_ctf_field_type *uint_35_type =
@@ -689,6 +700,7 @@ void append_complex_event(struct bt_ctf_stream_class *stream_class,
        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;
+       struct bt_object *obj;
 
        bt_ctf_field_type_set_alignment(int_16_type, 32);
        bt_ctf_field_type_integer_set_signed(int_16_type, 1);
@@ -842,6 +854,114 @@ void append_complex_event(struct bt_ctf_stream_class *stream_class,
        ok(bt_ctf_event_class_get_id(event_class) == 42,
                "bt_ctf_event_class_get_id returns the correct value");
 
+       /* Test event class attributes */
+       assert(obj = bt_object_integer_create_init(15));
+       ok(bt_ctf_event_class_set_attribute(NULL, "id", obj),
+               "bt_ctf_event_class_set_attribute handles a NULL event class correctly");
+       ok(bt_ctf_event_class_set_attribute(event_class, NULL, obj),
+               "bt_ctf_event_class_set_attribute handles a NULL name correctly");
+       ok(bt_ctf_event_class_set_attribute(event_class, "id", NULL),
+               "bt_ctf_event_class_set_attribute handles a NULL value correctly");
+       assert(!bt_object_integer_set(obj, -3));
+       ok(bt_ctf_event_class_set_attribute(event_class, "id", obj),
+               "bt_ctf_event_class_set_attribute fails with a negative \"id\" attribute");
+       assert(!bt_object_integer_set(obj, 11));
+       ret = bt_ctf_event_class_set_attribute(event_class, "id", obj);
+       ok(!ret && bt_ctf_event_class_get_id(event_class) == 11,
+               "bt_ctf_event_class_set_attribute succeeds in replacing the existing \"id\" attribute");
+       ret = bt_ctf_event_class_set_attribute(event_class, "name", obj);
+       ret &= bt_ctf_event_class_set_attribute(event_class, "model.emf.uri", obj);
+       ok(ret,
+               "bt_ctf_event_class_set_attribute cannot set \"name\" or \"model.emf.uri\" to an integer value");
+       BT_OBJECT_PUT(obj);
+
+       obj = bt_object_integer_create_init(5);
+       assert(obj);
+       ok(!bt_ctf_event_class_set_attribute(event_class, "loglevel", obj),
+               "bt_ctf_event_class_set_attribute succeeds in setting the \"loglevel\" attribute");
+       BT_OBJECT_PUT(obj);
+       ok(!bt_ctf_event_class_get_attribute_value_by_name(NULL, "loglevel"),
+               "bt_ctf_event_class_get_attribute_value_by_name handles a NULL event class correctly");
+       ok(!bt_ctf_event_class_get_attribute_value_by_name(event_class, NULL),
+               "bt_ctf_event_class_get_attribute_value_by_name handles a NULL name correctly");
+       ok(!bt_ctf_event_class_get_attribute_value_by_name(event_class, "meow"),
+               "bt_ctf_event_class_get_attribute_value_by_name fails with a non-existing attribute name");
+       obj = bt_ctf_event_class_get_attribute_value_by_name(event_class,
+               "loglevel");
+       int64_value = 0;
+       ret = bt_object_integer_get(obj, &int64_value);
+       ok(obj && !ret && int64_value == 5,
+               "bt_ctf_event_class_get_attribute_value_by_name returns the correct value");
+       BT_OBJECT_PUT(obj);
+
+       assert(obj = bt_object_string_create_init("nu name"));
+       assert(!bt_ctf_event_class_set_attribute(event_class, "name", obj));
+       ret_string = bt_ctf_event_class_get_name(event_class);
+       ok(!strcmp(ret_string, "nu name"),
+               "bt_ctf_event_class_set_attribute succeeds in replacing the existing \"name\" attribute");
+       ret = bt_ctf_event_class_set_attribute(event_class, "id", obj);
+       ret &= bt_ctf_event_class_set_attribute(event_class, "loglevel", obj);
+       ok(ret,
+               "bt_ctf_event_class_set_attribute cannot set \"id\" or \"loglevel\" to a string value");
+       BT_OBJECT_PUT(obj);
+       obj = bt_object_string_create_init("http://kernel.org/");
+       assert(obj);
+       assert(!bt_ctf_event_class_set_attribute(event_class, "model.emf.uri", obj));
+       BT_OBJECT_PUT(obj);
+
+       ok(bt_ctf_event_class_get_attribute_count(NULL),
+               "bt_ctf_event_class_get_attribute_count handles a NULL event class");
+       ok(bt_ctf_event_class_get_attribute_count(event_class) == 4,
+               "bt_ctf_event_class_get_attribute_count returns the correct count");
+       ok(!bt_ctf_event_class_get_attribute_name(NULL, 0),
+               "bt_ctf_event_class_get_attribute_name handles a NULL event class correctly");
+       ok(!bt_ctf_event_class_get_attribute_name(event_class, 4),
+               "bt_ctf_event_class_get_attribute_name handles a too large index correctly");
+       ok(!bt_ctf_event_class_get_attribute_value(NULL, 0),
+               "bt_ctf_event_class_get_attribute_value handles a NULL event class correctly");
+       ok(!bt_ctf_event_class_get_attribute_value(event_class, 4),
+               "bt_ctf_event_class_get_attribute_value handles a too large index correctly");
+
+       memset(&attrs_count, 0, sizeof(attrs_count));
+
+       for (i = 0; i < 4; ++i) {
+               ret_string = bt_ctf_event_class_get_attribute_name(event_class,
+                       i);
+               obj = bt_ctf_event_class_get_attribute_value(event_class, i);
+               assert(ret_string && obj);
+
+               if (!strcmp(ret_string, "id")) {
+                       attrs_count.id++;
+                       ok(bt_object_is_integer(obj),
+                               "bt_ctf_event_class_get_attribute_value returns the correct type (\"%s\")",
+                               ret_string);
+               } else if (!strcmp(ret_string, "name")) {
+                       attrs_count.name++;
+                       ok(bt_object_is_string(obj),
+                               "bt_ctf_event_class_get_attribute_value returns the correct type (\"%s\")",
+                               ret_string);
+               } else if (!strcmp(ret_string, "loglevel")) {
+                       attrs_count.loglevel++;
+                       ok(bt_object_is_integer(obj),
+                               "bt_ctf_event_class_get_attribute_value returns the correct type (\"%s\")",
+                               ret_string);
+               } else if (!strcmp(ret_string, "model.emf.uri")) {
+                       attrs_count.modelemfuri++;
+                       ok(bt_object_is_string(obj),
+                               "bt_ctf_event_class_get_attribute_value returns the correct type (\"%s\")",
+                               ret_string);
+               } else {
+                       attrs_count.unknown++;
+               }
+
+               BT_OBJECT_PUT(obj);
+       }
+
+       ok(attrs_count.unknown == 0, "event class has no unknown attributes");
+       ok(attrs_count.id == 1 && attrs_count.name == 1 &&
+               attrs_count.loglevel == 1 && attrs_count.modelemfuri == 1,
+               "event class has one instance of each known attribute");
+
        /* 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");
@@ -1497,7 +1617,7 @@ end:
 void test_empty_stream(struct bt_ctf_writer *writer)
 {
        int ret = 0;
-       struct bt_ctf_trace *trace = NULL;
+       struct bt_ctf_trace *trace = NULL, *ret_trace = NULL;
        struct bt_ctf_stream_class *stream_class = NULL;
        struct bt_ctf_stream *stream = NULL;
 
@@ -1515,16 +1635,26 @@ void test_empty_stream(struct bt_ctf_writer *writer)
                goto end;
        }
 
+       ok(bt_ctf_stream_class_get_trace(NULL) == NULL,
+               "bt_ctf_stream_class_get_trace handles NULL correctly");
+       ok(bt_ctf_stream_class_get_trace(stream_class) == NULL,
+               "bt_ctf_stream_class_get_trace returns NULL when stream class is orphaned");
+
        stream = bt_ctf_writer_create_stream(writer, stream_class);
        if (!stream) {
                diag("Failed to create writer stream");
                ret = -1;
                goto end;
        }
+
+       ret_trace = bt_ctf_stream_class_get_trace(stream_class);
+       ok(ret_trace == trace,
+               "bt_ctf_stream_class_get_trace returns the correct trace after a stream has been created");
 end:
        ok(ret == 0,
                "Created a stream class with default attributes and an empty stream");
        bt_ctf_trace_put(trace);
+       bt_ctf_trace_put(ret_trace);
        bt_ctf_stream_put(stream);
        bt_ctf_stream_class_put(stream_class);
 }
@@ -1535,7 +1665,8 @@ void test_instanciate_event_before_stream(struct bt_ctf_writer *writer)
        struct bt_ctf_trace *trace = NULL;
        struct bt_ctf_clock *clock = NULL;
        struct bt_ctf_stream_class *stream_class = NULL;
-       struct bt_ctf_stream *stream = NULL;
+       struct bt_ctf_stream *stream = NULL,
+               *ret_stream = NULL;
        struct bt_ctf_event_class *event_class = NULL;
        struct bt_ctf_event *event = NULL;
        struct bt_ctf_field_type *integer_type = NULL;
@@ -1616,16 +1747,26 @@ void test_instanciate_event_before_stream(struct bt_ctf_writer *writer)
                goto end;
        }
 
+       ok(bt_ctf_event_get_stream(NULL) == NULL,
+               "bt_ctf_event_get_stream handles NULL correctly");
+       ok(bt_ctf_event_get_stream(event) == NULL,
+               "bt_ctf_event_get_stream returns NULL on event which has not yet been appended to a stream");
+
        ret = bt_ctf_stream_append_event(stream, event);
        if (ret) {
                diag("Failed to append event to stream");
                goto end;
        }
+
+       ret_stream = bt_ctf_event_get_stream(event);
+       ok(ret_stream == stream,
+               "bt_ctf_event_get_stream returns an event's stream after it has been appended");
 end:
        ok(ret == 0,
                "Create an event before instanciating its associated stream");
        bt_ctf_trace_put(trace);
        bt_ctf_stream_put(stream);
+       bt_ctf_stream_put(ret_stream);
        bt_ctf_stream_class_put(stream_class);
        bt_ctf_event_class_put(event_class);
        bt_ctf_event_put(event);
@@ -1634,6 +1775,23 @@ end:
        bt_ctf_clock_put(clock);
 }
 
+void append_existing_event_class(struct bt_ctf_stream_class *stream_class)
+{
+       struct bt_ctf_event_class *event_class;
+
+       assert(event_class = bt_ctf_event_class_create("Simple Event"));
+       ok(bt_ctf_stream_class_add_event_class(stream_class, event_class),
+               "two event classes with the same name cannot cohabit within the same stream class");
+       bt_ctf_event_class_put(event_class);
+
+       event_class = bt_ctf_event_class_create("different name, ok");
+       assert(event_class);
+       assert(!bt_ctf_event_class_set_id(event_class, 11));
+       ok(bt_ctf_stream_class_add_event_class(stream_class, event_class),
+               "two event classes with the same ID cannot cohabit within the same stream class");
+       bt_ctf_event_class_put(event_class);
+}
+
 int main(int argc, char **argv)
 {
        char trace_path[] = "/tmp/ctfwriter_XXXXXX";
@@ -2229,6 +2387,9 @@ int main(int argc, char **argv)
        ok(!bt_ctf_stream_set_packet_header(stream1, packet_header),
                "Successfully set a stream's packet header");
 
+       ok(bt_ctf_writer_add_environment_field(writer, "new_field", "test") == 0,
+               "Add environment field to writer after stream creation");
+
        test_instanciate_event_before_stream(writer);
 
        append_simple_event(stream_class, stream1, clock);
@@ -2237,6 +2398,8 @@ int main(int argc, char **argv)
 
        append_complex_event(stream_class, stream1, clock);
 
+       append_existing_event_class(stream_class);
+
        test_empty_stream(writer);
 
        metadata_string = bt_ctf_writer_get_metadata_string(writer);
@@ -2247,7 +2410,6 @@ int main(int argc, char **argv)
        validate_trace(argv[2], trace_path);
 
        bt_ctf_clock_put(clock);
-       bt_ctf_stream_class_put(stream_class);
        bt_ctf_stream_class_put(ret_stream_class);
        bt_ctf_writer_put(writer);
        bt_ctf_stream_put(stream1);
@@ -2263,6 +2425,10 @@ int main(int argc, char **argv)
        bt_ctf_trace_put(trace);
        free(metadata_string);
 
+       ok(bt_ctf_stream_class_get_trace(stream_class) == NULL,
+               "bt_ctf_stream_class_get_trace returns NULL after its trace has been reclaimed");
+       bt_ctf_stream_class_put(stream_class);
+
        /* Remove all trace files and delete temporary trace directory */
        DIR *trace_dir = opendir(trace_path);
        if (!trace_dir) {
This page took 0.033768 seconds and 4 git commands to generate.