Fix unit test check.

[bertos.git] / bertos / cfg / test.h

diff --git a/bertos/cfg/test.h b/bertos/cfg/test.h
index 85c4d4276d93b7cbbb26cdb61c005e76a955acfa..b45515b044c1f110bd15fa7d76b9662b4183e777 100644 (file)
--- a/bertos/cfg/test.h
+++ b/bertos/cfg/test.h
@@ -71,7 +71,11 @@
 #ifndef CFG_TEST_H
 #define CFG_TEST_H
 
-#if (ARCH & ARCH_UNITTEST)
+#include "cfg/cfg_arch.h"
+
+#if defined(ARCH_UNITTEST) && (ARCH & ARCH_UNITTEST)
+
+       #define UNIT_TEST 1
 
        /**
         * Macro used to generate a main() for a test to be compiled
@@ -93,6 +97,7 @@
        #define TEST_ONLY(file) PP_STRINGIZE(file)
 
 #else /* !_TEST */
+       #define UNIT_TEST 0
 
        #define TEST_MAIN(module)  /* nothing */
        #define TEST_ONLY(file)    <cfg/nothing.h>