36 lines
1.2 KiB
Rust
36 lines
1.2 KiB
Rust
use super::super::file_linked::FileLinked;
|
|
|
|
use std::fs;
|
|
|
|
#[test]
|
|
fn test_mutate() -> Result<(), String> {
|
|
let tree = btree!(1, btree!(2), btree!(3, btree!(4),));
|
|
let mut linked_tree = FileLinked::new(tree, String::from("test.txt"))?;
|
|
|
|
assert_eq!(
|
|
format!("{}", linked_tree.readonly()),
|
|
"{\"val\":1,\"left\":{\"val\":2,\"left\":null,\"right\":null},\"right\":{\"val\":3,\"left\":{\"val\":4,\"left\":null,\"right\":null},\"right\":null}}"
|
|
);
|
|
|
|
linked_tree.mutate(|v1| v1.val = 10)?;
|
|
|
|
assert_eq!(
|
|
format!("{}", linked_tree.readonly()),
|
|
"{\"val\":10,\"left\":{\"val\":2,\"left\":null,\"right\":null},\"right\":{\"val\":3,\"left\":{\"val\":4,\"left\":null,\"right\":null},\"right\":null}}"
|
|
);
|
|
|
|
linked_tree.mutate(|v1| {
|
|
let mut left = v1.left.clone().unwrap();
|
|
left.val = 13;
|
|
v1.left = Some(left);
|
|
})?;
|
|
|
|
assert_eq!(
|
|
format!("{}", linked_tree.readonly()),
|
|
"{\"val\":10,\"left\":{\"val\":13,\"left\":null,\"right\":null},\"right\":{\"val\":3,\"left\":{\"val\":4,\"left\":null,\"right\":null},\"right\":null}}"
|
|
);
|
|
|
|
fs::remove_file("test.txt").expect("Unable to remove file");
|
|
|
|
Ok(())
|
|
}
|