Issue with modified FlexVec items

agerasev · Jun 12, 2024 · 4c11be3 · 4c11be3
1 parent ac3ba2b
commit 4c11be3
Showing 1 changed file with 17 additions and 1 deletion.
diff --git a/containers/src/flex.rs b/containers/src/flex.rs
@@ -14,7 +14,6 @@ where
     T: Flat + ?Sized,
     L: Flat + Length,
 {
-    _ghost: PhantomData<T>,
     len: L,
     _align: [FlexVecAlignAs<T, L>; 0],
     data: [u8],
@@ -445,4 +444,21 @@ mod tests {
         assert_eq!(flex_vec.iter().nth(2).unwrap().as_slice(), [2].as_slice());
         assert!(flex_vec.iter().nth(3).is_none());
     }
+
+    #[test]
+    fn push_modify() {
+        let mut bytes = AlignedBytes::new(4 + 4 * 6, 4);
+        let flex_vec = FlexVec::<FlatVec<i32, u16>, u16>::default_in_place(&mut bytes).unwrap();
+        assert_eq!(FlexVec::<FlatVec<i32, u16>, u16>::DATA_OFFSET, flex_vec.size());
+
+        flex_vec.push_default().unwrap().extend_from_slice(&[0, 1]).unwrap();
+        flex_vec.push_default().unwrap().extend_from_slice(&[2]).unwrap();
+
+        assert_eq!(flex_vec.iter_mut().next().unwrap().pop(), Some(1));
+
+        assert_eq!(flex_vec.len(), 2);
+        assert_eq!(flex_vec.iter().next().unwrap().as_slice(), [0].as_slice());
+        assert_eq!(flex_vec.iter().nth(1).unwrap().as_slice(), [2].as_slice());
+        assert!(flex_vec.iter().nth(2).is_none());
+    }
 }