X-Git-Url: https://codewiz.org/gitweb?a=blobdiff_plain;ds=inline;f=bertos%2Fcfg%2Ftest.h;h=f9c6ec21d4edf0562b4352de054ce118f3f59208;hb=8e61208aaf8657a4740dbf4b8bd8e880c4403353;hp=374250269fa0a54c073bc10800c2e8dc78c32ebb;hpb=e94a42a80537a6a2bd92be602d308eff4bc6d127;p=bertos.git

diff --git a/bertos/cfg/test.h b/bertos/cfg/test.h
index 37425026..f9c6ec21 100644
--- a/bertos/cfg/test.h
+++ b/bertos/cfg/test.h
@@ -49,15 +49,15 @@
  *
  * Then, at the end of your module_test you can write:
  * \code
- * #include TEST_ONLY(whatuneed.h)
- * #include TEST_ONLY(whatuneed.c)
- * #include TEST_ONLY(...)
+ * #if UNIT_TEST
+ *	#include <whatuneed.h>
+ *	#include <whatuneed.c>
+ *	#include <...>
  *
- * TEST_MAIN(module);
+ *	TEST_MAIN(module);
+ * #endif
  * \endcode
  *
- * The macro TEST_ONLY expand to nothing in non-TEST mode or to
- * the specified filename if _TEST is defined.
  * Including directly into your module the file.c you need to
  * run the test allows you to build and run the test compiling
  * only one file.
@@ -75,25 +75,28 @@
 
 #if defined(ARCH_UNITTEST) && (ARCH & ARCH_UNITTEST)
 	#define UNIT_TEST 1
+
+	/**
+	 * Macro used to generate a main() for a test to be compiled
+	 * on hosted platform.
+	 */
+	#define TEST_MAIN(module) \
+	int main(void) \
+	{ \
+		if (module##_testSetup() != 0) \
+			return 1; \
+		if (module##_testRun() != 0) \
+			return 2; \
+		if (module##_testTearDown() != 0) \
+			return 3; \
+		return 0; \
+	}
 #else /* !TEST */
 	#define UNIT_TEST 0
+
+	#define TEST_MAIN(module) /* nothing */
 #endif /* TEST */
 
-/**
- * Macro used to generate a main() for a test to be compiled
- * on hosted platform.
- */
-#define TEST_MAIN(module) \
-int main(void) \
-{ \
-	if (module##_testSetup() != 0) \
-		return 1; \
-	if (module##_testRun() != 0) \
-		return 2; \
-	if (module##_testTearDown() != 0) \
-		return 3; \
-	return 0; \
-}
 
 /**
  * Silent an assert in a test.