mariadb/newbrt/tests/cachetable-getandpin-test.c

#include <stdio.h>
#include <unistd.h>
#include <assert.h>

#include "test.h"
#include "cachetable.h"

void flush(CACHEFILE cf     __attribute__((__unused__)),
	   CACHEKEY key     __attribute__((__unused__)),
	   void *v          __attribute__((__unused__)),
	   void *extraargs  __attribute__((__unused__)),
	   long size        __attribute__((__unused__)),
	   BOOL write_me    __attribute__((__unused__)),
	   BOOL keep_me     __attribute__((__unused__)),
	   LSN lsn          __attribute__((__unused__)),
	   BOOL rename_p    __attribute__((__unused__))
	   ) {
    assert((long) key.b == size);
    if (!keep_me) free(v);
}

int fetch(CACHEFILE cf, CACHEKEY key, u_int32_t hash, void **vptr, long *sizep, void *extra, LSN *written_lsn) {
    cf = cf; hash = hash; extra = extra; written_lsn = written_lsn;
    *sizep = (long) key.b;
    *vptr = malloc(*sizep);
    return 0;
}

int fetch_error(CACHEFILE cf       __attribute__((__unused__)),
		CACHEKEY key       __attribute__((__unused__)),
		u_int32_t fullhash __attribute__((__unused__)),
		void **value       __attribute__((__unused__)),
		long *sizep        __attribute__((__unused__)),
		void*extraargs     __attribute__((__unused__)),
		LSN *written_lsn   __attribute__((__unused__))
		) {
    return -1;
}

void cachetable_getandpin_test(int n) {
    const int test_limit = 1024*1024;
    int r;
    CACHETABLE ct;
    r = toku_create_cachetable(&ct, test_limit, ZERO_LSN, NULL_LOGGER); assert(r == 0);
    char fname1[] = __FILE__ "test_getandpin.dat";
    unlink(fname1);
    CACHEFILE f1;
    r = toku_cachetable_openf(&f1, ct, fname1, O_RDWR|O_CREAT, 0777); assert(r == 0);

    int i;

    // test get_and_pin fails
    for (i=1; i<=n; i++) {
        u_int32_t hi;
        hi = toku_cachetable_hash(f1, make_blocknum(i));
        void *v; long size;
        r = toku_cachetable_get_and_pin(f1, make_blocknum(i), hi, &v, &size, flush, fetch_error, 0);
        assert(r == -1);
    }

    // test get_and_pin size
    for (i=1; i<=n; i++) {
        u_int32_t hi;
        hi = toku_cachetable_hash(f1, make_blocknum(i));
        void *v; long size;
        r = toku_cachetable_get_and_pin(f1, make_blocknum(i), hi, &v, &size, flush, fetch, 0);
        assert(r == 0);
        assert(size == i);

        r = toku_cachetable_unpin(f1, make_blocknum(i), hi, CACHETABLE_CLEAN, i);
        assert(r == 0);
    }
    toku_cachetable_verify(ct);

    r = toku_cachefile_close(&f1, NULL_LOGGER); assert(r == 0 && f1 == 0);
    r = toku_cachetable_close(&ct); assert(r == 0 && ct == 0);
}

int main(int argc, const char *argv[]) {
    int i;
    for (i=1; i<argc; i++) {
        if (strcmp(argv[i], "-v") == 0) {
            verbose++;
            continue;
        }
    }
    cachetable_getandpin_test(8);
    return 0;
}