Fix using grow to the same size.

servo · Jun 7, 2019 · b28c6b3 · b28c6b3
1 parent f96322b
commit b28c6b3
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/lib.rs b/lib.rs
@@ -665,6 +665,8 @@ impl<A: Array> SmallVec<A> {
                 if unspilled {
                     return;
                 }
+            } else {
+                return;
             }
             deallocate(ptr, cap);
         }
@@ -2329,4 +2331,18 @@ mod tests {
         v.push(4);
         assert_eq!(v[..], [4]);
     }
+
+    #[test]
+    fn grow_spilled_same_size() {
+        let mut v: SmallVec<[u8; 2]> = SmallVec::new();
+        v.push(0);
+        v.push(1);
+        v.push(2);
+        assert!(v.spilled());
+        assert_eq!(v.capacity(), 4);
+        // grow with the same capacity
+        v.grow(4);
+        assert_eq!(v.capacity(), 4);
+        assert_eq!(v[..], [0, 1, 2]);
+    }
 }