ir: move bt_ctf_stream_class_set_clock() to writer header
[babeltrace.git] / tests / lib / test_ctf_writer.c
index 22a7c3e6b44e18630d57199bd46e90b5dc646648..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>
@@ -58,7 +58,7 @@
 #define DEFAULT_CLOCK_TIME 0
 #define DEFAULT_CLOCK_VALUE 0
 
-#define NR_TESTS 584
+#define NR_TESTS 590
 
 static int64_t current_time = 42;
 
@@ -2611,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";
@@ -2918,38 +3066,6 @@ int main(int argc, char **argv)
        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),
@@ -3170,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);
@@ -3257,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.027589 seconds and 4 git commands to generate.