tdb2: tdb_append implemented, beef up tests.

[ccan] / ccan / tdb2 / test / run-enlarge_hash.c

diff --git a/ccan/tdb2/test/run-enlarge_hash.c b/ccan/tdb2/test/run-enlarge_hash.c
index 10ded59b205f0fdc2678799cb4dda250da454b6e..28299081550e372254c99e27a36138ec4230c85c 100644 (file)
--- a/ccan/tdb2/test/run-enlarge_hash.c
+++ b/ccan/tdb2/test/run-enlarge_hash.c
@@ -6,24 +6,63 @@
 #include <ccan/tap/tap.h>
 #include "logging.h"
 
+/* We rig the hash so adjacent-numbered records always clash. */
+static uint64_t clash(const void *key, size_t len, uint64_t seed, void *priv)
+{
+       return *(unsigned int *)key / 2;
+}
+
 int main(int argc, char *argv[])
 {
        unsigned int i;
        struct tdb_context *tdb;
-       int flags[] = { TDB_INTERNAL, TDB_DEFAULT,
-                       TDB_INTERNAL|TDB_CONVERT, TDB_CONVERT };
+       unsigned int v;
+       struct tdb_data key = { (unsigned char *)&v, sizeof(v) };
+       struct tdb_data data = { (unsigned char *)&v, sizeof(v) };
+       union tdb_attribute hattr = { .hash = { .base = { TDB_ATTRIBUTE_HASH },
+                                               .hash_fn = clash } };
+       int flags[] = { TDB_INTERNAL, TDB_DEFAULT, TDB_NOMMAP,
+                       TDB_INTERNAL|TDB_CONVERT, TDB_CONVERT, 
+                       TDB_NOMMAP|TDB_CONVERT };
 
-       plan_tests(sizeof(flags) / sizeof(flags[0]) * 2 + 1);
+       hattr.base.next = &tap_log_attr;
+
+       plan_tests(sizeof(flags) / sizeof(flags[0]) * 11 + 1);
        for (i = 0; i < sizeof(flags) / sizeof(flags[0]); i++) {
-               tdb = tdb_open("/tmp/run-new_database.tdb", flags[i],
-                              O_RDWR|O_CREAT|O_TRUNC, 0600, NULL);
-               tdb->log = tap_log_fn;
+               tdb = tdb_open("run-enlarge-hash.tdb", flags[i],
+                              O_RDWR|O_CREAT|O_TRUNC, 0600, &hattr);
                ok1(tdb);
-               if (tdb) {
-                       enlarge_hash(tdb);
-                       ok1(tdb_check(tdb, NULL, NULL) == 0);
-                       tdb_close(tdb);
-               }
+               if (!tdb)
+                       continue;
+
+               /* Put a single entry in. */
+               v = 0;
+               ok1(tdb_store(tdb, key, data, TDB_INSERT) == 0);
+               enlarge_hash(tdb);
+               ok1(tdb_check(tdb, NULL, NULL) == 0);
+
+               /* Put a non-clashing entry in. */
+               v = 2;
+               ok1(tdb_store(tdb, key, data, TDB_INSERT) == 0);
+               enlarge_hash(tdb);
+               ok1(tdb_check(tdb, NULL, NULL) == 0);
+
+               /* Now, make a clash. */
+               v = 1;
+               ok1(tdb_store(tdb, key, data, TDB_INSERT) == 0);
+               enlarge_hash(tdb);
+               ok1(tdb_check(tdb, NULL, NULL) == 0);
+
+               /* Clash at end. */
+               v = ((1 << tdb->header.v.hash_bits) - 1) * 2;
+               ok1(tdb_store(tdb, key, data, TDB_INSERT) == 0);
+               v++;
+               ok1(tdb_store(tdb, key, data, TDB_INSERT) == 0);
+               ok1(tdb_check(tdb, NULL, NULL) == 0);
+               enlarge_hash(tdb);
+               ok1(tdb_check(tdb, NULL, NULL) == 0);
+
+               tdb_close(tdb);
        }
        ok1(tap_log_messages == 0);
        return exit_status();