use btree_plus_store::{BTreeMap, BTreeStore};
use rand::{rngs::SmallRng, seq::SliceRandom, SeedableRng};

const SEED: &'static [u8; 32] = b"testseedtestseedtestseedtestseed";

#[test]
pub fn insert() {
    let store = BTreeStore::new();
    let mut btree = BTreeMap::new_in(&store);

    for (key, value) in &ITEMS {
        if let Some(_) = btree.insert(*key, *value) {
            println!("duplicate: {}", key);
        }
        btree.validate();
        println!("{:?}", btree);
    }

    assert_eq!(btree.len(), 100);
}

#[test]
pub fn remove() {
    let store = BTreeStore::new();
    let mut btree = BTreeMap::new_in(&store);

    let mut items = ITEMS;

    for (key, value) in &items {
        btree.insert(*key, *value);
    }

    let mut rng = SmallRng::from_seed(*SEED);
    items.shuffle(&mut rng);

    for (key, _) in &items {
        btree.remove(&key);
        btree.validate();
        println!("{:?}", btree);
    }

    assert!(btree.is_empty())
}

#[test]
pub fn update() {
    let store = BTreeStore::new();
    let mut btree = BTreeMap::new_in(&store);

    for (key, value) in &ITEMS {
        if key % 2 == 0 {
            btree.insert(*key, *value);
        }
    }

    for (key, value) in &ITEMS {
        btree.update(*key, |current_value| match current_value {
            Some(current_value) => {
                if current_value % 2 == 0 {
                    None
                } else {
                    Some(10000 - *value)
                }
            }
            None => Some(*value),
        });

        btree.validate();
        println!("{:?}", btree);
    }

    for (key, value) in &ITEMS {
        let shoud_be_present = *key % 2 == 1 || *value % 2 == 1;

        match btree.get(key) {
            Some(current_value) => {
                if !shoud_be_present {
                    panic!("binding {}:{} should not be present", *key, *value);
                }

                if *key % 2 == 0 && *value % 2 == 1 {
                    assert_eq!(*current_value, 10000 - *value)
                } else {
                    assert_eq!(*current_value, *value)
                }
            }
            None => {
                if shoud_be_present {
                    panic!("binding {}:{} should be present", *key, *value);
                }
            }
        }
    }
}

const ITEMS: [(usize, usize); 100] = [
    (4223, 5948),
    (8175, 4629),
    (1411, 7458),
    (9208, 4040),
    (1246, 2287),
    (6568, 7583),
    (5426, 491),
    (7850, 8789),
    (2034, 9388),
    (1408, 7331),
    (7346, 5820),
    (9712, 4253),
    (5430, 7253),
    (1662, 5278),
    (9322, 777),
    (9256, 8116),
    (7971, 8071),
    (648, 3082),
    (7510, 2207),
    (8394, 7839),
    (57, 8834),
    (7770, 5437),
    (6388, 6755),
    (9177, 9904),
    (6487, 5143),
    (2231, 688),
    (7389, 4472),
    (577, 1930),
    (9130, 3222),
    (2230, 8268),
    (1211, 2354),
    (9237, 3643),
    (2912, 8471),
    (8783, 4977),
    (4325, 9566),
    (9355, 528),
    (9814, 9342),
    (1641, 6027),
    (3009, 8304),
    (4199, 2688),
    (7011, 9579),
    (8391, 8562),
    (1097, 5448),
    (1224, 5844),
    (5309, 2846),
    (7493, 8845),
    (3682, 48),
    (9165, 2755),
    (9959, 7420),
    (8158, 2616),
    (3210, 7795),
    (4418, 7790),
    (5592, 4184),
    (4111, 885),
    (742, 952),
    (2486, 6088),
    (6797, 271),
    (8829, 3005),
    (6444, 5818),
    (6566, 8783),
    (913, 2886),
    (2325, 1260),
    (4382, 3045),
    (5451, 1473),
    (9376, 8133),
    (9036, 4924),
    (5202, 7364),
    (9190, 5619),
    (8190, 2892),
    (9493, 500),
    (3043, 8315),
    (9220, 6396),
    (6400, 5692),
    (2709, 8547),
    (1218, 7403),
    (581, 117),
    (2577, 9373),
    (9349, 3186),
    (9021, 4874),
    (4207, 1781),
    (5201, 5305),
    (7889, 1996),
    (6327, 6377),
    (8120, 2338),
    (8213, 9072),
    (865, 6524),
    (5858, 5331),
    (1904, 3594),
    (9950, 8859),
    (518, 6551),
    (2674, 7081),
    (9848, 618),
    (5120, 5595),
    (259, 9662),
    (3077, 863),
    (4519, 7217),
    (3931, 6743),
    (2575, 6810),
    (1553, 5964),
    (4493, 3677),
];