X-Git-Url: http://git.efficios.com/?p=babeltrace.git;a=blobdiff_plain;f=tests%2Flib%2Futils%2Frun-in.cpp;fp=tests%2Flib%2Futils%2Frun-in.cpp;h=3fef7d327b553f8073596869465c0047ec77e1c5;hp=01913872d209e19b303ebbbbd5812e90f02c40d4;hb=5d15e4cab25bba8cf293e7edb6adf3c6b1c68958;hpb=209ec9b4c609b0a989d33be9949b80a31544faa8

diff --git a/tests/lib/utils/run-in.cpp b/tests/lib/utils/run-in.cpp
index 01913872..3fef7d32 100644
--- a/tests/lib/utils/run-in.cpp
+++ b/tests/lib/utils/run-in.cpp
@@ -7,6 +7,7 @@
 #include <utility>
 
 #include "common/assert.h"
+#include "cpp-common/bt2/wrap.hpp"
 
 #include "run-in.hpp"
 
@@ -37,7 +38,7 @@ bt_component_class_initialize_method_status compClsInit(bt_self_component_source
     auto& data = runInDataFromMethodData(initMethodData);
 
     if (data.compCtxFunc) {
-        data.compCtxFunc(bt_self_component_source_as_self_component(selfComp));
+        data.compCtxFunc(bt2::wrap(bt_self_component_source_as_self_component(selfComp)));
     }
 
     return BT_COMPONENT_CLASS_INITIALIZE_METHOD_STATUS_OK;
@@ -51,7 +52,8 @@ compClsQuery(bt_self_component_class_source * const selfCompCls, bt_private_quer
     auto& data = runInDataFromMethodData(methodData);
 
     if (data.compClsCtxFunc) {
-        data.compClsCtxFunc(bt_self_component_class_source_as_self_component_class(selfCompCls));
+        data.compClsCtxFunc(
+            bt2::wrap(bt_self_component_class_source_as_self_component_class(selfCompCls)));
     }
 
     *result = bt_value_null;
@@ -66,7 +68,7 @@ msgIterClsInit(bt_self_message_iterator * const selfMsgIter,
         bt_self_component_port_output_as_self_component_port(port)));
 
     if (data.msgIterCtxFunc) {
-        data.msgIterCtxFunc(selfMsgIter);
+        data.msgIterCtxFunc(bt2::wrap(selfMsgIter));
     }
 
     return BT_MESSAGE_ITERATOR_CLASS_INITIALIZE_METHOD_STATUS_OK;