From: Jérémie Galarneau <jeremie.galarneau@efficios.com>
Date: Fri, 20 Mar 2015 02:35:23 +0000 (-0400)
Subject: ir: add tests for bt_ctf_event's stream accessor
X-Git-Tag: v2.0.0-pre1~1283
X-Git-Url: http://git.efficios.com/?p=babeltrace.git;a=commitdiff_plain;h=2fb29fdc231084f2178c0865c2cd0de5d1c32d87

ir: add tests for bt_ctf_event's stream accessor

Signed-off-by: Jérémie Galarneau <jeremie.galarneau@efficios.com>
---

diff --git a/tests/lib/test_ctf_writer.c b/tests/lib/test_ctf_writer.c
index 9d98d627..2aa0c7e0 100644
--- a/tests/lib/test_ctf_writer.c
+++ b/tests/lib/test_ctf_writer.c
@@ -1663,7 +1663,8 @@ void test_instanciate_event_before_stream(struct bt_ctf_writer *writer)
 	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_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;
@@ -1744,16 +1745,26 @@ void test_instanciate_event_before_stream(struct bt_ctf_writer *writer)
 		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);