Add test to verify dynstr_malloc_error_abort actually calls abort.

2 files changed, 39 insertions, 0 deletions

diff --git a/dynstr.c b/dynstr.c
index 28804bf..f445ea1 100644
--- a/dynstr.c
+++ b/dynstr.c
@@ -34,4 +34,5 @@ void dynstr_malloc_error_indicate(int error, size_t size, void *oldptr)
 
 void dynstr_malloc_error_abort(int error, size_t size, void *oldptr)
 {
+    abort();
 }
diff --git a/unittests.c b/unittests.c
index fd30081..28a9e9d 100644
--- a/unittests.c
+++ b/unittests.c
@@ -1,3 +1,7 @@
+#define _POSIX_C_SOURCE 200112L
+
+#include <setjmp.h>
+#include <signal.h>
 #include <stdbool.h>
 #include <stdio.h>
 #include <stdlib.h>
@@ -5,6 +9,13 @@
 #include "dynstr.h"
 
 
+#define FAIL(msg) \
+    do { \
+        printf("FAIL: %s\n", msg); \
+        return false; \
+    } while (0)
+
+
 #define FAIL_UNLESS_EQUAL(a,b) \
     do { \
         if ((a) != (b)) { \
@@ -31,6 +42,32 @@ static int test_sets_error_handler(void)
 }
 
 
+static jmp_buf env;
+static void abort_handler(int signo)
+{
+    longjmp(env, 1);
+}
+
+
+static int test_abort_handler_calls_abort(void)
+{
+    struct sigaction act;
+    struct sigaction oldact;
+    
+    if (setjmp(env) == 0) {
+        act.sa_handler = abort_handler;
+        act.sa_flags = 0;
+        sigaction(SIGABRT, &act, &oldact);
+        dynstr_set_malloc_error_handler(dynstr_malloc_error_abort);
+        dynstr_malloc_error_abort(0, 0, 0);
+        FAIL("dynstr_malloc_error_abort returned");
+    } else {
+        sigaction(SIGABRT, &oldact, NULL);
+    }
+    return true;
+}
+
+
 struct test {
     const char *name;
     int (*test)(void);
@@ -42,6 +79,7 @@ struct test {
 static const struct test tests[] = {
     TEST(test_default_error_handler_is_indicate),
     TEST(test_sets_error_handler),
+    TEST(test_abort_handler_calls_abort),
 };
 static const int num_tests = sizeof(tests) / sizeof(tests[0]);