use utils::{open_database,tmpdir,db_put_simple};
use leveldb::snapshots::Snapshots;
use leveldb::options::ReadOptions;
use leveldb::iterator::Iterable;

#[test]
fn test_snapshots() {
  let tmp = tmpdir("snapshots");
  let database = &mut open_database(tmp.path(), true);
  db_put_simple(database, [1], &[1]);
  let snapshot = database.snapshot();
  db_put_simple(database, [2], &[2]);
  let read_opts = ReadOptions::new();
  let res = snapshot.get(read_opts, [2]);
  assert!(res.is_ok());
  assert_eq!(None, res.unwrap());
}

#[test]
fn test_snapshot_iterator() {
  let tmp = tmpdir("snap_iterator");
  let database = &mut open_database(tmp.path(), true);
  db_put_simple(database, [1], &[1]);
  let snapshot = database.snapshot();
  db_put_simple(database, [2], &[2]);
  let read_opts = ReadOptions::new();
  let mut iter = snapshot.iter(read_opts);
  {
    let (key, _) = iter.next().unwrap();
    assert_eq!([1].to_vec().as_slice(), key);
  }
  {
    let next = iter.next();
    assert_eq!(None, next);
  }
}