Add tests and fix edge case

voltrevo · Aug 17, 2023 · f59f68a · f59f68a
1 parent 8313dcc
commit f59f68a
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/radix_tree/src/lib.rs b/radix_tree/src/lib.rs
@@ -115,8 +115,18 @@ mod tests {
       tree.push(i);
     }
 
-    tree.truncate(50);
+    assert_eq!(tree.len(), 100);
+
+    tree.truncate(100);
+    assert_eq!(tree.len(), 100);
+
+    tree.truncate(101);
+    assert_eq!(tree.len(), 100);
 
+    tree.truncate(1000);
+    assert_eq!(tree.len(), 100);
+
+    tree.truncate(50);
     assert_eq!(tree.len(), 50);
     assert_eq!(tree.depth(), 3);
 
@@ -125,7 +135,6 @@ mod tests {
     }
 
     tree.truncate(1);
-
     assert_eq!(tree.len(), 1);
     assert_eq!(tree.depth(), 1);
   }

diff --git a/radix_tree/src/radix_tree.rs b/radix_tree/src/radix_tree.rs
@@ -193,6 +193,10 @@ impl<T: Clone, const N: usize> RadixTree<T, N> {
     for p in path {
       match tree.data_mut() {
         RadixTreeData::Meta(meta) => {
+          if meta.len() <= p {
+            break;
+          }
+
           meta.truncate(p + 1);
           tree = &mut meta[p];
         }