use super::*;
use ruc::*;
macro_rules! s {
($i: expr) => {{ $i.as_bytes().to_vec() }};
}
#[test]
fn dagmaprawkey_functions() {
let mut i0 = DagMapRawKey::new(&mut Orphan::new(None)).unwrap();
i0.insert("k0", &s!("v0"));
assert_eq!(i0.get("k0").unwrap(), s!("v0"));
assert!(i0.get("k1").is_none());
let mut i0_raw = Orphan::new(Some(i0.into_inner()));
let mut i1 = DagMapRawKey::new(&mut i0_raw).unwrap();
i1.insert("k1", &s!("v1"));
assert_eq!(i1.get("k1").unwrap(), s!("v1"));
assert_eq!(i1.get("k0").unwrap(), s!("v0"));
let mut i1_raw = Orphan::new(Some(i1.into_inner()));
let mut i2 = DagMapRawKey::new(&mut i1_raw).unwrap();
i2.insert("k2", &s!("v2"));
assert_eq!(i2.get("k2").unwrap(), s!("v2"));
assert_eq!(i2.get("k1").unwrap(), s!("v1"));
assert_eq!(i2.get("k0").unwrap(), s!("v0"));
i2.insert("k2", &s!("v2x"));
assert_eq!(i2.get("k2").unwrap(), s!("v2x"));
assert_eq!(i2.get("k1").unwrap(), s!("v1"));
assert_eq!(i2.get("k0").unwrap(), s!("v0"));
i2.insert("k1", &s!("v1x"));
assert_eq!(i2.get("k2").unwrap(), s!("v2x"));
assert_eq!(i2.get("k1").unwrap(), s!("v1x"));
assert_eq!(i2.get("k0").unwrap(), s!("v0"));
i2.insert("k0", &s!("v0x"));
assert_eq!(i2.get("k2").unwrap(), s!("v2x"));
assert_eq!(i2.get("k1").unwrap(), s!("v1x"));
assert_eq!(i2.get("k0").unwrap(), s!("v0x"));
assert!(i1_raw.get_value().unwrap().get("k2").is_none());
assert_eq!(
i1_raw.get_value().unwrap().get("k1").unwrap(),
s!("v1").encode()
);
assert_eq!(
i1_raw.get_value().unwrap().get("k0").unwrap(),
s!("v0").encode()
);
assert!(i0_raw.get_value().unwrap().get("k2").is_none());
assert!(i0_raw.get_value().unwrap().get("k1").is_none());
assert_eq!(
i0_raw.get_value().unwrap().get("k0").unwrap(),
s!("v0").encode()
);
let mut head = i2.prune().unwrap();
sleep_ms!(1000);
assert_eq!(head.get("k2").unwrap(), s!("v2x"));
assert_eq!(head.get("k1").unwrap(), s!("v1x"));
assert_eq!(head.get("k0").unwrap(), s!("v0x"));
assert!(i1_raw.get_value().is_none());
assert!(i1_raw.get_value().is_none());
assert!(i1_raw.get_value().is_none());
assert!(i0_raw.get_value().is_none());
assert!(i0_raw.get_value().is_none());
assert!(i0_raw.get_value().is_none());
for i in 10u8..=255 {
head.insert(i.to_be_bytes(), &i.to_be_bytes().to_vec());
head = DagMapRawKey::new(&mut Orphan::new(Some(head.into_inner()))).unwrap();
}
let mut head = head.prune().unwrap();
sleep_ms!(1000);
for i in 10u8..=255 {
assert_eq!(head.get(i.to_be_bytes()).unwrap(), i.to_be_bytes().to_vec());
}
for i in 0u8..=254 {
head.remove(i.to_be_bytes());
assert!(head.get(i.to_be_bytes()).is_none());
}
*(head.get_mut(255u8.to_be_bytes()).unwrap()) = 0u8.to_be_bytes().to_vec();
assert_eq!(
head.get(255u8.to_be_bytes()).unwrap().as_slice(),
0u8.to_be_bytes()
);
}
#[test]
fn test_save_and_from_meta() {
let mut dag: DagMapRawKey<Vec<u8>> =
DagMapRawKey::new(&mut Orphan::new(None)).unwrap();
dag.insert("k1", &s!("v1"));
dag.insert("k2", &s!("v2"));
let id = dag.save_meta().unwrap();
assert_eq!(id, dag.instance_id());
let restored: DagMapRawKey<Vec<u8>> = DagMapRawKey::from_meta(id).unwrap();
assert_eq!(restored.get("k1").unwrap(), s!("v1"));
assert_eq!(restored.get("k2").unwrap(), s!("v2"));
}
#[test]
fn test_serde_roundtrip() {
let mut dag: DagMapRawKey<Vec<u8>> =
DagMapRawKey::new(&mut Orphan::new(None)).unwrap();
dag.insert("x", &s!("X"));
dag.insert("y", &s!("Y"));
let bytes = postcard::to_allocvec(&dag).unwrap();
let restored: DagMapRawKey<Vec<u8>> = postcard::from_bytes(&bytes).unwrap();
assert_eq!(restored.get("x").unwrap(), s!("X"));
assert_eq!(restored.get("y").unwrap(), s!("Y"));
}
#[test]
fn test_from_meta_nonexistent() {
assert!(DagMapRawKey::<Vec<u8>>::from_meta(u64::MAX).is_err());
}
#[test]
fn test_meta_with_parent_child() {
let mut p = DagMapRawKey::<Vec<u8>>::new(&mut Orphan::new(None)).unwrap();
p.insert("pk", &s!("pv"));
let mut c =
DagMapRawKey::<Vec<u8>>::new(&mut Orphan::new(Some(p.into_inner()))).unwrap();
c.insert("ck", &s!("cv"));
let id = c.save_meta().unwrap();
let restored: DagMapRawKey<Vec<u8>> = DagMapRawKey::from_meta(id).unwrap();
assert_eq!(restored.get("ck").unwrap(), s!("cv"));
assert_eq!(restored.get("pk").unwrap(), s!("pv"));
}