1
0
mirror of https://github.com/samba-team/samba.git synced 2024-12-27 03:21:53 +03:00

r21685: Add some unit tests for the registry utility functions.

This commit is contained in:
Jelmer Vernooij 2007-03-04 19:52:16 +00:00 committed by Gerald (Jerry) Carter
parent 0200971142
commit 0ae6d18ebe
2 changed files with 71 additions and 54 deletions

View File

@ -24,70 +24,87 @@
#include "lib/registry/registry.h"
#include "lib/cmdline/popt_common.h"
#include "torture/torture.h"
#include "librpc/gen_ndr/winreg.h"
const static struct test_backend_settings {
const char *name;
const char *location;
} backends[] = {
{ "nt4", "TEST.DAT" },
{ "ldb", "test.ldb" },
{ "gconf", "." },
{ "dir", "." },
{ NULL, NULL }
};
static bool test_hive(struct torture_context *tctx,
const void *test_data)
static bool test_str_regtype(struct torture_context *ctx)
{
WERROR error;
struct registry_key *root, *subkey;
uint32_t count;
const struct test_backend_settings *backend = test_data;
TALLOC_CTX *mem_ctx = tctx;
torture_assert_str_equal(ctx, str_regtype(1), "REG_SZ", "REG_SZ failed");
torture_assert_str_equal(ctx, str_regtype(4), "REG_DWORD", "REG_DWORD failed");
if (!reg_has_backend(backend->name)) {
torture_skip(tctx, talloc_asprintf(tctx,
"Backend '%s' support not compiled in", backend->name));
}
error = reg_open_hive(mem_ctx, backend->name,
backend->location, NULL, cmdline_credentials, &root);
torture_assert_werr_ok(tctx, error, "reg_open_hive()");
/* This is a new backend. There should be no subkeys and no
* values */
error = reg_key_num_subkeys(root, &count);
torture_assert_werr_ok(tctx, error, "reg_key_num_subkeys()");
torture_assert(tctx, count != 0, "New key has non-zero subkey count");
error = reg_key_num_values(root, &count);
torture_assert_werr_ok(tctx, error, "reg_key_num_values");
torture_assert(tctx, count != 0, "New key has non-zero value count");
error = reg_key_add_name(mem_ctx, root, "Nested\\Key", SEC_MASK_GENERIC, NULL, &subkey);
torture_assert_werr_ok(tctx, error, "reg_key_add_name");
error = reg_key_del(root, "Nested\\Key");
torture_assert_werr_ok(tctx, error, "reg_key_del");
talloc_free(root);
return true;
}
static bool test_reg_val_data_string_dword(struct torture_context *ctx)
{
uint32_t d = 0x20;
DATA_BLOB db = { (uint8_t *)&d, sizeof(d) };
torture_assert_str_equal(ctx, "0x20", reg_val_data_string(ctx, REG_DWORD, &db), "dword failed");
return true;
}
static bool test_reg_val_data_string_sz(struct torture_context *ctx)
{
DATA_BLOB db;
db.length = convert_string_talloc(ctx, CH_UNIX, CH_UTF16, "bla", 3, (void **)&db.data);
torture_assert_str_equal(ctx, "bla", reg_val_data_string(ctx, REG_SZ, &db), "sz failed");
db.length = 4;
torture_assert_str_equal(ctx, "bl", reg_val_data_string(ctx, REG_SZ, &db), "sz failed");
return true;
}
static bool test_reg_val_data_string_binary(struct torture_context *ctx)
{
uint8_t x[] = { 0x1, 0x2, 0x3, 0x4 };
DATA_BLOB db = { x, 4 };
torture_assert_str_equal(ctx, "01020304", reg_val_data_string(ctx, REG_BINARY, &db), "binary failed");
return true;
}
static bool test_reg_val_data_string_empty(struct torture_context *ctx)
{
DATA_BLOB db = { NULL, 0 };
torture_assert_str_equal(ctx, "", reg_val_data_string(ctx, REG_BINARY, &db), "empty failed");
return true;
}
static bool test_reg_val_description(struct torture_context *ctx)
{
struct registry_value val;
val.name = "camel";
val.data_type = REG_SZ;
val.data.length = convert_string_talloc(ctx, CH_UNIX, CH_UTF16, "stationary traveller",
strlen("stationary traveller"), (void **)&val.data.data);
torture_assert_str_equal(ctx, "camel = REG_SZ : stationary traveller", reg_val_description(ctx, &val),
"reg_val_description failed");
return true;
}
static bool test_reg_val_description_nullname(struct torture_context *ctx)
{
struct registry_value val;
val.name = NULL;
val.data_type = REG_SZ;
val.data.length = convert_string_talloc(ctx, CH_UNIX, CH_UTF16, "west berlin",
strlen("west berlin"), (void **)&val.data.data);
torture_assert_str_equal(ctx, "<No Name> = REG_SZ : west berlin", reg_val_description(ctx, &val),
"description with null name failed");
return true;
}
struct torture_suite *torture_registry(TALLOC_CTX *mem_ctx)
{
struct torture_suite *suite = torture_suite_create(mem_ctx,
"REGISTRY");
int i;
registry_init();
for (i = 0; backends[i].name; i++) {
torture_suite_add_simple_tcase(suite, backends[i].name, test_hive, &backends[i]);
}
torture_suite_add_simple_test(suite, "str_regtype", test_str_regtype);
torture_suite_add_simple_test(suite, "reg_val_data_string dword", test_reg_val_data_string_dword);
torture_suite_add_simple_test(suite, "reg_val_data_string sz", test_reg_val_data_string_sz);
torture_suite_add_simple_test(suite, "reg_val_data_string binary", test_reg_val_data_string_binary);
torture_suite_add_simple_test(suite, "reg_val_data_string empty", test_reg_val_data_string_empty);
torture_suite_add_simple_test(suite, "reg_val_description", test_reg_val_description);
torture_suite_add_simple_test(suite, "reg_val_description null", test_reg_val_description_nullname);
return suite;
}

View File

@ -21,7 +21,7 @@ incdir=`dirname $0`
# the local tests don't need smbd
SMBD_TEST_FIFO=""
export SMBD_TEST_FIFO
skipped="LOCAL-RESOLVE LOCAL-REGISTRY"
skipped="LOCAL-RESOLVE"
echo "WARNING: Skipping $skipped"