From 44aa4d3c37ce32ab961db446c81b936351d1d4e0 Mon Sep 17 00:00:00 2001
From: "Bradley C. Kuszmaul" <bradley@tokutek.com>
Date: Tue, 24 Jun 2008 14:04:52 +0000
Subject: [PATCH] Create test for #944. Addresses #944.
git-svn-id: file:///svn/tokudb@4720 c7de825b-a66e-492c-adef-691d508d4ae1
---
src/tests/test944.c | 135 ++++++++++++++++++++++++++++++++++++++++++++
1 file changed, 135 insertions(+)
create mode 100644 src/tests/test944.c
diff --git a/src/tests/test944.c b/src/tests/test944.c
new file mode 100644
index 00000000000..fea52c92b92
--- /dev/null
+++ b/src/tests/test944.c
@@ -0,0 +1,135 @@
+/* -*- mode: C; c-basic-offset: 4 -*- */
+#include <db.h>
+#include <sys/stat.h>
+#include "test.h"
+
+unsigned char N=8;
+int fact(int n) {
+ if (n<=2) return n;
+ else return n*fact(n-1);
+}
+
+void swapc (unsigned char *a, unsigned char *b) {
+ unsigned char tmp=*a;
+ *a=*b;
+ *b=tmp;
+}
+
+DB_ENV *env;
+DB *db;
+
+void run (int choice) {
+ unsigned char v[N];
+ int i;
+ int r;
+ for (i=0; i<N; i++) {
+ v[i]=i;
+ }
+ for (i=0; i<N; i++) {
+ int nchoices=N-i;
+ swapc(&v[i], &v[i+choice%nchoices]);
+ choice=choice/nchoices;
+ }
+ if (0) {
+ for (i=0; i<N; i++) {
+ printf("%d ", v[i]);
+ }
+
+ printf("\n");
+ }
+ DB_TXN *txn;
+ {
+ r=env->txn_begin(env, 0, &txn, 0); CKERR(r);
+ for (i=0; i<N; i+=2) {
+ DBT kdbt,vdbt;
+ char key=v[i];
+ char val=v[i+1];
+ r=db->put(db, txn, dbt_init(&kdbt, &key, 1), dbt_init(&vdbt, &val, 1), DB_YESOVERWRITE); CKERR(r);
+ }
+ r=txn->commit(txn, DB_TXN_NOSYNC); CKERR(r);
+ }
+ {
+ r=env->txn_begin(env, 0, &txn, 0); CKERR(r);
+ DBC *c;
+ r=db->cursor(db, txn, &c, 0); CKERR(r);
+ DBT kdbt,vdbt;
+ dbt_init_malloc(&kdbt);
+ dbt_init_malloc(&vdbt);
+ i=0;
+ while (0==(r=c->c_get(c, &kdbt, &vdbt, DB_NEXT))) {
+ printf("Got %d %d\n", *(unsigned char*)kdbt.data, *(unsigned char*)vdbt.data);
+ i++;
+ }
+ CKERR2(r, DB_NOTFOUND);
+ printf("i=%d N=%d\n", i, N);
+ assert(i==N/2);
+ r=c->c_close(c); CKERR(r);
+ r=txn->commit(txn, DB_TXN_NOSYNC); CKERR(r);
+ }
+ {
+ r=env->txn_begin(env, 0, &txn, 0); CKERR(r);
+ for (i=0; i<N; i+=2) {
+ DBT kdbt;
+ char key=v[i];
+ r=db->del(db, txn, dbt_init(&kdbt, &key, 1), 0); CKERR(r);
+ printf(" del %d\n", key);
+ }
+ r=txn->commit(txn, DB_TXN_NOSYNC); CKERR(r);
+ }
+ return;
+#if 0
+ char v101=101, v102=102, v1=1, v2=2;
+ {
+ r=env->txn_begin(env, 0, &txn, 0); CKERR(r);
+ DBT k,v;
+ r=db->put(db, txn, dbt_init(&k, &v1, 1), dbt_init(&v, &v101, 1), 0); CKERR(r);
+ r=db->put(db, txn, dbt_init(&k, &v2, 1), dbt_init(&v, &v102, 1), 0); CKERR(r);
+ r=txn->commit(txn, 0); CKERR(r);
+ }
+ {
+ r=env->txn_begin(env, 0, &txn, 0); CKERR(r);
+ DBC *c;
+ r=db->cursor(db, txn, &c, 0); CKERR(r);
+ DBT k,v;
+ r=c->c_get(c, dbt_init_malloc(&k), dbt_init_malloc(&v), DB_FIRST); CKERR(r);
+ assert(*(char*)k.data==v1); assert(*(char*)v.data==v101);
+ r=c->c_get(c, dbt_init_malloc(&k), dbt_init_malloc(&v), DB_NEXT); CKERR(r);
+ assert(*(char*)k.data==v2); assert(*(char*)v.data==v102);
+ r=c->c_get(c, dbt_init_malloc(&k), dbt_init_malloc(&v), DB_NEXT); assert(r!=0);
+ r=c->c_close(c); CKERR(r);
+ r=txn->commit(txn, 0); CKERR(r);
+ }
+#endif
+}
+
+int main(int argc, const char *argv[]) {
+ parse_args(argc, argv);
+
+ system("rm -rf " ENVDIR);
+ mkdir(ENVDIR, 0777);
+
+ int r;
+
+ DB_TXN *txn;
+ {
+ r = db_env_create(&env, 0); CKERR(r);
+ r=env->open(env, ENVDIR, DB_INIT_LOCK|DB_INIT_LOG|DB_INIT_MPOOL|DB_INIT_TXN|DB_CREATE|DB_PRIVATE, 0777); CKERR(r);
+ env->set_errfile(env, stderr);
+ r=env->txn_begin(env, 0, &txn, 0); CKERR(r);
+ r=db_create(&db, env, 0); CKERR(r);
+ r=db->set_flags(db, DB_DUP|DB_DUPSORT);
+ r=db->open(db, txn, "foo.db", 0, DB_BTREE, DB_CREATE, 0777); CKERR(r);
+ r=txn->commit(txn, 0); CKERR(r);
+ }
+ int i;
+ printf("fact(%d)=%d\n", N, fact(N));
+ for (i=0; i<fact(N); i++) {
+ run(i);
+ }
+ {
+ r=db->close(db, 0); CKERR(r);
+ r=env->close(env, 0); CKERR(r);
+ }
+
+ return 0;
+}