X-Git-Url: http://git.efficios.com/?p=babeltrace.git;a=blobdiff_plain;f=tests%2Flib%2Futils%2Frun-in.hpp;fp=tests%2Flib%2Futils%2Frun-in.hpp;h=01774f9053f5e424dcea154d84b31d73f3f955c2;hp=0855e05e062ba697b3f9b4b4b51ad3a11833e634;hb=5d15e4cab25bba8cf293e7edb6adf3c6b1c68958;hpb=209ec9b4c609b0a989d33be9949b80a31544faa8

diff --git a/tests/lib/utils/run-in.hpp b/tests/lib/utils/run-in.hpp
index 0855e05e..01774f90 100644
--- a/tests/lib/utils/run-in.hpp
+++ b/tests/lib/utils/run-in.hpp
@@ -11,9 +11,13 @@
 
 #include <babeltrace2/babeltrace.h>
 
-using RunInCompClsQueryFunc = std::function<void(bt_self_component_class *)>;
-using RunInCompClsInitFunc = std::function<void(bt_self_component *)>;
-using RunInMsgIterClsInitFunc = std::function<void(bt_self_message_iterator *)>;
+#include "cpp-common/bt2/self-component-class.hpp"
+#include "cpp-common/bt2/self-component-port.hpp"
+#include "cpp-common/bt2/self-message-iterator.hpp"
+
+using RunInCompClsQueryFunc = std::function<void(bt2::SelfComponentClass)>;
+using RunInCompClsInitFunc = std::function<void(bt2::SelfComponent)>;
+using RunInMsgIterClsInitFunc = std::function<void(bt2::SelfMessageIterator)>;
 
 /*
  * Runs: