1 files changed, 16 insertions, 0 deletions
diff --git a/testsuites/support/include/buffer_test_io.h b/testsuites/support/include/buffer_test_io.h
index 03d5419b76..3cadc481f5 100644
--- a/testsuites/support/include/buffer_test_io.h
+++ b/testsuites/support/include/buffer_test_io.h
@@ -5,6 +5,8 @@
 #ifndef __BUFFER_TEST_IO_h
 #define __BUFFER_TEST_IO_h
 
+#include <rtems/test.h>
+
 #ifdef __cplusplus
 extern "C" {
 #endif
@@ -54,6 +56,10 @@ extern "C" {
     do { \
     } while (0)
 
+  #define TEST_BEGIN() rtems_test_begink()
+
+  #define TEST_END() rtems_test_endk()
+
 /*
  *  BUFFER TEST OUTPUT
  */
@@ -152,6 +158,10 @@ extern "C" {
       fflush(stdout); \
     } while (0)
 
+  #define TEST_BEGIN() rtems_test_begin()
+
+  #define TEST_END() rtems_test_end()
+
 /*
  *  USE IPRINT
  */
@@ -197,6 +207,12 @@ extern "C" {
       fflush(stdout); \
     } while (0)
 
+  #define TEST_BEGIN() \
+    rtems_test_begin_with_plugin((rtems_printk_plugin_t) fiprintf, stderr)
+
+  #define TEST_END() \
+    rtems_test_end_with_plugin((rtems_printk_plugin_t) fiprintf, stderr)
+
 #endif
 
 #ifdef __cplusplus