X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=tests%2Flib%2Ftest_ctf_writer.c;h=e6120d2242e8adf7d47798d3db9c1ea3c480d6db;hb=3b3b162e3dee8ebc5f5d4a99089ce83befff2ab2;hp=608172cf926ce9096ab9623fd3a57c0d9888776b;hpb=5edae678ef55c4242b659e03becb5d6424898271;p=babeltrace.git diff --git a/tests/lib/test_ctf_writer.c b/tests/lib/test_ctf_writer.c index 608172cf..e6120d22 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,24 +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*/ - bt_ctf_field_type_structure_add_field(event_context_type, uint_12_type, - "event_specific_context"); - 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"); @@ -639,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 = @@ -674,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); @@ -827,6 +854,112 @@ 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); + + assert(obj = bt_object_integer_create_init(5)); + 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); + assert(obj = bt_object_string_create_init("http://kernel.org/")); + 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"); @@ -1479,6 +1612,146 @@ end: bt_ctf_event_class_put(event_class); } +void test_empty_stream(struct bt_ctf_writer *writer) +{ + int ret = 0; + struct bt_ctf_trace *trace = NULL; + struct bt_ctf_stream_class *stream_class = NULL; + struct bt_ctf_stream *stream = NULL; + + trace = bt_ctf_writer_get_trace(writer); + if (!trace) { + diag("Failed to get trace from writer"); + ret = -1; + goto end; + } + + stream_class = bt_ctf_stream_class_create("empty_stream"); + if (!stream_class) { + diag("Failed to create stream class"); + ret = -1; + goto end; + } + + stream = bt_ctf_writer_create_stream(writer, stream_class); + if (!stream) { + diag("Failed to create writer stream"); + ret = -1; + goto end; + } +end: + ok(ret == 0, + "Created a stream class with default attributes and an empty stream"); + bt_ctf_trace_put(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; + 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; + } + + ret = bt_ctf_stream_append_event(stream, event); + if (ret) { + diag("Failed to append event to stream"); + goto end; + } +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_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); +} + int main(int argc, char **argv) { char trace_path[] = "/tmp/ctfwriter_XXXXXX"; @@ -1497,18 +1770,24 @@ int main(int argc, char **argv) struct utsname name; char hostname[BABELTRACE_HOST_NAME_MAX]; struct bt_ctf_clock *clock, *ret_clock; - struct bt_ctf_stream_class *stream_class; + struct bt_ctf_stream_class *stream_class, *ret_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, + *packet_header_type, + *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"); @@ -1527,6 +1806,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) { @@ -1545,6 +1834,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; @@ -1636,11 +2042,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, @@ -1655,6 +2056,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"); @@ -1716,7 +2129,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, @@ -1746,8 +2159,67 @@ 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"); - /* Create a "uint5_t" equivalent custom packet context field */ - packet_context_field_type = bt_ctf_field_type_integer_create(5); + /* 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"); + packet_header_type = bt_ctf_trace_get_packet_header_type(trace); + ok(packet_header_type, + "bt_ctf_trace_get_packet_header_type returns a packet header"); + ok(bt_ctf_field_type_get_type_id(packet_header_type) == CTF_TYPE_STRUCT, + "bt_ctf_trace_get_packet_header_type returns a packet header of type struct"); + ret_field_type = bt_ctf_field_type_structure_get_field_type_by_name( + packet_header_type, "magic"); + ok(ret_field_type, "Default packet header type contains a \"magic\" field"); + bt_ctf_field_type_put(ret_field_type); + ret_field_type = bt_ctf_field_type_structure_get_field_type_by_name( + packet_header_type, "uuid"); + ok(ret_field_type, "Default packet header type contains a \"uuid\" field"); + bt_ctf_field_type_put(ret_field_type); + ret_field_type = bt_ctf_field_type_structure_get_field_type_by_name( + packet_header_type, "stream_id"); + ok(ret_field_type, "Default packet header type contains a \"stream_id\" field"); + bt_ctf_field_type_put(ret_field_type); + + packet_header_field_type = bt_ctf_field_type_integer_create(22); + ok(!bt_ctf_field_type_structure_add_field(packet_header_type, + packet_header_field_type, "custom_trace_packet_header_field"), + "Added a custom trace packet header field successfully"); + + ok(bt_ctf_trace_set_packet_header_type(NULL, packet_header_type) < 0, + "bt_ctf_trace_set_packet_header_type handles a NULL trace correctly"); + ok(bt_ctf_trace_set_packet_header_type(trace, NULL) < 0, + "bt_ctf_trace_set_packet_header_type handles a NULL packet_header_type correctly"); + ok(!bt_ctf_trace_set_packet_header_type(trace, packet_header_type), + "Set a trace packet_header_type successfully"); ok(bt_ctf_stream_class_get_packet_context_type(NULL) == NULL, "bt_ctf_stream_class_get_packet_context_type handles NULL correctly"); @@ -1768,6 +2240,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"); @@ -1799,11 +2274,29 @@ int main(int argc, char **argv) stream_class); ok(ret_field_type == stream_event_context_type, "bt_ctf_stream_class_get_event_context_type returns the correct field type."); + bt_ctf_field_type_put(ret_field_type); /* Instantiate a stream and append events */ stream1 = bt_ctf_writer_create_stream(writer, stream_class); ok(stream1, "Instanciate a stream class from writer"); + ok(bt_ctf_stream_get_class(NULL) == NULL, + "bt_ctf_stream_get_class correctly handles NULL"); + ret_stream_class = bt_ctf_stream_get_class(stream1); + ok(ret_stream_class, + "bt_ctf_stream_get_class returns a stream class"); + ok(ret_stream_class == stream_class, + "Returned stream class is of the correct type"); + + /* + * Try to modify the packet context type after a stream has been + * created. + */ + ret = bt_ctf_field_type_structure_add_field(packet_header_type, + packet_header_field_type, "should_fail"); + ok(ret < 0, + "Trace packet header type can't be modified once a stream has been instanciated"); + /* * Try to modify the packet context type after a stream has been * created. @@ -1811,7 +2304,7 @@ int main(int argc, char **argv) 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"); + "Packet context type can't be modified once a stream has been instanciated"); /* * Try to modify the stream event context type after a stream has been @@ -1820,18 +2313,50 @@ int main(int argc, char **argv) ret = bt_ctf_field_type_structure_add_field(stream_event_context_type, integer_type, "should_fail"); ok(ret < 0, - "Stream event context type can't be modified once a stream class has been instanciated"); + "Stream event context type can't be modified once a stream has been instanciated"); /* Should fail after instanciating a stream (frozen) */ ok(bt_ctf_stream_class_set_clock(stream_class, clock), "Changes to a stream class that was already instantiated fail"); + /* Populate the custom packet header field only once for all tests */ + ok(bt_ctf_stream_get_packet_header(NULL) == NULL, + "bt_ctf_stream_get_packet_header handles NULL correctly"); + packet_header = bt_ctf_stream_get_packet_header(stream1); + ok(packet_header, + "bt_ctf_stream_get_packet_header returns a packet header"); + ret_field_type = bt_ctf_field_get_type(packet_header); + ok(ret_field_type == packet_header_type, + "Stream returns a packet header of the appropriate type"); + bt_ctf_field_type_put(ret_field_type); + packet_header_field = bt_ctf_field_structure_get_field(packet_header, + "custom_trace_packet_header_field"); + ok(packet_header_field, + "Packet header structure contains a custom field with the appropriate name"); + ret_field_type = bt_ctf_field_get_type(packet_header_field); + ok(ret_field_type == packet_header_field_type, + "Custom packet header field is of the expected type"); + ok(!bt_ctf_field_unsigned_integer_set_value(packet_header_field, + 54321), "Set custom packet header value successfully"); + ok(bt_ctf_stream_set_packet_header(stream1, NULL) < 0, + "bt_ctf_stream_set_packet_header handles a NULL packet header correctly"); + ok(bt_ctf_stream_set_packet_header(NULL, packet_header) < 0, + "bt_ctf_stream_set_packet_header handles a NULL stream correctly"); + ok(bt_ctf_stream_set_packet_header(stream1, packet_header_field) < 0, + "bt_ctf_stream_set_packet_header rejects a packet header of the wrong type"); + ok(!bt_ctf_stream_set_packet_header(stream1, packet_header), + "Successfully set a stream's packet header"); + + 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); + test_empty_stream(writer); + metadata_string = bt_ctf_writer_get_metadata_string(writer); ok(metadata_string, "Get metadata string"); @@ -1841,6 +2366,7 @@ int main(int argc, char **argv) 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); bt_ctf_field_type_put(packet_context_type); @@ -1848,6 +2374,10 @@ int main(int argc, char **argv) bt_ctf_field_type_put(integer_type); bt_ctf_field_type_put(stream_event_context_type); bt_ctf_field_type_put(ret_field_type); + bt_ctf_field_type_put(packet_header_type); + bt_ctf_field_type_put(packet_header_field_type); + bt_ctf_field_put(packet_header); + bt_ctf_field_put(packet_header_field); bt_ctf_trace_put(trace); free(metadata_string); @@ -1867,6 +2397,5 @@ int main(int argc, char **argv) rmdir(trace_path); closedir(trace_dir); - return 0; }