ldb: Add tests for new ldb handle and event context behaviour

Signed-off-by: Andrew Bartlett <abartlet@samba.org>
Reviewed-by: Garming Sam <garming@catalyst.net.nz>

diff --git a/lib/ldb/tests/ldb_mod_op_test.c b/lib/ldb/tests/ldb_mod_op_test.c
index be1a13cff820f1bfeb6ef33243c5e3d9b547ba06..7d8b074e8f26dc45a3c9a4f7a11f1ecf0f440d0f 100644 (file)
--- a/lib/ldb/tests/ldb_mod_op_test.c
+++ b/lib/ldb/tests/ldb_mod_op_test.c
@@ -367,6 +367,57 @@ static void test_ldb_del_noexist(void **state)
        assert_int_equal(ret, LDB_ERR_NO_SUCH_OBJECT);
 }
 
+static void test_ldb_handle(void **state)
+{
+       int ret;
+       struct ldbtest_ctx *test_ctx = talloc_get_type_abort(*state,
+                                                       struct ldbtest_ctx);
+       TALLOC_CTX *tmp_ctx;
+       struct ldb_dn *basedn;
+       struct ldb_request *request = NULL;
+       struct ldb_request *request2 = NULL;
+       struct ldb_result *res = NULL;
+       const char *attrs[] = { "cn", NULL };
+
+       tmp_ctx = talloc_new(test_ctx);
+       assert_non_null(tmp_ctx);
+
+       basedn = ldb_dn_new_fmt(tmp_ctx, test_ctx->ldb, "dc=test");
+       assert_non_null(basedn);
+
+       res = talloc_zero(tmp_ctx, struct ldb_result);
+       assert_non_null(res);
+
+       ret = ldb_build_search_req(&request, test_ctx->ldb, tmp_ctx,
+                                  basedn, LDB_SCOPE_BASE,
+                                  NULL, attrs, NULL, res,
+                                  ldb_search_default_callback,
+                                  NULL);
+       assert_int_equal(ret, 0);
+
+       /* We are against ldb_tdb, so expect private event contexts */
+       assert_ptr_not_equal(ldb_handle_get_event_context(request->handle),
+                            ldb_get_event_context(test_ctx->ldb));
+
+       ret = ldb_build_search_req(&request2, test_ctx->ldb, tmp_ctx,
+                                  basedn, LDB_SCOPE_BASE,
+                                  NULL, attrs, NULL, res,
+                                  ldb_search_default_callback,
+                                  request);
+       assert_int_equal(ret, 0);
+
+       /* Expect that same event context will be chained */
+       assert_ptr_equal(ldb_handle_get_event_context(request->handle),
+                        ldb_handle_get_event_context(request2->handle));
+
+       /* Now force this to use the global context */
+       ldb_handle_use_global_event_context(request2->handle);
+       assert_ptr_equal(ldb_handle_get_event_context(request2->handle),
+                        ldb_get_event_context(test_ctx->ldb));
+
+       talloc_free(tmp_ctx);
+}
+
 static void add_keyval(struct ldbtest_ctx *test_ctx,
                       const char *key,
                       const char *val)
@@ -1902,6 +1953,9 @@ int main(int argc, const char **argv)
                cmocka_unit_test_setup_teardown(test_ldb_del_noexist,
                                                ldbtest_setup,
                                                ldbtest_teardown),
+               cmocka_unit_test_setup_teardown(test_ldb_handle,
+                                               ldbtest_setup,
+                                               ldbtest_teardown),
                cmocka_unit_test_setup_teardown(test_transactions,
                                                ldbtest_setup,
                                                ldbtest_teardown),