X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=tests%2Flib%2Ftest_ir_visit.c;h=50498c2cf4d7a600c2b6aa2fa844d0f8a54e1e4f;hb=8deee0391c182fffc2e3c41f343a008df0e42624;hp=7c1c8c261705bf49e3aacd79391dfa87fc07eec7;hpb=44ea72c5be36c31f53d21df1ea6fba8df939ce4c;p=babeltrace.git

diff --git a/tests/lib/test_ir_visit.c b/tests/lib/test_ir_visit.c
index 7c1c8c26..50498c2c 100644
--- a/tests/lib/test_ir_visit.c
+++ b/tests/lib/test_ir_visit.c
@@ -73,6 +73,7 @@ struct bt_event_class *init_event_class(const char *name)
 {
 	int ret;
 	struct bt_event_class *ec = bt_event_class_create(name);
+	struct bt_field_type *payload_ft = NULL;;
 	struct bt_field_type *int_field =
 			bt_field_type_integer_create(8);
 
@@ -80,16 +81,20 @@ struct bt_event_class *init_event_class(const char *name)
 		goto error;
 	}
 
-	ret = bt_event_class_add_field(ec, int_field, "an_int_field");
+	payload_ft = bt_event_class_get_payload_field_type(ec);
+	ret = bt_field_type_structure_add_field(payload_ft,
+		int_field, "an_int_field");
 	if (ret) {
 		goto error;
 	}
 
 	BT_PUT(int_field);
+	BT_PUT(payload_ft);
 	return ec;
 error:
 	BT_PUT(ec);
 	BT_PUT(int_field);
+	BT_PUT(payload_ft);
 	return NULL;
 }
 
@@ -125,10 +130,10 @@ static void set_stream_class_field_types(
 	assert(ret == 0);
 	bt_put(ft);
 
-	ret = bt_stream_class_set_packet_context_type(stream_class,
+	ret = bt_stream_class_set_packet_context_field_type(stream_class,
 		packet_context_type);
 	assert(ret == 0);
-	ret = bt_stream_class_set_event_header_type(stream_class,
+	ret = bt_stream_class_set_event_header_field_type(stream_class,
 		event_header_type);
 	assert(ret == 0);
 
@@ -151,7 +156,7 @@ static void set_trace_packet_header(struct bt_trace *trace)
 	assert(ret == 0);
 	bt_put(ft);
 
-	ret = bt_trace_set_packet_header_type(trace,
+	ret = bt_trace_set_packet_header_field_type(trace,
 		packet_header_type);
 	assert(ret == 0);