test cachetable_unpin_and_remove. addresses #1237

git-svn-id: file:///svn/toku/tokudb.1195@7612 c7de825b-a66e-492c-adef-691d508d4ae1

newbrt/tests/Makefile

@@ -136,6 +136,7 @@ REGRESSION_TESTS = \
 	cachetable-count-pinned-test \
 	cachetable-debug-test \
 	cachetable-debug-test \
+	cachetable-unpin-and-remove-test \
 	fifo-test \
 	list-test \
 	keyrange \

newbrt/tests/cachetable-unpin-and-remove-test.c

+#include "includes.h"
+#include "test.h"
+
+static void
+flush (CACHEFILE f __attribute__((__unused__)),
+       CACHEKEY k  __attribute__((__unused__)),
+       void *v     __attribute__((__unused__)),
+       void *e     __attribute__((__unused__)),
+       long s      __attribute__((__unused__)),
+       BOOL w      __attribute__((__unused__)),
+       BOOL keep   __attribute__((__unused__)),
+       LSN m       __attribute__((__unused__)),
+       BOOL r      __attribute__((__unused__))
+       ) {
+    /* Do nothing */
+}
+
+static int
+fetch (CACHEFILE f        __attribute__((__unused__)),
+       CACHEKEY k         __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 0;
+}
+
+// test simple unpin and remove
+static void
+cachetable_unpin_and_remove_test (int n) {
+    if (verbose) printf("%s %d\n", __FUNCTION__, n);
+    const int test_limit = 2*n;
+    int r;
+    int i;
+
+    CACHETABLE ct;
+    r = toku_create_cachetable(&ct, test_limit, ZERO_LSN, NULL_LOGGER); assert(r == 0);
+    char fname1[] = __FILE__ "test1.dat";
+    unlink(fname1);
+    CACHEFILE f1;
+    r = toku_cachetable_openf(&f1, ct, fname1, O_RDWR|O_CREAT, 0777); assert(r == 0);
+
+    // generate some random keys
+    CACHEKEY keys[n]; int nkeys = n;
+    for (i=0; i<n; i++) {
+        keys[i].b = random();
+    }
+
+    // put the keys into the cachetable
+    for (i=0; i<n; i++) {
+        u_int32_t hi = toku_cachetable_hash(f1, make_blocknum(keys[i].b));
+        r = toku_cachetable_put(f1, make_blocknum(keys[i].b), hi, (void *) keys[i].b, 1, flush, fetch, 0);
+        assert(r == 0);
+    }
+    
+    // unpin and remove
+    CACHEKEY oldkeys[n];
+    while (nkeys > 0) {
+        i = random() % nkeys;
+        u_int32_t hi = toku_cachetable_hash(f1, make_blocknum(keys[i].b));
+        r = toku_cachetable_unpin_and_remove(f1, keys[i]);
+        assert(r == 0);
+
+        toku_cachefile_verify(f1);
+
+        // verify that k is removed
+        void *v;
+        r = toku_cachetable_maybe_get_and_pin(f1, make_blocknum(keys[i].b), hi, &v);
+        assert(r != 0);
+
+        oldkeys[i] = keys[i]; keys[i] = keys[nkeys-1]; nkeys -= 1;
+    }
+
+    // verify that all are really removed
+    for (i=0; i<n; i++) {
+        r = toku_cachetable_unpin_and_remove(f1, keys[i]);
+        // assert(r != 0);
+        if (r == 0) printf("%s:%d warning %d\n", __FILE__, __LINE__, r);
+    }
+
+    // verify that the cachtable is empty
+    int nentries;
+    toku_cachetable_get_state(ct, &nentries, NULL, NULL, NULL);
+    assert(nentries == 0);
+
+    r = toku_cachefile_close(&f1, NULL_LOGGER); assert(r == 0 && f1 == 0);
+    r = toku_cachetable_close(&ct); assert(r == 0 && ct == 0);
+}
+
+// test remove when the pair in being written
+static void
+cachetable_put_evict_remove_test (int n) {
+    if (verbose) printf("%s %d\n", __FUNCTION__, n);
+    const int test_limit = 2*n;
+    int r;
+
+    CACHETABLE ct;
+    r = toku_create_cachetable(&ct, test_limit, ZERO_LSN, NULL_LOGGER); assert(r == 0);
+    char fname1[] = __FILE__ "test1.dat";
+    unlink(fname1);
+    CACHEFILE f1;
+    r = toku_cachetable_openf(&f1, ct, fname1, O_RDWR|O_CREAT, 0777); assert(r == 0);
+
+    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[]) {
+    default_parse_args(argc, argv);
+    cachetable_unpin_and_remove_test(8);
+    cachetable_put_evict_remove_test(8);
+    return 0;
+}