Index: src/platform/vboot_reference/cgptlib/tests/cgpt_test.h |
diff --git a/src/platform/vboot_reference/cgptlib/tests/cgpt_test.h b/src/platform/vboot_reference/cgptlib/tests/cgpt_test.h |
index 4a8b115ff92bc9e71306f02b663d38095f415add..3a4191a8de368e8ad4fd61327627633303c7ddb6 100644 |
--- a/src/platform/vboot_reference/cgptlib/tests/cgpt_test.h |
+++ b/src/platform/vboot_reference/cgptlib/tests/cgpt_test.h |
@@ -2,8 +2,8 @@ |
* Use of this source code is governed by a BSD-style license that can be |
* found in the LICENSE file. |
*/ |
-#ifndef VBOOT_REFERENCE_CGPTLIB_TESTS_CGPT_TEST_H_ |
-#define VBOOT_REFERENCE_CGPTLIB_TESTS_CGPT_TEST_H_ |
+#ifndef VBOOT_REFERENCE_CGPT_TEST_H_ |
+#define VBOOT_REFERENCE_CGPT_TEST_H_ |
#include <stdio.h> |
@@ -12,16 +12,23 @@ enum { |
TEST_OK = 0, |
}; |
+#define TEST_CASE(func) #func, func |
+typedef int (*test_func)(void); |
+ |
+#define ARRAY_SIZE(array) (sizeof(array) / sizeof(array[0])) |
+ |
/* ANSI Color coding sequences. */ |
#define COL_GREEN "\e[1;32m" |
#define COL_RED "\e[0;31m" |
#define COL_STOP "\e[m" |
#define EXPECT(expr) \ |
- if (!expr) { \ |
- printf(COL_RED " fail " COL_STOP "in expression %s in %s() line %d\n",\ |
- #expr, __FUNCTION__, __LINE__); \ |
- return TEST_FAIL; \ |
- } |
+ do { \ |
+ if (!(expr)) { \ |
+ printf(COL_RED " fail " COL_STOP "in expression %s in %s() line %d\n",\ |
+ #expr, __FUNCTION__, __LINE__); \ |
+ return TEST_FAIL; \ |
+ } \ |
+ } while (0) |
-#endif /* VBOOT_REFERENCE_CGPTLIB_TESTS_CGPT_TEST_H_ */ |
+#endif /* VBOOT_REFERENCE_CGPT_TEST_H_ */ |