Create test for #556. Addresses #556.

git-svn-id: file:///svn/tokudb@2941 c7de825b-a66e-492c-adef-691d508d4ae1

src/tests/test_abort2.c

+/* Put some insert messages into an internal buffer (by first creating a DB, filling it up, then closing it, and reopening, and inserting a few things)
+ * Then perform a transaction that overwrites some of those internal things.
+ * Then abort the transaction.
+ * Make sure those middle things made it back into the tree.
+ */
+
+#include <db.h>
+#include <sys/stat.h>
+#include "test.h"
+
+DB_ENV *env;
+DB *db;
+DB_TXN *txn;
+
+void insert (int i, int j) {
+    char hello[30], there[230];
+    DBT key,data;
+    snprintf(hello, sizeof(hello), "hello%d", i);
+    snprintf(there, sizeof(there), "%dthere%d %*s", j, i, 10+i%40, "padding");
+    int r = db->put(db, txn,
+		    dbt_init(&key,  hello, strlen(hello)+1),
+		    dbt_init(&data, there, strlen(there)+1),
+		    0);
+    CKERR(r);
+}
+
+void do_test_abort2 (void) {
+    int r;
+    system("rm -rf " ENVDIR);
+    r=mkdir(ENVDIR, 0777);       assert(r==0);
+
+    r=db_env_create(&env, 0); assert(r==0);
+    env->set_errfile(env, stderr);
+    r=env->open(env, ENVDIR, DB_INIT_LOCK|DB_INIT_LOG|DB_INIT_MPOOL|DB_INIT_TXN|DB_CREATE|DB_PRIVATE, 0777); CKERR(r);
+    r=db_create(&db, env, 0); CKERR(r);
+    r=db->set_pagesize(db, 4096); // Use a small page
+    r=env->txn_begin(env, 0, &txn, 0); assert(r==0);
+    r=db->open(db, txn, "foo.db", 0, DB_BTREE, DB_CREATE, 0777); CKERR(r);
+    r=txn->commit(txn, 0);    assert(r==0);
+
+    r=env->txn_begin(env, 0, &txn, 0); assert(r==0);
+    {
+	int i;
+	for (i=0; i<1000; i++) {
+	    insert(4*i, 0);
+	}
+    }
+    r=txn->commit(txn, 0); CKERR(r);
+    r=db->close(db, 0); CKERR(r);
+    r=env->close(env, 0); CKERR(r);
+
+    printf("%s:%d\n", __FILE__, __LINE__);
+
+    // Now do a few inserts.
+    r=db_env_create(&env, 0); assert(r==0);
+    env->set_errfile(env, stderr);
+    r=env->open(env, ENVDIR, DB_INIT_LOCK|DB_INIT_LOG|DB_INIT_MPOOL|DB_INIT_TXN|DB_CREATE|DB_PRIVATE, 0777); CKERR(r);
+    r=db_create(&db, env, 0); CKERR(r);
+    r=env->txn_begin(env, 0, &txn, 0); CKERR(r);
+    r=db->open(db, txn, "foo.db", 0, DB_BTREE, 0, 0777); CKERR(r);
+#ifndef TOKUDB
+    {
+	u_int32_t ps;
+	r=db->get_pagesize(db, &ps); CKERR(r);
+	assert(ps==4096);
+    }
+#endif
+    r=txn->commit(txn, 0);    assert(r==0);
+
+    r=env->txn_begin(env, 0, &txn, 0); assert(r==0);
+    insert(3, 0);
+    insert(5, 0);
+    insert(7, 0);
+    r=txn->commit(txn, 0); CKERR(r);
+
+    r=env->txn_begin(env, 0, &txn, 0); assert(r==0);
+    insert(7, 1);
+    r=txn->abort(txn); CKERR(r);
+
+    // Don't do a query on "hello7", because that will force things out of the buffer.
+
+    r=env->txn_begin(env, 0, &txn, 0); assert(r==0);
+    {
+	DBT key,data;
+	memset(&data, 0, sizeof(data));
+	r = db->get(db, txn, dbt_init(&key, "hello7", strlen("hello7")+1), &data, 0);
+	CKERR(r);
+	printf("data is %s\n", (char*)data.data);
+	assert(((char*)data.data)[0]=='0');
+    }
+    r=txn->abort(txn); CKERR(r);
+    
+    r=db->close(db, 0); CKERR(r);
+    r=env->close(env, 0); CKERR(r);
+
+}
+
+int main (int argc, const char *argv[]) {
+    parse_args(argc, argv);
+    do_test_abort2();
+    return 0;
+}