use mmdb_core::{mmdb_set_base_dir, MapxRaw};
use ruc::*;
use std::borrow::Cow;

#[test]
fn basic_cases() {
    let cnt = 200;
    info_omit!(mmdb_set_base_dir(&format!(
        "/tmp/mmdb_testing/{}",
        rand::random::<u64>()
    )));

    let hdr = {
        let mut hdr_i = MapxRaw::new();

        assert_eq!(0, hdr_i.len());
        (0..cnt).for_each(|i: usize| {
            assert!(hdr_i.get(&i.to_be_bytes()).is_none());
        });

        (0..cnt)
            .map(|i: usize| (i.to_be_bytes(), i.to_be_bytes()))
            .for_each(|(i, b)| {
                hdr_i.entry(&i).or_insert(&b);
                assert_eq!(&hdr_i.get(&i).unwrap()[..], &i[..]);
                assert_eq!(&hdr_i.remove(&i).unwrap()[..], &b[..]);
                assert!(hdr_i.get(&i).is_none());
                assert!(hdr_i.insert(&i, &b).is_none());
                assert!(hdr_i.insert(&i, &b).is_some());
            });

        assert_eq!(cnt, hdr_i.len());

        pnk!(msgpack::to_vec(&hdr_i))
    };

    let mut reloaded = pnk!(msgpack::from_slice::<MapxRaw>(&hdr));

    assert_eq!(cnt, reloaded.len());

    (0..cnt).map(|i: usize| i.to_be_bytes()).for_each(|i| {
        assert_eq!(&i[..], &reloaded.get(&i).unwrap()[..]);
    });

    (1..cnt).map(|i: usize| i.to_be_bytes()).for_each(|i| {
        *reloaded.get_mut(&i).unwrap() = i.to_vec();
        assert_eq!(&reloaded.get(&i).unwrap()[..], &i[..]);
        assert!(reloaded.contains_key(&i));
        assert!(reloaded.remove(&i).is_some());
        assert!(!reloaded.contains_key(&i));
    });

    assert_eq!(1, reloaded.len());
    reloaded.clear();
    assert!(reloaded.is_empty());

    reloaded.insert(&[1], &[1]);
    reloaded.insert(&[4], &[4]);
    reloaded.insert(&[6], &[6]);
    reloaded.insert(&[80], &[80]);

    assert!(
        reloaded
            .range(Cow::Borrowed(&[][..])..Cow::Borrowed(&[1][..]))
            .next()
            .is_none()
    );
    assert_eq!(
        vec![4],
        reloaded
            .range(Cow::Borrowed(&[2][..])..Cow::Borrowed(&[10][..]))
            .next()
            .unwrap()
            .1
    );
    assert_eq!(
        vec![6],
        reloaded
            .range(Cow::Borrowed(&[2][..])..Cow::Borrowed(&[10][..]))
            .rev()
            .next()
            .unwrap()
            .1
    );

    assert_eq!(vec![80], reloaded.get_ge(&[79]).unwrap().1);
    assert_eq!(vec![80], reloaded.get_ge(&[80]).unwrap().1);
    assert_eq!(vec![80], reloaded.get_le(&[80]).unwrap().1);
    assert_eq!(vec![80], reloaded.get_le(&[100]).unwrap().1);
}