#include <babeltrace/ctf-writer/event-fields.h>
#include <babeltrace/ctf-ir/stream-class.h>
#include <babeltrace/ctf/events.h>
+#include <babeltrace/objects.h>
#include <unistd.h>
#include <stdlib.h>
#include <stdio.h>
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");
ok(bt_ctf_writer_add_environment_field(writer, "test_field",
NULL),
"bt_ctf_writer_add_environment_field error with NULL field value");
- ok(bt_ctf_trace_add_environment_field_integer(NULL, "test_env", 0),
- "bt_ctf_trace_add_environment_field_integer handles a NULL trace correctly");
- ok(bt_ctf_trace_add_environment_field_integer(trace, NULL, 0),
- "bt_ctf_trace_add_environment_field_integer handles a NULL environment field name");
- ok(!bt_ctf_trace_add_environment_field_integer(trace, "test_env",
- 123456),
- "Add an integer environment field to a trace instance");
+
+ /* 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) == 2,
+ 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_type */
- ok(bt_ctf_trace_get_environment_field_type(trace, 2) ==
- BT_ENVIRONMENT_FIELD_TYPE_UNKNOWN,
- "bt_ctf_trace_get_environment_field_type handles an invalid index correctly");
- ok(bt_ctf_trace_get_environment_field_type(NULL, 0) ==
- BT_ENVIRONMENT_FIELD_TYPE_UNKNOWN,
- "bt_ctf_trace_get_environment_field_type handles a NULL trace correctly");
- ok(bt_ctf_trace_get_environment_field_type(trace, 1) ==
- BT_ENVIRONMENT_FIELD_TYPE_INTEGER,
- "bt_ctf_trace_get_environment_field_type the correct type of environment field");
-
/* 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");
+ "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_string */
- ok(bt_ctf_trace_get_environment_field_value_string(NULL, 0) == NULL,
- "bt_ctf_trace_get_environment_field_value_string handles a NULL trace correctly");
- ok(bt_ctf_trace_get_environment_field_value_string(trace, -1) == NULL,
- "bt_ctf_trace_get_environment_field_value_string handles an invalid index correctly");
- ok(bt_ctf_trace_get_environment_field_value_string(trace, 1) == NULL,
- "bt_ctf_trace_get_environment_field_value_string validates environment field type");
- ret_string = bt_ctf_trace_get_environment_field_value_string(trace, 0);
- ok(ret_string && !strcmp(ret_string, hostname),
- "bt_ctf_trace_get_environment_field_value_string returns a correct value");
-
- /* Test bt_ctf_trace_get_environment_field_value_integer */
- ok(bt_ctf_trace_get_environment_field_value_integer(NULL, 0, &ret_int64_t) < 0,
- "bt_ctf_trace_get_environment_field_value_integer handles a NULL trace correctly");
- ok(bt_ctf_trace_get_environment_field_value_integer(trace, 42, &ret_int64_t) < 0,
- "bt_ctf_trace_get_environment_field_value_integer handles an invalid index correctly");
- ok(bt_ctf_trace_get_environment_field_value_integer(trace, 1, NULL) < 0,
- "bt_ctf_trace_get_environment_field_value_integer handles a NULL value argument correctly");
- ok(bt_ctf_trace_get_environment_field_value_integer(trace, 0, &ret_int64_t) < 0,
- "bt_ctf_trace_get_environment_field_value_integer validates environment field type");
- ok(!bt_ctf_trace_get_environment_field_value_integer(trace, 1, &ret_int64_t),
- "bt_ctf_trace_get_environment_field_value_integer returns a value");
- ok(ret_int64_t == 123456,
- "bt_ctf_trace_get_environment_field_value_integer returned a correct value");
+ /* 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");