s4:messaging/tests: correctly teardown imessage contexts

Signed-off-by: Stefan Metzmacher <metze@samba.org>
Reviewed-by: Andreas Schneider <asn@samba.org>

diff --git a/source4/lib/messaging/tests/irpc.c b/source4/lib/messaging/tests/irpc.c
index 466b47fdb34d2dd9bd2be102e46ddb68a427ec11..2848d264023bebe49eb7e7d6defedfd6917c8112 100644 (file)
--- a/source4/lib/messaging/tests/irpc.c
+++ b/source4/lib/messaging/tests/irpc.c
@@ -258,14 +258,14 @@ static bool irpc_setup(struct torture_context *tctx, void **_data)
 
        data->ev = tctx->ev;
        torture_assert(tctx, data->msg_ctx1 = 
-                      imessaging_init(tctx,
+                      imessaging_init(data,
                                      tctx->lp_ctx,
                                      cluster_id(0, MSG_ID1),
                                      data->ev),
                       "Failed to init first messaging context");
 
        torture_assert(tctx, data->msg_ctx2 = 
-                      imessaging_init(tctx,
+                      imessaging_init(data,
                                      tctx->lp_ctx,
                                      cluster_id(0, MSG_ID2), 
                                      data->ev),
@@ -281,6 +281,16 @@ static bool irpc_setup(struct torture_context *tctx, void **_data)
        return true;
 }
 
+static bool irpc_teardown(struct torture_context *tctx, void *_data)
+{
+       struct irpc_test_data *data =
+               talloc_get_type_abort(_data,
+               struct irpc_test_data);
+
+       TALLOC_FREE(data);
+       return true;
+}
+
 struct torture_suite *torture_local_irpc(TALLOC_CTX *mem_ctx)
 {
        struct torture_suite *suite = torture_suite_create(mem_ctx, "irpc");
@@ -295,6 +305,7 @@ struct torture_suite *torture_local_irpc(TALLOC_CTX *mem_ctx)
        values[4] = random() & 0xFFFFFFFF;
 
        tcase->setup = irpc_setup;
+       tcase->teardown = irpc_teardown;
 
        for (i = 0; i < 5; i++) {
                torture_tcase_add_test_const(tcase, "addone", test_addone,