tests/lib/utils: add RunIn::onMsgIterNext()

[babeltrace.git] / tests / lib / utils / run-in.hpp

diff --git a/tests/lib/utils/run-in.hpp b/tests/lib/utils/run-in.hpp
index d08e931a125cb8e67719ec8765c389b49f405a99..54e8ec20c3204401990e36f8e57ffa87b89b6a3b 100644 (file)
--- a/tests/lib/utils/run-in.hpp
+++ b/tests/lib/utils/run-in.hpp
@@ -4,8 +4,8 @@
  * Copyright (C) 2020-2023 EfficiOS, inc.
  */
 
-#ifndef TESTS_LIB_UTILS_H
-#define TESTS_LIB_UTILS_H
+#ifndef TESTS_LIB_UTILS_RUN_IN_HPP
+#define TESTS_LIB_UTILS_RUN_IN_HPP
 
 #include <babeltrace2/babeltrace.h>
 
@@ -38,6 +38,12 @@ public:
      * Called when initializing the message iterator `self`.
      */
     virtual void onMsgIterInit(bt2::SelfMessageIterator self);
+
+    /*
+     * Called within the "next" method of `self` to return the messages
+     * `msgs`.
+     */
+    virtual void onMsgIterNext(bt2::SelfMessageIterator self, bt2::ConstMessageArray& msgs);
 };
 
 /*
@@ -46,4 +52,4 @@ public:
  */
 void runIn(RunIn& runIn);
 
-#endif /* TESTS_LIB_UTILS_H */
+#endif /* TESTS_LIB_UTILS_RUN_IN_HPP */