[src/trunk]: src/regress/lib/libc/cdb Add a simple test for cdb(5).

[rmind <rmind%NetBSD.org@localhost>]

details:   https://anonhg.NetBSD.org/src/rev/19e24cce5ef5
branches:  trunk
changeset: 780367:19e24cce5ef5
user:      rmind <rmind%NetBSD.org@localhost>
date:      Sat Jul 21 22:22:55 2012 +0000

description:
Add a simple test for cdb(5).

diffstat:

 regress/lib/libc/cdb/cdb_test.c |  90 +++++++++++++++++++++++++++++++++++++++++
 1 files changed, 90 insertions(+), 0 deletions(-)

diffs (94 lines):

diff -r 48bddcadb842 -r 19e24cce5ef5 regress/lib/libc/cdb/cdb_test.c
--- /dev/null   Thu Jan 01 00:00:00 1970 +0000
+++ b/regress/lib/libc/cdb/cdb_test.c   Sat Jul 21 22:22:55 2012 +0000
@@ -0,0 +1,90 @@
+/*     $NetBSD: cdb_test.c,v 1.1 2012/07/21 22:22:55 rmind Exp $       */
+
+/*
+ * This file is in the Public Domain.
+ */
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <fcntl.h>
+#include <unistd.h>
+#include <inttypes.h>
+#include <assert.h>
+
+#include "cdbr.h"
+#include "cdbw.h"
+
+#define        CDB_FILE                "/tmp/test.cdb"
+
+#define DATASOURCE_SIZE                8192
+
+static uint64_t                        k[DATASOURCE_SIZE];
+static uint64_t                        d[DATASOURCE_SIZE];
+
+static void
+build_cdb(uint32_t (*seedgen)(void))
+{
+       struct cdbw *cdbw = cdbw_open();
+       int i, fd, ret;
+
+       for (i = 0; i < DATASOURCE_SIZE; i++) {
+               k[i] = ((uint64_t)i << 32UL) | random();
+               d[i] = random();
+               ret = cdbw_put(cdbw, &k[i], sizeof(k[0]), &d[i], sizeof(d[0]));
+               assert(ret == 0);
+       }
+
+       fd = open(CDB_FILE, O_RDWR | O_CREAT, 0644);
+       assert(fd > 0);
+
+       ret = cdbw_output(cdbw, fd, "test", seedgen);
+       assert(ret == 0);
+
+       cdbw_close(cdbw);
+       close(fd);
+}
+
+static void
+test_cdb(void)
+{
+       struct cdbr *cdbr = cdbr_open(CDB_FILE, CDBR_DEFAULT);
+       int i, ret;
+
+       assert(cdbr != NULL);
+       assert(cdbr_entries(cdbr) == DATASOURCE_SIZE);
+
+       for (i = 0; i < DATASOURCE_SIZE; i++) {
+               const void *val;
+               size_t len;
+
+               ret = cdbr_find(cdbr, &k[i], sizeof(k[0]), &val, &len);
+               assert(ret == 0);
+               assert(len == sizeof(uint64_t));
+
+               const uint64_t num = *(const uint64_t *)val;
+               if (d[i] == num) {
+                       continue;
+               }
+               fprintf(stderr, "%d: 0x%"PRIu64" != 0x%"PRIu64"\n", i, d[i], num);
+               abort();
+       }
+       cdbr_close(cdbr);
+}
+
+int
+main(int argc, char **argv)
+{
+       int i;
+
+       srandom(time(NULL) ^ getpid());
+
+       for (i = 0; i < 64; i++) {
+               build_cdb((uint32_t (*)(void))random);
+               test_cdb();
+
+               build_cdb(cdbw_stable_seeder);
+               test_cdb();
+       }
+
+       return 0;
+}