mariadb/locktree/tests/test.h

/* -*- mode: C++; c-basic-offset: 4; indent-tabs-mode: nil -*- */
// vim: ft=cpp:expandtab:ts=8:sw=4:softtabstop=4:
#ident "$Id$"
#ident "Copyright (c) 2007-2012 Tokutek Inc.  All rights reserved."
#ident "The technology is licensed by the Massachusetts Institute of Technology, Rutgers State University of New Jersey, and the Research Foundation of State University of New York at Stony Brook under United States of America Serial No. 11/760379 and to the patents and/or patent applications resulting from it."

#ifndef TOKU_TEST_H
#define TOKU_TEST_H

#include <ft/ybt.h>
#include <limits.h>

namespace toku {

__attribute__((__unused__))
static DBT min_dbt(void) {
    static int64_t min = INT_MIN;
    DBT dbt;
    toku_fill_dbt(&dbt, &min, sizeof(int64_t));
    dbt.flags = DB_DBT_USERMEM;
    return dbt;
}

__attribute__((__unused__))
static DBT max_dbt(void) {
    static int64_t max = INT_MAX;
    DBT dbt;
    toku_fill_dbt(&dbt, &max, sizeof(int64_t));
    dbt.flags = DB_DBT_USERMEM;
    return dbt;
}

__attribute__((__unused__))
static const DBT *get_dbt(int64_t key) {
    static const int NUM_DBTS = 1000;
    static bool initialized;
    static int64_t static_ints[NUM_DBTS];
    static DBT static_dbts[NUM_DBTS];
    invariant(key < NUM_DBTS);
    if (!initialized) {
        for (int i = 0; i < NUM_DBTS; i++) {
            static_ints[i] = i;
            toku_fill_dbt(&static_dbts[i],
                    &static_ints[i],
                    sizeof(int64_t));
            static_dbts[i].flags = DB_DBT_USERMEM;
        }
        initialized = true;
    }

    invariant(key < NUM_DBTS);
    return &static_dbts[key];
}

__attribute__((__unused__))
static int compare_dbts(DB *db, const DBT *key1, const DBT *key2) {
    (void) db;

    // this emulates what a "infinity-aware" comparator object does
    if (toku_dbt_is_infinite(key1) || toku_dbt_is_infinite(key2)) {
        return toku_dbt_infinite_compare(key1, key2);
    } else {
        invariant(key1->size == sizeof(int64_t));
        invariant(key2->size == sizeof(int64_t));
        int64_t a = *(int64_t*) key1->data;
        int64_t b = *(int64_t*) key2->data;
        if (a < b) {
            return -1;
        } else if (a == b) {
            return 0;
        } else {
            return 1;
        }
    }
}

} /* namespace toku */

#endif