Commit fd2d01ab authored by Rich Prohaska's avatar Rich Prohaska Committed by Yoni Fogel

add some tests for the removal of the root fifo. addresses #1691

git-svn-id: file:///svn/toku/tokudb@11326 c7de825b-a66e-492c-adef-691d508d4ae1
parent f574339d
...@@ -68,7 +68,7 @@ static void root_fifo_1(int n) { ...@@ -68,7 +68,7 @@ static void root_fifo_1(int n) {
for (i=0; i<n; i++) { for (i=0; i<n; i++) {
DB *db = null_db; DB *db = null_db;
r = db_create(&db, env, 0); assert(r == 0); assert(db != NULL); r = db_create(&db, env, 0); assert(r == 0); assert(db != NULL);
if (verbose) printf("db %p brt %p\n", db, db->i->brt); if (0) printf("db %p brt %p\n", db, db->i->brt);
r = db->open(db, txn, "test.db", 0, DB_BTREE, DB_CREATE, S_IRWXU+S_IRWXG+S_IRWXO); r = db->open(db, txn, "test.db", 0, DB_BTREE, DB_CREATE, S_IRWXU+S_IRWXG+S_IRWXO);
assert(r == 0); assert(r == 0);
......
...@@ -68,7 +68,7 @@ static void root_fifo_1(int n) { ...@@ -68,7 +68,7 @@ static void root_fifo_1(int n) {
for (i=0; i<n; i++) { for (i=0; i<n; i++) {
DB *db = null_db; DB *db = null_db;
r = db_create(&db, env, 0); assert(r == 0); assert(db != NULL); r = db_create(&db, env, 0); assert(r == 0); assert(db != NULL);
if (verbose) printf("db %p brt %p\n", db, db->i->brt); if (0) printf("db %p brt %p\n", db, db->i->brt);
r = db->open(db, txn, "test.db", 0, DB_BTREE, DB_CREATE, S_IRWXU+S_IRWXG+S_IRWXO); r = db->open(db, txn, "test.db", 0, DB_BTREE, DB_CREATE, S_IRWXU+S_IRWXG+S_IRWXO);
assert(r == 0); assert(r == 0);
......
// test txn commit after db close
#include "test.h"
#include <sys/stat.h>
#include "../ydb-internal.h"
DB_ENV *null_env = NULL;
DB *null_db = NULL;
DB_TXN *null_txn = NULL;
DBC *null_cursor = NULL;
static void create_non_empty(int n) {
DB_ENV *env = null_env;
int r;
r = db_env_create(&env, 0); assert(r == 0); assert(env != NULL);
r = env->set_data_dir(env, ENVDIR);
r = env->set_lg_dir(env, ENVDIR);
r = env->open(env,
0,
DB_INIT_MPOOL+DB_INIT_LOG+DB_INIT_LOCK+DB_INIT_TXN+DB_PRIVATE+DB_CREATE,
S_IRWXU+S_IRWXG+S_IRWXO);
assert(r == 0);
DB_TXN *txn = null_txn;
r = env->txn_begin(env, null_txn, &txn, 0); assert(r == 0); assert(txn != NULL);
DB *db = null_db;
r = db_create(&db, env, 0); assert(r == 0); assert(db != NULL);
r = db->open(db, txn, "test.db", 0, DB_BTREE, DB_CREATE, S_IRWXU+S_IRWXG+S_IRWXO);
assert(r == 0);
int i;
for (i=n; i<2*n; i++) {
DBT key, val;
int k = toku_htonl(i);
r = db->put(db, txn, dbt_init(&key, &k, sizeof k), dbt_init(&val, &i, sizeof i), 0);
assert(r == 0);
}
r = db->close(db, 0); assert(r == 0); db = null_db;
r = txn->commit(txn, 0); assert(r == 0); txn = null_txn;
r = env->close(env, 0); assert(r == 0); env = null_env;
}
static void root_fifo_verify(DB_ENV *env, int n) {
if (verbose) printf("%s:%d %d\n", __FUNCTION__, __LINE__, n);
int r;
DB *db = null_db;
r = db_create(&db, env, 0); assert(r == 0); assert(db != NULL);
r = db->open(db, null_txn, "test.db", 0, DB_BTREE, DB_CREATE, S_IRWXU+S_IRWXG+S_IRWXO);
assert(r == 0);
DB_TXN *txn = null_txn;
r = env->txn_begin(env, null_txn, &txn, 0); assert(r == 0); assert(txn != NULL);
DBC *cursor = null_cursor;
r = db->cursor(db, txn, &cursor, 0); assert(r == 0);
int i;
for (i = 0; ; i++) {
DBT key, val;
memset(&key, 0, sizeof key); memset(&val, 0, sizeof val);
r = cursor->c_get(cursor, &key, &val, DB_NEXT);
if (r != 0) break;
int k;
assert(key.size == sizeof k);
memcpy(&k, key.data, key.size);
assert((int)toku_ntohl(k) == i);
}
assert(i == 2*n);
r = cursor->c_close(cursor); assert(r == 0); cursor = null_cursor;
r = txn->commit(txn, 0); assert(r == 0); txn = null_txn;
r = db->close(db, 0); assert(r == 0); db = null_db;
}
static void root_fifo_31(int n) {
if (verbose) printf("%s:%d %d\n", __FUNCTION__, __LINE__, n);
int r;
// create the env
system("rm -rf " ENVDIR);
toku_os_mkdir(ENVDIR, S_IRWXU+S_IRWXG+S_IRWXO);
// populate
create_non_empty(n);
DB_ENV *env = null_env;
r = db_env_create(&env, 0); assert(r == 0); assert(env != NULL);
r = env->set_data_dir(env, ENVDIR);
r = env->set_lg_dir(env, ENVDIR);
r = env->open(env,
0,
DB_INIT_MPOOL+DB_INIT_LOG+DB_INIT_LOCK+DB_INIT_TXN+DB_PRIVATE+DB_CREATE,
S_IRWXU+S_IRWXG+S_IRWXO);
assert(r == 0);
DB_TXN *txn = null_txn;
r = env->txn_begin(env, null_txn, &txn, 0); assert(r == 0); assert(txn != NULL);
int i;
for (i=0; i<n; i++) {
DB *db = null_db;
r = db_create(&db, env, 0); assert(r == 0); assert(db != NULL);
if (0) printf("db %p brt %p\n", db, db->i->brt);
r = db->open(db, txn, "test.db", 0, DB_BTREE, DB_CREATE, S_IRWXU+S_IRWXG+S_IRWXO);
assert(r == 0);
DBT key, val;
int k = toku_htonl(i);
r = db->put(db, txn, dbt_init(&key, &k, sizeof k), dbt_init(&val, &i, sizeof i), 0);
assert(r == 0);
r = db->close(db, 0); assert(r == 0); db = null_db;
}
r = txn->commit(txn, 0); assert(r == 0); txn = null_txn;
// verify the db
root_fifo_verify(env, n);
// cleanup
r = env->close(env, 0); assert(r == 0); env = null_env;
}
int test_main(int argc, const char *argv[]) {
int i;
int n = -1;
// parse_args(argc, argv);
for (i = 1; i < argc; i++) {
if (strcmp(argv[i], "-v") == 0) {
verbose = 1;
continue;
}
if (strcmp(argv[i], "-n") == 0) {
if (i+1 < argc)
n = atoi(argv[++i]);
continue;
}
}
if (n >= 0)
root_fifo_31(n);
else
for (i=0; i<100; i++)
root_fifo_31(i);
return 0;
}
// test txn commit after db close
#include "test.h"
#include <sys/stat.h>
#include "../ydb-internal.h"
DB_ENV *null_env = NULL;
DB *null_db = NULL;
DB_TXN *null_txn = NULL;
DBC *null_cursor = NULL;
static void create_non_empty(int n) {
DB_ENV *env = null_env;
int r;
r = db_env_create(&env, 0); assert(r == 0); assert(env != NULL);
r = env->set_data_dir(env, ENVDIR);
r = env->set_lg_dir(env, ENVDIR);
r = env->open(env,
0,
DB_INIT_MPOOL+DB_INIT_LOG+DB_INIT_LOCK+DB_INIT_TXN+DB_PRIVATE+DB_CREATE,
S_IRWXU+S_IRWXG+S_IRWXO);
assert(r == 0);
DB_TXN *txn = null_txn;
r = env->txn_begin(env, null_txn, &txn, 0); assert(r == 0); assert(txn != NULL);
DB *db = null_db;
r = db_create(&db, env, 0); assert(r == 0); assert(db != NULL);
r = db->open(db, txn, "test.db", 0, DB_BTREE, DB_CREATE, S_IRWXU+S_IRWXG+S_IRWXO);
assert(r == 0);
int i;
for (i=n; i<2*n; i++) {
DBT key, val;
int k = toku_htonl(i);
r = db->put(db, txn, dbt_init(&key, &k, sizeof k), dbt_init(&val, &i, sizeof i), 0);
assert(r == 0);
}
r = db->close(db, 0); assert(r == 0); db = null_db;
r = txn->commit(txn, 0); assert(r == 0); txn = null_txn;
r = env->close(env, 0); assert(r == 0); env = null_env;
}
static void root_fifo_verify(DB_ENV *env, int n) {
if (verbose) printf("%s:%d %d\n", __FUNCTION__, __LINE__, n);
int r;
DB *db = null_db;
r = db_create(&db, env, 0); assert(r == 0); assert(db != NULL);
r = db->open(db, null_txn, "test.db", 0, DB_BTREE, DB_CREATE, S_IRWXU+S_IRWXG+S_IRWXO);
assert(r == 0);
DB_TXN *txn = null_txn;
r = env->txn_begin(env, null_txn, &txn, 0); assert(r == 0); assert(txn != NULL);
DBC *cursor = null_cursor;
r = db->cursor(db, txn, &cursor, 0); assert(r == 0);
int i;
for (i = n; ; i++) {
DBT key, val;
memset(&key, 0, sizeof key); memset(&val, 0, sizeof val);
r = cursor->c_get(cursor, &key, &val, DB_NEXT);
if (r != 0) break;
int k;
assert(key.size == sizeof k);
memcpy(&k, key.data, key.size);
assert((int)toku_ntohl(k) == i);
}
assert(i == 2*n);
r = cursor->c_close(cursor); assert(r == 0); cursor = null_cursor;
r = txn->commit(txn, 0); assert(r == 0); txn = null_txn;
r = db->close(db, 0); assert(r == 0); db = null_db;
}
static void root_fifo_32(int n) {
if (verbose) printf("%s:%d %d\n", __FUNCTION__, __LINE__, n);
int r;
// create the env
system("rm -rf " ENVDIR);
toku_os_mkdir(ENVDIR, S_IRWXU+S_IRWXG+S_IRWXO);
// populate
create_non_empty(n);
DB_ENV *env = null_env;
r = db_env_create(&env, 0); assert(r == 0); assert(env != NULL);
r = env->set_data_dir(env, ENVDIR);
r = env->set_lg_dir(env, ENVDIR);
r = env->open(env,
0,
DB_INIT_MPOOL+DB_INIT_LOG+DB_INIT_LOCK+DB_INIT_TXN+DB_PRIVATE+DB_CREATE,
S_IRWXU+S_IRWXG+S_IRWXO);
assert(r == 0);
DB_TXN *txn = null_txn;
r = env->txn_begin(env, null_txn, &txn, 0); assert(r == 0); assert(txn != NULL);
int i;
for (i=0; i<n; i++) {
DB *db = null_db;
r = db_create(&db, env, 0); assert(r == 0); assert(db != NULL);
if (0) printf("db %p brt %p\n", db, db->i->brt);
r = db->open(db, txn, "test.db", 0, DB_BTREE, DB_CREATE, S_IRWXU+S_IRWXG+S_IRWXO);
assert(r == 0);
DBT key, val;
int k = toku_htonl(i);
r = db->put(db, txn, dbt_init(&key, &k, sizeof k), dbt_init(&val, &i, sizeof i), 0);
assert(r == 0);
r = db->close(db, 0); assert(r == 0); db = null_db;
}
r = txn->abort(txn); assert(r == 0); txn = null_txn;
// verify the db
root_fifo_verify(env, n);
// cleanup
r = env->close(env, 0); assert(r == 0); env = null_env;
}
int test_main(int argc, const char *argv[]) {
int i;
int n = -1;
// parse_args(argc, argv);
for (i = 1; i < argc; i++) {
if (strcmp(argv[i], "-v") == 0) {
verbose = 1;
continue;
}
if (strcmp(argv[i], "-n") == 0) {
if (i+1 < argc)
n = atoi(argv[++i]);
continue;
}
}
if (n >= 0)
root_fifo_32(n);
else
for (i=0; i<100; i++)
root_fifo_32(i);
return 0;
}
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment