diff --git a/src/tests/inflate2.c b/src/tests/inflate2.c new file mode 100644 index 00000000000..1714c220761 --- /dev/null +++ b/src/tests/inflate2.c @@ -0,0 +1,139 @@ +/* -*- mode: C; c-basic-offset: 4 -*- */ + +/* Idea: inflate a node by + * create a 2-level tree + * Nodes are A B C D E F G H + * Fill them up sequentially so they'll all be near 4MB. + * Close the file + * Insert some more to H (buffered in the root) + * Delete stuff from G (so that H merges with G) + * Merge first flushes G then merges the nodes, and maybe leaves G too big. + */ + +#include "test.h" + +DB_ENV *env; +DB *db; +const char dbname[] = "foo.db"; +const int envflags = DB_INIT_MPOOL|DB_CREATE|DB_THREAD |DB_INIT_LOCK|DB_INIT_LOG; + +static void +open_em (void) +{ + int r; + r = db_env_create(&env, 0); CKERR(r); + r = env->open(env, ENVDIR, envflags, S_IRWXU+S_IRWXG+S_IRWXO); CKERR(r); + r = db_create(&db, env, 0); CKERR(r); + r = db->open(db, NULL, dbname, NULL, DB_BTREE, DB_CREATE, 0666); CKERR(r); +} + +static void +close_em (void) +{ + int r; + r = db->close(db, 0); CKERR(r); + r = env->close(env, 0); CKERR(r); +} + +static void +reopen_em (void) +{ + close_em(); + open_em(); +} + + +static void +setup(void) +{ + system("rm -rf " ENVDIR); + toku_os_mkdir(ENVDIR, S_IRWXU+S_IRWXG+S_IRWXO); + int r; + r = db_env_create(&env, 0); CKERR(r); + r = env->open(env, ENVDIR, envflags, S_IRWXU+S_IRWXG+S_IRWXO); CKERR(r); + r = db_create(&db, env, 0); CKERR(r); + r = db->set_pagesize(db, 8192); CKERR(r); + r = db->open(db, NULL, dbname, NULL, DB_BTREE, DB_CREATE, 0666); CKERR(r); +} + +char vdata[150]; + +static void +insert_n (u_int32_t ah) { + u_int32_t an = htonl(ah); + DBT key = {.size = 4, .data=&an }; + DBT val = {.size = sizeof(vdata), .data=vdata}; + int r = db->put(db, NULL, &key, &val, DB_YESOVERWRITE); + CKERR(r); +} + +static void +get_n (u_int32_t ah, int expect_r) +{ + u_int32_t an = htonl(ah); + DBT key = {.size = 4, .data=&an }; + DBT val = {.data=0, .flags = DB_DBT_MALLOC}; + int r = db->get(db, NULL, &key, &val, 0); + assert(r==expect_r); + if (r==0) toku_free(val.data); +} + + +static void +delete_n_now (u_int32_t ah) +{ + u_int32_t an = htonl(ah); + DBT key = {.size = 4, .data=&an }; + int r = db->del(db, NULL, &key, DB_DELETE_ANY); +#ifdef USE_BDB + assert(r==0 || r==DB_NOTFOUND); +#else + CKERR(r); +#endif + get_n(ah, DB_NOTFOUND); +} + +static void +doit (void) +{ + u_int32_t N=46; + u_int32_t BIG=1<<16; + for (u_int32_t i=0; i<2*N; i++) { + insert_n(i); + } + insert_n(BIG); + insert_n(BIG+1); + reopen_em(); + for (u_int32_t i=0; i