use std::fs;
use cdb2::{CDB, CDBWriter};
macro_rules! noerr {
( $e:expr ) => {
if let Err(x) = $e {
panic!("{}", x);
}
};
}
#[test]
fn test_make() {
let filename = "tests/make.cdb";
let mut cdb = CDBWriter::create(filename).unwrap();
noerr!(cdb.add(b"one", b"Hello"));
noerr!(cdb.add(b"two", b"Goodbye"));
noerr!(cdb.add(b"one", b", World!"));
noerr!(cdb.add(b"this key will be split across two reads", b"Got it."));
noerr!(cdb.finish());
let cdb = CDB::open(filename).unwrap();
assert_eq!(cdb.find(b"two").next().unwrap().unwrap(), b"Goodbye");
assert_eq!(
cdb.find(b"this key will be split across two reads")
.next()
.unwrap()
.unwrap(),
b"Got it."
);
let mut i = cdb.find(b"one");
assert_eq!(i.next().unwrap().unwrap(), b"Hello");
assert_eq!(i.next().unwrap().unwrap(), b", World!");
let mut i = cdb.iter();
let next = i.next().unwrap().unwrap();
assert_eq!(next.0, b"one");
assert_eq!(next.1, b"Hello");
let next = i.next().unwrap().unwrap();
assert_eq!(next.0, b"two");
assert_eq!(next.1, b"Goodbye");
let next = i.next().unwrap().unwrap();
assert_eq!(next.0, b"one");
assert_eq!(next.1, b", World!");
let next = i.next().unwrap().unwrap();
assert_eq!(next.0, b"this key will be split across two reads");
assert_eq!(next.1, b"Got it.");
noerr!(fs::remove_file(filename));
}