X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=tests%2Flib%2Ftest_ctf_writer.c;h=756d168055246b276221d9b25b52f0b20d764678;hb=3975bd7e68ab519547cfb2e63fef3c12eeabdf5c;hp=52106f2d6e581044a0dbd230fec9744ffe66044f;hpb=fdf80f327cd463eacb1d8a209091733b0cea2073;p=babeltrace.git diff --git a/tests/lib/test_ctf_writer.c b/tests/lib/test_ctf_writer.c index 52106f2d..756d1680 100644 --- a/tests/lib/test_ctf_writer.c +++ b/tests/lib/test_ctf_writer.c @@ -3,7 +3,7 @@ * * CTF Writer test * - * Copyright 2013 - Jérémie Galarneau + * Copyright 2013 - 2015 - Jérémie Galarneau * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by @@ -26,7 +26,9 @@ #include #include #include +#include #include +#include #include #include #include @@ -158,12 +160,12 @@ result: /* Output the metadata and parser output as diagnostic */ while (getline(&line, &len, metadata_fp) > 0) { - diag("%s", line); + fprintf(stderr, "# %s", line); } rewind(parser_output_fp); while (getline(&line, &len, parser_output_fp) > 0) { - diag("%s", line); + fprintf(stderr, "# %s", line); } free(line); @@ -453,8 +455,30 @@ void append_simple_event(struct bt_ctf_stream_class *stream_class, "integer_field"); bt_ctf_event_class_add_field(simple_event_class, float_type, "float_field"); - bt_ctf_stream_class_add_event_class(stream_class, - simple_event_class); + + assert(!bt_ctf_event_class_set_id(simple_event_class, 13)); + + /* Set an event context type which will contain a single integer*/ + ok(!bt_ctf_field_type_structure_add_field(event_context_type, uint_12_type, + "event_specific_context"), + "Add event specific context field"); + ok(bt_ctf_event_class_get_context_type(NULL) == NULL, + "bt_ctf_event_class_get_context_type handles NULL correctly"); + ok(bt_ctf_event_class_get_context_type(simple_event_class) == NULL, + "bt_ctf_event_class_get_context_type returns NULL when no event context type is set"); + + ok(bt_ctf_event_class_set_context_type(simple_event_class, NULL) < 0, + "bt_ctf_event_class_set_context_type handles a NULL context type correctly"); + ok(bt_ctf_event_class_set_context_type(NULL, event_context_type) < 0, + "bt_ctf_event_class_set_context_type handles a NULL event class correctly"); + ok(!bt_ctf_event_class_set_context_type(simple_event_class, event_context_type), + "Set an event class' context type successfully"); + returned_type = bt_ctf_event_class_get_context_type(simple_event_class); + ok(returned_type == event_context_type, + "bt_ctf_event_class_get_context_type returns the appropriate type"); + bt_ctf_field_type_put(returned_type); + + bt_ctf_stream_class_add_event_class(stream_class, 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"); @@ -468,6 +492,15 @@ void append_simple_event(struct bt_ctf_stream_class *stream_class, 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_id(NULL, 0), + "bt_ctf_stream_class_get_event_class_by_id handles NULL correctly"); + ok(!bt_ctf_stream_class_get_event_class_by_id(stream_class, 2), + "bt_ctf_stream_class_get_event_class_by_id returns NULL when the requested ID doesn't exist"); + ret_event_class = + bt_ctf_stream_class_get_event_class_by_id(stream_class, 13); + ok(ret_event_class == simple_event_class, + "bt_ctf_stream_class_get_event_class_by_id returns a 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"); @@ -480,25 +513,6 @@ void append_simple_event(struct bt_ctf_stream_class *stream_class, "bt_ctf_stream_class_get_event_class_by_name returns a correct event class"); bt_ctf_event_class_put(ret_event_class); - /* Set an event context type which will contain a single integer*/ - ok(!bt_ctf_field_type_structure_add_field(event_context_type, uint_12_type, - "event_specific_context"), - "Add event specific context field"); - ok(bt_ctf_event_class_get_context_type(NULL) == NULL, - "bt_ctf_event_class_get_context_type handles NULL correctly"); - ok(bt_ctf_event_class_get_context_type(simple_event_class) == NULL, - "bt_ctf_event_class_get_context_type returns NULL when no event context type is set"); - ok(bt_ctf_event_class_set_context_type(simple_event_class, NULL) < 0, - "bt_ctf_event_class_set_context_type handles a NULL context type correctly"); - ok(bt_ctf_event_class_set_context_type(NULL, event_context_type) < 0, - "bt_ctf_event_class_set_context_type handles a NULL event class correctly"); - ok(!bt_ctf_event_class_set_context_type(simple_event_class, event_context_type), - "Set an event class' context type successfully"); - returned_type = bt_ctf_event_class_get_context_type(simple_event_class); - ok(returned_type == event_context_type, - "bt_ctf_event_class_get_context_type returns the appropriate type"); - bt_ctf_field_type_put(returned_type); - simple_event = bt_ctf_event_create(simple_event_class); ok(simple_event, "Instantiate an event containing a single integer field"); @@ -552,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"); @@ -640,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 = @@ -675,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); @@ -828,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"); @@ -1483,8 +1617,7 @@ end: void test_empty_stream(struct bt_ctf_writer *writer) { int ret = 0; - struct bt_ctf_trace *trace = NULL; - struct bt_ctf_clock *clock = NULL; + struct bt_ctf_trace *trace = NULL, *ret_trace = NULL; struct bt_ctf_stream_class *stream_class = NULL; struct bt_ctf_stream *stream = NULL; @@ -1502,21 +1635,163 @@ 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_clock_put(clock); + bt_ctf_trace_put(ret_trace); bt_ctf_stream_put(stream); bt_ctf_stream_class_put(stream_class); } +void test_instanciate_event_before_stream(struct bt_ctf_writer *writer) +{ + int ret = 0; + 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, + *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; + struct bt_ctf_field *integer = NULL; + + trace = bt_ctf_writer_get_trace(writer); + if (!trace) { + diag("Failed to get trace from writer"); + ret = -1; + goto end; + } + + clock = bt_ctf_trace_get_clock(trace, 0); + if (!clock) { + diag("Failed to get clock from trace"); + ret = -1; + goto end; + } + + stream_class = bt_ctf_stream_class_create("event_before_stream_test"); + if (!stream_class) { + diag("Failed to create stream class"); + ret = -1; + goto end; + } + + ret = bt_ctf_stream_class_set_clock(stream_class, clock); + if (ret) { + diag("Failed to set stream class clock"); + goto end; + } + + event_class = bt_ctf_event_class_create("some_event_class_name"); + integer_type = bt_ctf_field_type_integer_create(32); + if (!integer_type) { + diag("Failed to create integer field type"); + ret = -1; + goto end; + } + + ret = bt_ctf_event_class_add_field(event_class, integer_type, + "integer_field"); + if (ret) { + diag("Failed to add field to event class"); + goto end; + } + + ret = bt_ctf_stream_class_add_event_class(stream_class, + event_class); + if (ret) { + diag("Failed to add event class to stream class"); + } + + event = bt_ctf_event_create(event_class); + if (!event) { + diag("Failed to create event"); + ret = -1; + goto end; + } + + integer = bt_ctf_event_get_payload_by_index(event, 0); + if (!integer) { + diag("Failed to get integer field payload from event"); + ret = -1; + goto end; + } + + ret = bt_ctf_field_unsigned_integer_set_value(integer, 1234); + if (ret) { + diag("Failed to set integer field value"); + goto end; + } + + stream = bt_ctf_writer_create_stream(writer, stream_class); + if (!stream) { + diag("Failed to create writer stream"); + ret = -1; + 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); + bt_ctf_field_type_put(integer_type); + bt_ctf_field_put(integer); + 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"; @@ -1546,10 +1821,13 @@ int main(int argc, char **argv) *packet_header_field_type, *integer_type, *stream_event_context_type, - *ret_field_type; + *ret_field_type, + *event_header_field_type; struct bt_ctf_field *packet_header, *packet_header_field; struct bt_ctf_trace *trace; int ret; + int64_t ret_int64_t; + struct bt_object *obj; if (argc < 3) { printf("Usage: tests-ctf-writer path_to_ctf_parser_test path_to_babeltrace\n"); @@ -1568,6 +1846,16 @@ int main(int argc, char **argv) writer = bt_ctf_writer_create(trace_path); ok(writer, "bt_ctf_create succeeds in creating trace with path"); + 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_set_byte_order(trace, BT_CTF_BYTE_ORDER_BIG_ENDIAN) == 0, + "Set a trace's byte order to big endian"); + ok(bt_ctf_trace_get_byte_order(trace) == BT_CTF_BYTE_ORDER_BIG_ENDIAN, + "bt_ctf_trace_get_byte_order returns a correct endianness"); + /* Add environment context to the trace */ ret = gethostname(hostname, sizeof(hostname)); if (ret < 0) { @@ -1586,6 +1874,123 @@ int main(int argc, char **argv) NULL), "bt_ctf_writer_add_environment_field error with NULL field value"); + /* Test bt_ctf_trace_set_environment_field with an integer object */ + obj = bt_object_integer_create_init(23); + assert(obj); + ok(bt_ctf_trace_set_environment_field(NULL, "test_env_int_obj", obj), + "bt_ctf_trace_set_environment_field handles a NULL trace correctly"); + ok(bt_ctf_trace_set_environment_field(trace, NULL, obj), + "bt_ctf_trace_set_environment_field handles a NULL name correctly"); + ok(bt_ctf_trace_set_environment_field(trace, "test_env_int_obj", NULL), + "bt_ctf_trace_set_environment_field handles a NULL value correctly"); + ok(!bt_ctf_trace_set_environment_field(trace, "test_env_int_obj", obj), + "bt_ctf_trace_set_environment_field succeeds in adding an integer object"); + BT_OBJECT_PUT(obj); + + /* Test bt_ctf_trace_set_environment_field with a string object */ + obj = bt_object_string_create_init("the value"); + assert(obj); + ok(!bt_ctf_trace_set_environment_field(trace, "test_env_str_obj", obj), + "bt_ctf_trace_set_environment_field succeeds in adding a string object"); + BT_OBJECT_PUT(obj); + + /* Test bt_ctf_trace_set_environment_field_integer */ + ok(bt_ctf_trace_set_environment_field_integer(NULL, "test_env_int", + -194875), + "bt_ctf_trace_set_environment_field_integer handles a NULL trace correctly"); + ok(bt_ctf_trace_set_environment_field_integer(trace, NULL, -194875), + "bt_ctf_trace_set_environment_field_integer handles a NULL name correctly"); + ok(!bt_ctf_trace_set_environment_field_integer(trace, "test_env_int", + -164973), + "bt_ctf_trace_set_environment_field_integer succeeds"); + + /* Test bt_ctf_trace_set_environment_field_string */ + ok(bt_ctf_trace_set_environment_field_string(NULL, "test_env_str", + "yeah"), + "bt_ctf_trace_set_environment_field_string handles a NULL trace correctly"); + ok(bt_ctf_trace_set_environment_field_string(trace, NULL, "yeah"), + "bt_ctf_trace_set_environment_field_string handles a NULL name correctly"); + ok(bt_ctf_trace_set_environment_field_string(trace, "test_env_str", + NULL), + "bt_ctf_trace_set_environment_field_string handles a NULL value correctly"); + ok(!bt_ctf_trace_set_environment_field_string(trace, "test_env_str", + "oh yeah"), + "bt_ctf_trace_set_environment_field_string succeeds"); + + /* Test bt_ctf_trace_get_environment_field_count */ + ok(bt_ctf_trace_get_environment_field_count(NULL) < 0, + "bt_ctf_trace_get_environment_field_count handles a NULL trace correctly"); + ok(bt_ctf_trace_get_environment_field_count(trace) == 5, + "bt_ctf_trace_get_environment_field_count returns a correct number of environment fields"); + + /* Test bt_ctf_trace_get_environment_field_name */ + ok(bt_ctf_trace_get_environment_field_name(NULL, 0) == NULL, + "bt_ctf_trace_get_environment_field_name handles a NULL trace correctly"); + ok(bt_ctf_trace_get_environment_field_name(trace, -1) == NULL, + "bt_ctf_trace_get_environment_field_name handles an invalid index correctly (negative)"); + ok(bt_ctf_trace_get_environment_field_name(trace, 5) == NULL, + "bt_ctf_trace_get_environment_field_name handles an invalid index correctly (too large)"); + ret_string = bt_ctf_trace_get_environment_field_name(trace, 0); + ok(ret_string && !strcmp(ret_string, "host"), + "bt_ctf_trace_get_environment_field_name returns a correct field name"); + ret_string = bt_ctf_trace_get_environment_field_name(trace, 1); + ok(ret_string && !strcmp(ret_string, "test_env_int_obj"), + "bt_ctf_trace_get_environment_field_name returns a correct field name"); + ret_string = bt_ctf_trace_get_environment_field_name(trace, 2); + ok(ret_string && !strcmp(ret_string, "test_env_str_obj"), + "bt_ctf_trace_get_environment_field_name returns a correct field name"); + ret_string = bt_ctf_trace_get_environment_field_name(trace, 3); + ok(ret_string && !strcmp(ret_string, "test_env_int"), + "bt_ctf_trace_get_environment_field_name returns a correct field name"); + ret_string = bt_ctf_trace_get_environment_field_name(trace, 4); + ok(ret_string && !strcmp(ret_string, "test_env_str"), + "bt_ctf_trace_get_environment_field_name returns a correct field name"); + + /* Test bt_ctf_trace_get_environment_field_value */ + ok(bt_ctf_trace_get_environment_field_value(NULL, 0) == NULL, + "bt_ctf_trace_get_environment_field_value handles a NULL trace correctly"); + ok(bt_ctf_trace_get_environment_field_value(trace, -1) == NULL, + "bt_ctf_trace_get_environment_field_value handles an invalid index correctly (negative)"); + ok(bt_ctf_trace_get_environment_field_value(trace, 5) == NULL, + "bt_ctf_trace_get_environment_field_value handles an invalid index correctly (too large)"); + obj = bt_ctf_trace_get_environment_field_value(trace, 1); + ret = bt_object_integer_get(obj, &ret_int64_t); + ok(!ret && ret_int64_t == 23, + "bt_ctf_trace_get_environment_field_value succeeds in getting an integer value"); + BT_OBJECT_PUT(obj); + obj = bt_ctf_trace_get_environment_field_value(trace, 2); + ret = bt_object_string_get(obj, &ret_string); + ok(!ret && ret_string && !strcmp(ret_string, "the value"), + "bt_ctf_trace_get_environment_field_value succeeds in getting a string value"); + BT_OBJECT_PUT(obj); + + /* Test bt_ctf_trace_get_environment_field_value_by_name */ + ok(!bt_ctf_trace_get_environment_field_value_by_name(NULL, + "test_env_str"), + "bt_ctf_trace_get_environment_field_value_by_name handles a NULL trace correctly"); + ok(!bt_ctf_trace_get_environment_field_value_by_name(trace, NULL), + "bt_ctf_trace_get_environment_field_value_by_name handles a NULL name correctly"); + ok(!bt_ctf_trace_get_environment_field_value_by_name(trace, "oh oh"), + "bt_ctf_trace_get_environment_field_value_by_name returns NULL or an unknown field name"); + obj = bt_ctf_trace_get_environment_field_value_by_name(trace, + "test_env_str"); + ret = bt_object_string_get(obj, &ret_string); + ok(!ret && ret_string && !strcmp(ret_string, "oh yeah"), + "bt_ctf_trace_get_environment_field_value_by_name succeeds in getting an existing field"); + BT_OBJECT_PUT(obj); + + /* Test environment field replacement */ + ok(!bt_ctf_trace_set_environment_field_integer(trace, "test_env_int", + 654321), + "bt_ctf_trace_set_environment_field_integer succeeds with an existing name"); + ok(bt_ctf_trace_get_environment_field_count(trace) == 5, + "bt_ctf_trace_set_environment_field_integer with an existing key does not increase the environment size"); + obj = bt_ctf_trace_get_environment_field_value(trace, 3); + ret = bt_object_integer_get(obj, &ret_int64_t); + ok(!ret && ret_int64_t == 654321, + "bt_ctf_trace_get_environment_field_value successfully replaces an existing field"); + BT_OBJECT_PUT(obj); + if (uname(&name)) { perror("uname"); return -1; @@ -1677,11 +2082,6 @@ 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, @@ -1696,6 +2096,18 @@ int main(int argc, char **argv) ok(ret_clock == clock, "bt_ctf_trace_get_clock returns the right clock instance"); bt_ctf_clock_put(ret_clock); + ok(!bt_ctf_trace_get_clock_by_name(trace, NULL), + "bt_ctf_trace_get_clock_by_name correctly handles NULL (trace)"); + ok(!bt_ctf_trace_get_clock_by_name(NULL, clock_name), + "bt_ctf_trace_get_clock_by_name correctly handles NULL (clock name)"); + ok(!bt_ctf_trace_get_clock_by_name(NULL, NULL), + "bt_ctf_trace_get_clock_by_name correctly handles NULL (both)"); + ret_clock = bt_ctf_trace_get_clock_by_name(trace, clock_name); + ok(ret_clock == clock, + "bt_ctf_trace_get_clock_by_name returns the right clock instance"); + bt_ctf_clock_put(ret_clock); + ok(!bt_ctf_trace_get_clock_by_name(trace, "random"), + "bt_ctf_trace_get_clock_by_name fails when the requested clock doesn't exist"); ok(!bt_ctf_clock_get_name(NULL), "bt_ctf_clock_get_name correctly handles NULL"); @@ -1757,7 +2169,7 @@ int main(int argc, char **argv) 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"), + ok(ret_string && !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, @@ -1787,6 +2199,35 @@ int main(int argc, char **argv) ok(bt_ctf_stream_class_get_id(stream_class) == 123, "bt_ctf_stream_class_get_id returns the correct value"); + /* Validate default event header fields */ + ok(bt_ctf_stream_class_get_event_header_type(NULL) == NULL, + "bt_ctf_stream_class_get_event_header_type handles NULL correctly"); + ret_field_type = bt_ctf_stream_class_get_event_header_type( + stream_class); + ok(ret_field_type, + "bt_ctf_stream_class_get_event_header_type returns an event header type"); + ok(bt_ctf_field_type_get_type_id(ret_field_type) == CTF_TYPE_STRUCT, + "Default event header type is a structure"); + event_header_field_type = + bt_ctf_field_type_structure_get_field_type_by_name( + ret_field_type, "id"); + ok(event_header_field_type, + "Default event header type contains an \"id\" field"); + ok(bt_ctf_field_type_get_type_id( + event_header_field_type) == CTF_TYPE_INTEGER, + "Default event header \"id\" field is an integer"); + bt_ctf_field_type_put(event_header_field_type); + event_header_field_type = + bt_ctf_field_type_structure_get_field_type_by_name( + ret_field_type, "timestamp"); + ok(event_header_field_type, + "Default event header type contains a \"timestamp\" field"); + ok(bt_ctf_field_type_get_type_id( + event_header_field_type) == CTF_TYPE_INTEGER, + "Default event header \"timestamp\" field is an integer"); + bt_ctf_field_type_put(event_header_field_type); + bt_ctf_field_type_put(ret_field_type); + /* Add a custom trace packet header field */ ok(bt_ctf_trace_get_packet_header_type(NULL) == NULL, "bt_ctf_trace_get_packet_header_type handles NULL correctly"); @@ -1820,9 +2261,6 @@ int main(int argc, char **argv) ok(!bt_ctf_trace_set_packet_header_type(trace, packet_header_type), "Set a trace packet_header_type successfully"); - /* 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"); @@ -1842,6 +2280,9 @@ int main(int argc, char **argv) ok(bt_ctf_stream_class_set_packet_context_type(stream_class, integer_type) < 0, "bt_ctf_stream_class_set_packet_context_type rejects a packet context that is not a structure"); + /* Create a "uint5_t" equivalent custom packet context field */ + packet_context_field_type = bt_ctf_field_type_integer_create(5); + ret = bt_ctf_field_type_structure_add_field(packet_context_type, packet_context_field_type, "custom_packet_context_field"); ok(ret == 0, "Packet context field added successfully"); @@ -1946,12 +2387,19 @@ 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); packet_resize_test(stream_class, stream1, clock); 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); @@ -1962,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); @@ -1978,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) { @@ -1994,6 +2445,5 @@ int main(int argc, char **argv) rmdir(trace_path); closedir(trace_dir); - return 0; }