lmdb-rs/lmdb-sys/tests/simple.rs

extern crate lmdb_rkv_sys;

use lmdb_rkv_sys::*;

use std::ffi::{c_void, CString};
use std::ptr;

// https://github.com/victorporof/lmdb/blob/mdb.master/libraries/liblmdb/moz-test.c

macro_rules! E {
    ($expr:expr) => {{
        match $expr {
            ::MDB_SUCCESS => (),
            err_code => assert!(false, "Failed with code {}", err_code),
        }
    }};
}

macro_rules! str {
    ($expr:expr) => {
        ::CString::new($expr).unwrap().as_ptr()
    };
}

#[test]
#[cfg(all(target_os = "windows", target_arch = "x86"))]
#[should_panic(expected = "Failed with code -30793")]
fn test_simple_win_32() {
    test_simple()
}

#[test]
#[cfg(not(all(target_os = "windows", target_arch = "x86")))]
fn test_simple_other() {
    test_simple()
}

fn test_simple() {
    let mut env: *mut MDB_env = ptr::null_mut();
    let mut dbi: MDB_dbi = 0;
    let mut key = MDB_val {
        mv_size: 0,
        mv_data: ptr::null_mut(),
    };
    let mut data = MDB_val {
        mv_size: 0,
        mv_data: ptr::null_mut(),
    };
    let mut txn: *mut MDB_txn = ptr::null_mut();
    let sval = str!("foo") as *mut c_void;
    let dval = str!("bar") as *mut c_void;

    unsafe {
        E!(mdb_env_create(&mut env));
        E!(mdb_env_set_maxdbs(env, 2));
        E!(mdb_env_open(env, str!("./tests/fixtures/testdb"), 0, 0664));

        E!(mdb_txn_begin(env, ptr::null_mut(), 0, &mut txn));
        E!(mdb_dbi_open(txn, str!("subdb"), MDB_CREATE, &mut dbi));
        E!(mdb_txn_commit(txn));

        key.mv_size = 3;
        key.mv_data = sval;
        data.mv_size = 3;
        data.mv_data = dval;

        E!(mdb_txn_begin(env, ptr::null_mut(), 0, &mut txn));
        E!(mdb_put(txn, dbi, &mut key, &mut data, 0));
        E!(mdb_txn_commit(txn));

        mdb_dbi_close(env, dbi);
        mdb_env_close(env);
    }
}