test938b.c 2.66 KB
Newer Older
1
/* -*- mode: C; c-basic-offset: 4 -*- */
2
#include <toku_portability.h>
Yoni Fogel's avatar
Yoni Fogel committed
3
#include <toku_portability.h>
4 5 6 7 8 9 10
#include <db.h>
#include <sys/stat.h>
#include "test.h"

DB_ENV *env;
DB *db;

11 12
static void
run (void) {
13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39
    int r;
    DB_TXN *txn;
    char v101=101, v102=102, v1=1, v2=2;
    int vN=0;
    int N=0;
    {
	r=env->txn_begin(env, 0, &txn, 0);                            CKERR(r);
	DBT k,v;
	int i;
	r=db->put(db, txn, dbt_init(&k, &v1, 1), dbt_init(&v, &v101, 1), DB_YESOVERWRITE); CKERR(r);
	r=db->put(db, txn, dbt_init(&k, &v2, 1), dbt_init(&v, &v102, 1), DB_YESOVERWRITE); CKERR(r);
	for (i=0; i<N; i++) {
	    int iv = htonl(i);
	    r=db->put(db, txn, dbt_init(&k, &vN, 1), dbt_init(&v, &iv, 4), DB_YESOVERWRITE); 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;
	int i;
 	for (i=0; i<N; i++) {
	    r=c->c_get(c, dbt_init_malloc(&k), dbt_init_malloc(&v), DB_NEXT); CKERR(r);
	    assert(k.size==1);          assert(v.size==4);
40
	    assert(*(char*)k.data==vN); assert((int)ntohl(*(int*)v.data)==i);
41 42
	}

43
	r=c->c_get(c, dbt_init(&k, 0, 0), dbt_init(&v, 0, 0), DB_NEXT);  CKERR(r);
44
	assert(*(char*)k.data==v1); assert(*(char*)v.data==v101);
45
	r=c->c_get(c, dbt_init(&k, 0, 0), dbt_init(&v, 0, 0), DB_NEXT);  CKERR(r);
46
	assert(*(char*)k.data==v2); assert(*(char*)v.data==v102);
47
	r=c->c_get(c, dbt_init(&k, 0, 0), dbt_init(&v, 0, 0), DB_NEXT);  assert(r!=0);
48 49 50 51 52
	r=c->c_close(c);                                                   CKERR(r);
	r=txn->commit(txn, 0);                                             CKERR(r);
    }
}

Yoni Fogel's avatar
Yoni Fogel committed
53 54
int
test_main(int argc, const char *argv[]) {
55 56 57
    parse_args(argc, argv);

    system("rm -rf " ENVDIR);
58
    toku_os_mkdir(ENVDIR, S_IRWXU+S_IRWXG+S_IRWXO);
59 60 61 62 63 64

    int r;

    DB_TXN *txn;
    {
        r = db_env_create(&env, 0);                                   CKERR(r);
65
	r=env->open(env, ENVDIR, DB_INIT_LOCK|DB_INIT_LOG|DB_INIT_MPOOL|DB_INIT_TXN|DB_CREATE|DB_PRIVATE, S_IRWXU+S_IRWXG+S_IRWXO); CKERR(r);
66 67 68 69
	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);
70
	r=db->open(db, txn, "foo.db", 0, DB_BTREE, DB_CREATE, S_IRWXU+S_IRWXG+S_IRWXO);  CKERR(r);
71 72 73 74 75 76 77 78 79 80
	r=txn->commit(txn, 0);                                        CKERR(r);
    }
    run();
    {
	r=db->close(db, 0);                                           CKERR(r);
	r=env->close(env, 0);                                         CKERR(r);
    }

    return 0;
}