diff options
| -rw-r--r-- | unittest/unittest.h | 36 |
1 files changed, 34 insertions, 2 deletions
diff --git a/unittest/unittest.h b/unittest/unittest.h index 0ed005f..2350f55 100644 --- a/unittest/unittest.h +++ b/unittest/unittest.h @@ -3,13 +3,18 @@ #include <stdlib.h> #include <stdio.h> +#include <string.h> + #include "stringify.h" #define TEST_PASS 0 #define TEST_FAIL 1 +#define test_print(fmt, args...) \ + fprintf(stderr, "%s:%d:" fmt, __FILE__, __LINE__, ##args) + #define assert_err(fmt, args...) \ - fprintf(stderr, "%s:%d: Assertion error: " fmt, __FILE__, __LINE__, ##args) + test_print("Assertion error: " fmt, ##args) /* Assert the expression 'expr' */ #define test_assert(expr) ({if (!(expr)) { \ @@ -33,7 +38,34 @@ /* Assert that a != b */ #define test_assert_neq(a, b) ({ \ if ((a) == (b)) { \ - assert_err(__stringify(a) " != " __stringify(b) "\n"); \ + assert_err(__stringify(a) " == " __stringify(b) "\n"); \ + exit(TEST_FAIL); \ + }}) + +/* Assert that a is NULL */ +#define test_assert_null(a) ({ \ + if ((a) != NULL) { \ + assert_err(__stringify(a) " not NULL\n"); \ + exit(TEST_FAIL); \ + }}) + +/* Assert that a is not NULL */ +#define test_assert_not_null(a) ({ \ + if ((a) == NULL) { \ + assert_err(__stringify(a) " is NULL\n"); \ + exit(TEST_FAIL); \ + }}) + +static inline int __xstrmatch(const char *ctx, const char *arg) +{ + size_t len = strlen(ctx); + return (strlen(arg) == len && strncmp(arg, ctx, len) == 0); +} + +/* Assert that string a matches string b */ +#define test_assert_str_eq(a, b) ({ \ + if (!__xstrmatch((a), (b))) { \ + assert_err(__stringify(a) " does not match " __stringify(b) "\n"); \ exit(TEST_FAIL); \ }}) |