Fix TinyVec::resize across inline/heap boundary

Lokathor · Jan 13, 2020 · 001b33e · 001b33e
1 parent d0bdff5
commit 001b33e
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/src/tiny_vec.rs b/src/tiny_vec.rs
@@ -405,7 +405,12 @@ impl<A: Arrayish> TinyVec<A> {
     A::Item: Clone,
   {
     match self {
-      TinyVec::Inline(a) => a.resize(new_len, new_val),
+      TinyVec::Inline(a) => if new_len > A::CAPACITY {
+        self.move_to_the_heap();
+        self.resize(new_len, new_val);
+      } else {
+        a.resize(new_len, new_val);
+      },
       TinyVec::Heap(v) => v.resize(new_len, new_val),
     }
   }

diff --git a/tests/tinyvec.rs b/tests/tinyvec.rs
@@ -3,7 +3,6 @@
 use tinyvec::*;
 use std::iter::FromIterator;
 
-#[test]
 fn TinyVec_drain() {
   let mut tv: TinyVec<[i32; 10]> = Default::default();
   tv.push(1);
@@ -31,3 +30,10 @@ fn TinyVec_drain() {
   assert_eq!(Vec::from_iter(tv.clone().drain(1..=1)), vec![2]);
   assert_eq!(Vec::from_iter(tv.clone().drain(1..=2)), vec![2, 3]);
 }
+
+#[test]
+fn TinyVec_resize() {
+  let mut tv: TinyVec<[i32; 10]> = Default::default();
+  tv.resize(20, 5);
+  assert_eq!(&tv[..], &[5; 20]);
+}