Add a basic test for set_mem_functions

svn:r1216

diff --git a/test/regress.c b/test/regress.c
index 8c3107ff074f9ac9e26777cd805f9eed74312037..9dc9ffa74f80e4c7587f55b87918588de5ada393 100644 (file)
--- a/test/regress.c
+++ b/test/regress.c
@@ -1534,6 +1534,56 @@ end:
        }
 }
 
+int
+check_dummy_mem_ok(void *_mem)
+{
+       char *mem = _mem;
+       mem -= 16;
+       return !memcmp(mem, "{[<guardedram>]}", 16);
+}
+
+static void *
+dummy_malloc(size_t len)
+{
+       char *mem = malloc(len+16);
+       memcpy(mem, "{[<guardedram>]}", 16);
+       return mem+16;
+}
+
+static void *
+dummy_realloc(void *_mem, size_t len)
+{
+       char *mem = _mem;
+       if (!mem)
+               return dummy_malloc(len);
+       tt_want(check_dummy_mem_ok(_mem));
+       mem -= 16;
+       mem = realloc(mem, len+16);
+       return mem+16;
+}
+
+static void
+dummy_free(void *_mem)
+{
+       char *mem = _mem;
+       tt_want(check_dummy_mem_ok(_mem));
+       mem -= 16;
+       free(mem);
+}
+
+static void
+test_mm_functions(void *arg)
+{
+       struct event_base *b = NULL;
+       event_set_mem_functions(dummy_malloc, dummy_realloc, dummy_free);
+       b = event_base_new();
+       tt_assert(b);
+       tt_assert(check_dummy_mem_ok(b));
+end:
+       if (b)
+               event_base_free(b);
+}
+
 struct testcase_t main_testcases[] = {
         /* Some converted-over tests */
         { "methods", test_methods, TT_FORK, NULL, NULL },
@@ -1564,6 +1614,7 @@ struct testcase_t main_testcases[] = {
        { "event_once", test_event_once, TT_ISOLATED, &basic_setup, NULL },
        { "event_pending", test_event_pending, TT_ISOLATED, &basic_setup,
          NULL },
+       { "mm_functions", test_mm_functions, TT_FORK, NULL, NULL },
 
 #ifndef WIN32
         LEGACY(fork, TT_ISOLATED),