X-Git-Url: http://git.ozlabs.org/?p=ccan;a=blobdiff_plain;f=ccan%2Frbtree%2Ftest%2Frun.c;h=e1ce38730cd0e1f9c3c41bc3f25a4ac1cf9481b2;hp=5f58fd05eeb19ee9131a63de1e49e6ed22d0726e;hb=737eacb13ce9550f9820cbc55eec8b38fa2fa6a4;hpb=8d20b53ecd76773591e95db3bdfcbbd5129efb9e

diff --git a/ccan/rbtree/test/run.c b/ccan/rbtree/test/run.c
index 5f58fd05..e1ce3873 100644
--- a/ccan/rbtree/test/run.c
+++ b/ccan/rbtree/test/run.c
@@ -1,8 +1,25 @@
+#include <ccan/failtest/failtest.h>
 #include <ccan/rbtree/rbtree.c>
 #include <ccan/tap/tap.h>
 #include <ccan/talloc/talloc.h>
 #include <string.h>
 
+/* We want to test talloc failure paths. */
+static void *my_malloc(size_t size)
+{
+	return malloc(size);
+}
+
+static void my_free(void *ptr)
+{
+	free(ptr);
+}
+
+static void *my_realloc(void *ptr, size_t size)
+{
+	return realloc(ptr, size);
+}
+
 static void *insert_callback(void *param, void *data)
 {
 	ok1(data == param);
@@ -18,6 +35,8 @@ int main(void)
 	/* This is how many tests you plan to run */
 	plan_tests(19);
 
+	talloc_set_allocator(my_malloc, my_free, my_realloc);
+
 	rb = trbt_create(ctx, 0);
 	ok1(rb);
 	ok1(talloc_is_parent(rb, ctx));
@@ -73,5 +92,5 @@ int main(void)
 	talloc_free(ctx);
 
 	/* This exits depending on whether all tests passed */
-	return exit_status();
+	failtest_exit(exit_status());
 }