ir: move bt_ctf_stream_class_set_clock() to writer header
[babeltrace.git] / tests / lib / test_ctf_writer.c
index 576c4cf023320dafd08a1fcd383ffad6b3e8dbd3..eefee00eb0d8806c13b163115c4f2891b84098bb 100644 (file)
@@ -25,7 +25,7 @@
 #include <babeltrace/ctf-writer/event.h>
 #include <babeltrace/ctf-writer/event-types.h>
 #include <babeltrace/ctf-writer/event-fields.h>
-#include <babeltrace/ctf-ir/stream-class.h>
+#include <babeltrace/ctf-writer/stream-class.h>
 #include <babeltrace/ref.h>
 #include <babeltrace/ctf/events.h>
 #include <babeltrace/values.h>
@@ -56,8 +56,9 @@
 #define DEFAULT_CLOCK_OFFSET_S 0
 #define DEFAULT_CLOCK_IS_ABSOLUTE 0
 #define DEFAULT_CLOCK_TIME 0
+#define DEFAULT_CLOCK_VALUE 0
 
-#define NR_TESTS 594
+#define NR_TESTS 590
 
 static int64_t current_time = 42;
 
@@ -280,89 +281,6 @@ close_fp:
        }
 }
 
-void event_copy_tests(struct bt_ctf_event *event)
-{
-       struct bt_ctf_event *copy;
-       struct bt_ctf_event_class *orig_event_class;
-       struct bt_ctf_event_class *copy_event_class;
-       struct bt_ctf_stream *orig_stream;
-       struct bt_ctf_stream *copy_stream;
-       struct bt_ctf_field *orig_field;
-       struct bt_ctf_field *copy_field;
-
-       /* copy */
-       ok(!bt_ctf_event_copy(NULL),
-               "bt_ctf_event_copy handles NULL correctly");
-       copy = bt_ctf_event_copy(event);
-       ok(copy, "bt_ctf_event_copy returns a valid pointer");
-
-       /* validate event class */
-       orig_event_class = bt_ctf_event_get_class(event);
-       assert(orig_event_class);
-       copy_event_class = bt_ctf_event_get_class(copy);
-       ok(orig_event_class == copy_event_class,
-               "original and copied events share the same event class pointer");
-       bt_put(orig_event_class);
-       bt_put(copy_event_class);
-
-       /* validate stream */
-       orig_stream = bt_ctf_event_get_stream(event);
-       copy_stream = bt_ctf_event_get_stream(copy);
-
-       if (!orig_stream) {
-               ok(!copy_stream, "original and copied events have no stream");
-       } else {
-               ok(orig_stream == copy_stream,
-                       "original and copied events share the same stream pointer");
-       }
-       bt_put(orig_stream);
-       bt_put(copy_stream);
-
-       /* header */
-       orig_field = bt_ctf_event_get_header(event);
-       copy_field = bt_ctf_event_get_header(copy);
-
-       if (!orig_field) {
-               ok(!copy_field, "original and copied events have no header");
-       } else {
-               ok(orig_field != copy_field,
-                       "original and copied events headers are different pointers");
-       }
-
-       bt_put(orig_field);
-       bt_put(copy_field);
-
-       /* context */
-       orig_field = bt_ctf_event_get_event_context(event);
-       copy_field = bt_ctf_event_get_event_context(copy);
-
-       if (!orig_field) {
-               ok(!copy_field, "original and copied events have no context");
-       } else {
-               ok(orig_field != copy_field,
-                       "original and copied events contexts are different pointers");
-       }
-
-       bt_put(orig_field);
-       bt_put(copy_field);
-
-       /* payload */
-       orig_field = bt_ctf_event_get_payload_field(event);
-       copy_field = bt_ctf_event_get_payload_field(copy);
-
-       if (!orig_field) {
-               ok(!copy_field, "original and copied events have no payload");
-       } else {
-               ok(orig_field != copy_field,
-                       "original and copied events payloads are different pointers");
-       }
-
-       bt_put(orig_field);
-       bt_put(copy_field);
-
-       bt_put(copy);
-}
-
 void append_simple_event(struct bt_ctf_stream_class *stream_class,
                struct bt_ctf_stream *stream, struct bt_ctf_clock *clock)
 {
@@ -694,7 +612,9 @@ void append_simple_event(struct bt_ctf_stream_class *stream_class,
        ok(bt_ctf_clock_set_time(clock, current_time) == 0, "Set clock time");
 
        /* Populate stream event context */
-       stream_event_context = bt_ctf_stream_get_event_context(stream);
+       stream_event_context =
+               bt_ctf_event_get_stream_event_context(simple_event);
+       assert(stream_event_context);
        stream_event_context_field = bt_ctf_field_structure_get_field(
                stream_event_context, "common_event_context");
        bt_ctf_field_unsigned_integer_set_value(stream_event_context_field, 42);
@@ -721,7 +641,6 @@ void append_simple_event(struct bt_ctf_stream_class *stream_class,
        ok(!bt_ctf_event_set_event_context(simple_event, event_context),
                "Set an event context successfully");
 
-       event_copy_tests(simple_event);
        ok(bt_ctf_stream_append_event(stream, simple_event) == 0,
                "Append simple event to trace stream");
 
@@ -824,7 +743,8 @@ 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, *an_array_field, *ret_field;
+               *variant_field, *an_array_field, *stream_event_ctx_field,
+               *stream_event_ctx_int_field, *ret_field;
        uint64_t ret_unsigned_int;
        int64_t ret_signed_int;
        const char *ret_string;
@@ -1306,6 +1226,14 @@ void append_complex_event(struct bt_ctf_stream_class *stream_class,
                bt_put(int_16_field);
        }
 
+       stream_event_ctx_field = bt_ctf_event_get_stream_event_context(event);
+       assert(stream_event_ctx_field);
+       stream_event_ctx_int_field = bt_ctf_field_structure_get_field(
+               stream_event_ctx_field, "common_event_context");
+       BT_PUT(stream_event_ctx_field);
+       bt_ctf_field_unsigned_integer_set_value(stream_event_ctx_int_field, 17);
+       BT_PUT(stream_event_ctx_int_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");
@@ -2180,7 +2108,7 @@ void packet_resize_test(struct bt_ctf_stream_class *stream_class,
        uint64_t ret_uint64;
        int events_appended = 0;
        struct bt_ctf_field *packet_context = NULL,
-               *packet_context_field = NULL, *event_context = NULL;
+               *packet_context_field = NULL, *stream_event_context = NULL;
        struct bt_ctf_field_type *ep_field_1_type = NULL;
        struct bt_ctf_field_type *ep_a_string_type = NULL;
        struct bt_ctf_field_type *ep_type = NULL;
@@ -2223,22 +2151,6 @@ void packet_resize_test(struct bt_ctf_stream_class *stream_class,
                "bt_ctf_event_get_payload_by_index handles an invalid index correctly");
        bt_put(event);
 
-       ok(bt_ctf_stream_get_event_context(NULL) == NULL,
-               "bt_ctf_stream_get_event_context handles NULL correctly");
-       event_context = bt_ctf_stream_get_event_context(stream);
-       ok(event_context,
-               "bt_ctf_stream_get_event_context returns a stream event context");
-       ok(bt_ctf_stream_set_event_context(NULL, event_context) < 0,
-               "bt_ctf_stream_set_event_context handles a NULL stream correctly");
-       ok(bt_ctf_stream_set_event_context(stream, NULL) < 0,
-               "bt_ctf_stream_set_event_context handles a NULL stream event context correctly");
-       ok(!bt_ctf_stream_set_event_context(stream, event_context),
-               "bt_ctf_stream_set_event_context correctly set a stream event context");
-       ret_field = bt_ctf_field_create(ep_field_1_type);
-       ok(bt_ctf_stream_set_event_context(stream, ret_field) < 0,
-               "bt_ctf_stream_set_event_context rejects an event context of incorrect type");
-       bt_put(ret_field);
-
        for (i = 0; i < PACKET_RESIZE_TEST_LENGTH; i++) {
                event = bt_ctf_event_create(event_class);
                struct bt_ctf_field *integer =
@@ -2257,8 +2169,11 @@ void packet_resize_test(struct bt_ctf_stream_class *stream_class,
                bt_put(string);
 
                /* Populate stream event context */
-               integer = bt_ctf_field_structure_get_field(event_context,
+               stream_event_context =
+                       bt_ctf_event_get_stream_event_context(event);
+               integer = bt_ctf_field_structure_get_field(stream_event_context,
                        "common_event_context");
+               BT_PUT(stream_event_context);
                ret |= bt_ctf_field_unsigned_integer_set_value(integer,
                        i % 42);
                bt_put(integer);
@@ -2305,7 +2220,7 @@ end:
        bt_put(string_type);
        bt_put(packet_context);
        bt_put(packet_context_field);
-       bt_put(event_context);
+       bt_put(stream_event_context);
        bt_put(event_class);
        bt_put(ep_field_1_type);
        bt_put(ep_a_string_type);
@@ -2696,6 +2611,154 @@ void append_existing_event_class(struct bt_ctf_stream_class *stream_class)
        bt_put(event_class);
 }
 
+void test_trace_stream_class_clock(void)
+{
+       struct bt_ctf_trace *trace = NULL;
+       struct bt_ctf_stream_class *sc1 = NULL;
+       struct bt_ctf_stream_class *sc2 = NULL;
+       struct bt_ctf_clock *sc1_clock = NULL;
+       struct bt_ctf_clock *sc2_clock = NULL;
+       const char *clock_name = "hello";
+
+       trace = bt_ctf_trace_create();
+       assert(trace);
+       sc1 = bt_ctf_stream_class_create(NULL);
+       assert(sc1);
+       sc2 = bt_ctf_stream_class_create(NULL);
+       assert(sc2);
+       sc1_clock = bt_ctf_clock_create(clock_name);
+       assert(sc1_clock);
+       sc2_clock = bt_ctf_clock_create(clock_name);
+       assert(sc2_clock);
+
+       ok(!bt_ctf_stream_class_set_clock(sc1, sc1_clock),
+               "bt_ctf_stream_class_set_clock() succeeds for sc1");
+       ok(!bt_ctf_stream_class_set_clock(sc2, sc2_clock),
+               "bt_ctf_stream_class_set_clock() succeeds for sc2");
+       ok(!bt_ctf_trace_add_stream_class(trace, sc1),
+               "bt_ctf_trace_add_stream_class() succeeds with sc1");
+       ok(bt_ctf_trace_add_stream_class(trace, sc2),
+               "bt_ctf_trace_add_stream_class() fails with sc2 (different clock, same name)");
+
+       BT_PUT(trace);
+       BT_PUT(sc1);
+       BT_PUT(sc2);
+       BT_PUT(sc1_clock);
+       BT_PUT(sc2_clock);
+}
+
+static
+struct bt_ctf_event_class *create_minimal_event_class(void)
+{
+       struct bt_ctf_event_class *ec = NULL;
+       struct bt_ctf_field_type *int_ft = NULL;
+       int ret;
+
+       int_ft = bt_ctf_field_type_integer_create(23);
+       assert(int_ft);
+       ec = bt_ctf_event_class_create("minimal");
+       assert(ec);
+       ret = bt_ctf_event_class_add_field(ec, int_ft, "field");
+       assert(!ret);
+       BT_PUT(int_ft);
+
+       return ec;
+}
+
+static
+void test_create_writer_stream_from_stream_class(void)
+{
+       int ret;
+       char trace_path[] = "/tmp/ctfwriter_XXXXXX";
+       const char *writer_stream_name = "writer stream instance";
+       struct bt_ctf_writer *writer = NULL;
+       struct bt_ctf_trace *writer_trace = NULL;
+       struct bt_ctf_stream_class *writer_sc = NULL;
+       struct bt_ctf_stream *writer_stream = NULL;
+       struct bt_ctf_trace *non_writer_trace = NULL;
+       struct bt_ctf_stream_class *non_writer_sc = NULL;
+       struct bt_ctf_stream *non_writer_stream = NULL;
+       struct bt_ctf_event_class *ec = NULL;
+       struct bt_ctf_event *event = NULL;
+       struct bt_ctf_field_type *empty_struct_ft = NULL;
+       struct bt_ctf_field *int_field = NULL;
+
+       if (!bt_mkdtemp(trace_path)) {
+               perror("# perror");
+       }
+
+       /* Create empty structure field type (event header) */
+       empty_struct_ft = bt_ctf_field_type_structure_create();
+       assert(empty_struct_ft);
+
+       /* Create writer, writer stream class, and writer stream */
+       writer = bt_ctf_writer_create(trace_path);
+       assert(writer);
+       writer_trace = bt_ctf_writer_get_trace(writer);
+       ok(writer_trace, "bt_ctf_writer_get_trace() returns a trace");
+       writer_sc = bt_ctf_stream_class_create("writer_sc");
+       assert(writer_sc);
+       ret = bt_ctf_stream_class_set_event_header_type(writer_sc,
+               empty_struct_ft);
+       assert(!ret);
+       ret = bt_ctf_trace_add_stream_class(writer_trace, writer_sc);
+       assert(!ret);
+       writer_stream = bt_ctf_stream_create(writer_sc, writer_stream_name);
+       assert(writer_stream);
+       ok(!strcmp(bt_ctf_stream_get_name(writer_stream), writer_stream_name),
+               "bt_ctf_stream_get_name() returns the stream's name");
+
+       /* Create non-writer trace, stream class, and stream */
+       non_writer_trace = bt_ctf_trace_create();
+       assert(non_writer_trace);
+       non_writer_sc = bt_ctf_stream_class_create("nonwriter_sc");
+       assert(non_writer_sc);
+       ret = bt_ctf_stream_class_set_event_header_type(non_writer_sc,
+               empty_struct_ft);
+       assert(!ret);
+       ret = bt_ctf_trace_add_stream_class(non_writer_trace, non_writer_sc);
+       assert(!ret);
+       non_writer_stream = bt_ctf_stream_create(non_writer_sc, NULL);
+       assert(non_writer_stream);
+
+       /* Create event class and event */
+       ec = create_minimal_event_class();
+       assert(ec);
+       ret = bt_ctf_stream_class_add_event_class(writer_sc, ec);
+       assert(!ret);
+       event = bt_ctf_event_create(ec);
+       assert(event);
+       int_field = bt_ctf_event_get_payload_by_index(event, 0);
+       assert(int_field);
+       bt_ctf_field_unsigned_integer_set_value(int_field, 17);
+
+       /*
+        * Verify non-writer stream: it should be impossible to append
+        * an event to it.
+        */
+       ok(bt_ctf_stream_append_event(non_writer_stream, event),
+               "bt_ctf_stream_append_event() fails with a non-writer stream");
+
+       /*
+        * Verify writer stream: it should be possible to append an
+        * event to it.
+        */
+       ok(!bt_ctf_stream_append_event(writer_stream, event),
+               "bt_ctf_stream_append_event() succeeds with a writer stream");
+
+       BT_PUT(writer);
+       BT_PUT(writer_trace);
+       BT_PUT(writer_sc);
+       BT_PUT(writer_stream);
+       BT_PUT(non_writer_trace);
+       BT_PUT(non_writer_sc);
+       BT_PUT(non_writer_stream);
+       BT_PUT(ec);
+       BT_PUT(event);
+       BT_PUT(int_field);
+       BT_PUT(empty_struct_ft);
+}
+
 int main(int argc, char **argv)
 {
        char trace_path[] = "/tmp/ctfwriter_XXXXXX";
@@ -2990,45 +3053,19 @@ int main(int argc, char **argv)
                "bt_ctf_clock_get_time succeeds");
        ok(get_time == DEFAULT_CLOCK_TIME,
                "bt_ctf_clock_get_time returns the correct default time");
+       ok(bt_ctf_clock_get_value(clock) == DEFAULT_CLOCK_VALUE,
+               "bt_ctf_clock_get_value returns the correct default value");
+       ok(bt_ctf_clock_set_value(clock, current_time) == 0,
+               "Set clock value");
+       ok(bt_ctf_clock_get_value(clock) == current_time,
+               "bt_ctf_clock_get_value returns the correct value once it is set");
        ok(bt_ctf_clock_set_time(clock, current_time) == 0,
                "Set clock time");
        ok(bt_ctf_clock_get_time(clock, &get_time) == 0,
                "bt_ctf_clock_get_time succeeds");
-       ok(get_time == current_time,
+       ok(get_time >= current_time - 1 && get_time <= current_time + 1,
                "bt_ctf_clock_get_time returns the correct time once it is set");
 
-       ok(bt_ctf_writer_add_clock(writer, clock) == 0,
-               "Add clock to writer instance");
-       ok(bt_ctf_writer_add_clock(writer, clock),
-               "Verify a clock can't be added twice to a writer instance");
-
-       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_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_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");
        ok(!bt_ctf_clock_get_description(NULL),
@@ -3249,6 +3286,38 @@ int main(int argc, char **argv)
        stream1 = bt_ctf_writer_create_stream(writer, stream_class);
        ok(stream1, "Instanciate a stream class from writer");
 
+       /*
+        * Creating a stream through a writer adds the given stream
+        * class to the writer's trace, thus registering the stream
+        * class's clock to the trace.
+        */
+       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_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_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_stream_get_class(NULL) == NULL,
                "bt_ctf_stream_get_class correctly handles NULL");
        ret_stream_class = bt_ctf_stream_get_class(stream1);
@@ -3336,6 +3405,10 @@ int main(int argc, char **argv)
        ok(bt_ctf_writer_add_environment_field(writer, "new_field", "test") == 0,
                "Add environment field to writer after stream creation");
 
+       test_trace_stream_class_clock();
+
+       test_create_writer_stream_from_stream_class();
+
        test_instanciate_event_before_stream(writer);
 
        append_simple_event(stream_class, stream1, clock);
This page took 0.028809 seconds and 4 git commands to generate.