• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 use core::alloc::{Allocator, Layout};
2 use core::assert_eq;
3 use core::iter::IntoIterator;
4 use core::num::NonZeroUsize;
5 use core::ptr::NonNull;
6 use std::alloc::System;
7 use std::assert_matches::assert_matches;
8 use std::borrow::Cow;
9 use std::cell::Cell;
10 use std::collections::TryReserveErrorKind::*;
11 use std::fmt::Debug;
12 use std::hint;
13 use std::iter::InPlaceIterable;
14 use std::mem;
15 use std::mem::{size_of, swap};
16 use std::ops::Bound::*;
17 use std::panic::{catch_unwind, AssertUnwindSafe};
18 use std::rc::Rc;
19 use std::sync::atomic::{AtomicU32, Ordering};
20 use std::vec::{Drain, IntoIter};
21 
22 struct DropCounter<'a> {
23     count: &'a mut u32,
24 }
25 
26 impl Drop for DropCounter<'_> {
drop(&mut self)27     fn drop(&mut self) {
28         *self.count += 1;
29     }
30 }
31 
32 #[test]
test_small_vec_struct()33 fn test_small_vec_struct() {
34     assert_eq!(size_of::<Vec<u8>>(), size_of::<usize>() * 3);
35 }
36 
37 #[test]
test_double_drop()38 fn test_double_drop() {
39     struct TwoVec<T> {
40         x: Vec<T>,
41         y: Vec<T>,
42     }
43 
44     let (mut count_x, mut count_y) = (0, 0);
45     {
46         let mut tv = TwoVec { x: Vec::new(), y: Vec::new() };
47         tv.x.push(DropCounter { count: &mut count_x });
48         tv.y.push(DropCounter { count: &mut count_y });
49 
50         // If Vec had a drop flag, here is where it would be zeroed.
51         // Instead, it should rely on its internal state to prevent
52         // doing anything significant when dropped multiple times.
53         drop(tv.x);
54 
55         // Here tv goes out of scope, tv.y should be dropped, but not tv.x.
56     }
57 
58     assert_eq!(count_x, 1);
59     assert_eq!(count_y, 1);
60 }
61 
62 #[test]
test_reserve()63 fn test_reserve() {
64     let mut v = Vec::new();
65     assert_eq!(v.capacity(), 0);
66 
67     v.reserve(2);
68     assert!(v.capacity() >= 2);
69 
70     for i in 0..16 {
71         v.push(i);
72     }
73 
74     assert!(v.capacity() >= 16);
75     v.reserve(16);
76     assert!(v.capacity() >= 32);
77 
78     v.push(16);
79 
80     v.reserve(16);
81     assert!(v.capacity() >= 33)
82 }
83 
84 #[test]
test_zst_capacity()85 fn test_zst_capacity() {
86     assert_eq!(Vec::<()>::new().capacity(), usize::MAX);
87 }
88 
89 #[test]
test_indexing()90 fn test_indexing() {
91     let v: Vec<isize> = vec![10, 20];
92     assert_eq!(v[0], 10);
93     assert_eq!(v[1], 20);
94     let mut x: usize = 0;
95     assert_eq!(v[x], 10);
96     assert_eq!(v[x + 1], 20);
97     x = x + 1;
98     assert_eq!(v[x], 20);
99     assert_eq!(v[x - 1], 10);
100 }
101 
102 #[test]
test_debug_fmt()103 fn test_debug_fmt() {
104     let vec1: Vec<isize> = vec![];
105     assert_eq!("[]", format!("{:?}", vec1));
106 
107     let vec2 = vec![0, 1];
108     assert_eq!("[0, 1]", format!("{:?}", vec2));
109 
110     let slice: &[isize] = &[4, 5];
111     assert_eq!("[4, 5]", format!("{slice:?}"));
112 }
113 
114 #[test]
test_push()115 fn test_push() {
116     let mut v = vec![];
117     v.push(1);
118     assert_eq!(v, [1]);
119     v.push(2);
120     assert_eq!(v, [1, 2]);
121     v.push(3);
122     assert_eq!(v, [1, 2, 3]);
123 }
124 
125 #[test]
test_extend()126 fn test_extend() {
127     let mut v = Vec::new();
128     let mut w = Vec::new();
129 
130     v.extend(w.clone());
131     assert_eq!(v, &[]);
132 
133     v.extend(0..3);
134     for i in 0..3 {
135         w.push(i)
136     }
137 
138     assert_eq!(v, w);
139 
140     v.extend(3..10);
141     for i in 3..10 {
142         w.push(i)
143     }
144 
145     assert_eq!(v, w);
146 
147     v.extend(w.clone()); // specializes to `append`
148     assert!(v.iter().eq(w.iter().chain(w.iter())));
149 
150     // Zero sized types
151     #[derive(PartialEq, Debug)]
152     struct Foo;
153 
154     let mut a = Vec::new();
155     let b = vec![Foo, Foo];
156 
157     a.extend(b);
158     assert_eq!(a, &[Foo, Foo]);
159 
160     // Double drop
161     let mut count_x = 0;
162     {
163         let mut x = Vec::new();
164         let y = vec![DropCounter { count: &mut count_x }];
165         x.extend(y);
166     }
167     assert_eq!(count_x, 1);
168 }
169 
170 #[test]
test_extend_from_slice()171 fn test_extend_from_slice() {
172     let a: Vec<isize> = vec![1, 2, 3, 4, 5];
173     let b: Vec<isize> = vec![6, 7, 8, 9, 0];
174 
175     let mut v: Vec<isize> = a;
176 
177     v.extend_from_slice(&b);
178 
179     assert_eq!(v, [1, 2, 3, 4, 5, 6, 7, 8, 9, 0]);
180 }
181 
182 #[test]
test_extend_ref()183 fn test_extend_ref() {
184     let mut v = vec![1, 2];
185     v.extend(&[3, 4, 5]);
186 
187     assert_eq!(v.len(), 5);
188     assert_eq!(v, [1, 2, 3, 4, 5]);
189 
190     let w = vec![6, 7];
191     v.extend(&w);
192 
193     assert_eq!(v.len(), 7);
194     assert_eq!(v, [1, 2, 3, 4, 5, 6, 7]);
195 }
196 
197 #[test]
test_slice_from_ref()198 fn test_slice_from_ref() {
199     let values = vec![1, 2, 3, 4, 5];
200     let slice = &values[1..3];
201 
202     assert_eq!(slice, [2, 3]);
203 }
204 
205 #[test]
test_slice_from_mut()206 fn test_slice_from_mut() {
207     let mut values = vec![1, 2, 3, 4, 5];
208     {
209         let slice = &mut values[2..];
210         assert!(slice == [3, 4, 5]);
211         for p in slice {
212             *p += 2;
213         }
214     }
215 
216     assert!(values == [1, 2, 5, 6, 7]);
217 }
218 
219 #[test]
test_slice_to_mut()220 fn test_slice_to_mut() {
221     let mut values = vec![1, 2, 3, 4, 5];
222     {
223         let slice = &mut values[..2];
224         assert!(slice == [1, 2]);
225         for p in slice {
226             *p += 1;
227         }
228     }
229 
230     assert!(values == [2, 3, 3, 4, 5]);
231 }
232 
233 #[test]
test_split_at_mut()234 fn test_split_at_mut() {
235     let mut values = vec![1, 2, 3, 4, 5];
236     {
237         let (left, right) = values.split_at_mut(2);
238         {
239             let left: &[_] = left;
240             assert!(&left[..left.len()] == &[1, 2]);
241         }
242         for p in left {
243             *p += 1;
244         }
245 
246         {
247             let right: &[_] = right;
248             assert!(&right[..right.len()] == &[3, 4, 5]);
249         }
250         for p in right {
251             *p += 2;
252         }
253     }
254 
255     assert_eq!(values, [2, 3, 5, 6, 7]);
256 }
257 
258 #[test]
test_clone()259 fn test_clone() {
260     let v: Vec<i32> = vec![];
261     let w = vec![1, 2, 3];
262 
263     assert_eq!(v, v.clone());
264 
265     let z = w.clone();
266     assert_eq!(w, z);
267     // they should be disjoint in memory.
268     assert!(w.as_ptr() != z.as_ptr())
269 }
270 
271 #[test]
test_clone_from()272 fn test_clone_from() {
273     let mut v = vec![];
274     let three: Vec<Box<_>> = vec![Box::new(1), Box::new(2), Box::new(3)];
275     let two: Vec<Box<_>> = vec![Box::new(4), Box::new(5)];
276     // zero, long
277     v.clone_from(&three);
278     assert_eq!(v, three);
279 
280     // equal
281     v.clone_from(&three);
282     assert_eq!(v, three);
283 
284     // long, short
285     v.clone_from(&two);
286     assert_eq!(v, two);
287 
288     // short, long
289     v.clone_from(&three);
290     assert_eq!(v, three)
291 }
292 
293 #[test]
test_retain()294 fn test_retain() {
295     let mut vec = vec![1, 2, 3, 4];
296     vec.retain(|&x| x % 2 == 0);
297     assert_eq!(vec, [2, 4]);
298 }
299 
300 #[test]
test_retain_predicate_order()301 fn test_retain_predicate_order() {
302     for to_keep in [true, false] {
303         let mut number_of_executions = 0;
304         let mut vec = vec![1, 2, 3, 4];
305         let mut next_expected = 1;
306         vec.retain(|&x| {
307             assert_eq!(next_expected, x);
308             next_expected += 1;
309             number_of_executions += 1;
310             to_keep
311         });
312         assert_eq!(number_of_executions, 4);
313     }
314 }
315 
316 #[test]
317 #[cfg_attr(not(panic = "unwind"), ignore = "test requires unwinding support")]
test_retain_pred_panic_with_hole()318 fn test_retain_pred_panic_with_hole() {
319     let v = (0..5).map(Rc::new).collect::<Vec<_>>();
320     catch_unwind(AssertUnwindSafe(|| {
321         let mut v = v.clone();
322         v.retain(|r| match **r {
323             0 => true,
324             1 => false,
325             2 => true,
326             _ => panic!(),
327         });
328     }))
329     .unwrap_err();
330     // Everything is dropped when predicate panicked.
331     assert!(v.iter().all(|r| Rc::strong_count(r) == 1));
332 }
333 
334 #[test]
335 #[cfg_attr(not(panic = "unwind"), ignore = "test requires unwinding support")]
test_retain_pred_panic_no_hole()336 fn test_retain_pred_panic_no_hole() {
337     let v = (0..5).map(Rc::new).collect::<Vec<_>>();
338     catch_unwind(AssertUnwindSafe(|| {
339         let mut v = v.clone();
340         v.retain(|r| match **r {
341             0 | 1 | 2 => true,
342             _ => panic!(),
343         });
344     }))
345     .unwrap_err();
346     // Everything is dropped when predicate panicked.
347     assert!(v.iter().all(|r| Rc::strong_count(r) == 1));
348 }
349 
350 #[test]
351 #[cfg_attr(not(panic = "unwind"), ignore = "test requires unwinding support")]
test_retain_drop_panic()352 fn test_retain_drop_panic() {
353     struct Wrap(Rc<i32>);
354 
355     impl Drop for Wrap {
356         fn drop(&mut self) {
357             if *self.0 == 3 {
358                 panic!();
359             }
360         }
361     }
362 
363     let v = (0..5).map(|x| Rc::new(x)).collect::<Vec<_>>();
364     catch_unwind(AssertUnwindSafe(|| {
365         let mut v = v.iter().map(|r| Wrap(r.clone())).collect::<Vec<_>>();
366         v.retain(|w| match *w.0 {
367             0 => true,
368             1 => false,
369             2 => true,
370             3 => false, // Drop panic.
371             _ => true,
372         });
373     }))
374     .unwrap_err();
375     // Other elements are dropped when `drop` of one element panicked.
376     // The panicked wrapper also has its Rc dropped.
377     assert!(v.iter().all(|r| Rc::strong_count(r) == 1));
378 }
379 
380 #[test]
test_retain_maybeuninits()381 fn test_retain_maybeuninits() {
382     // This test aimed to be run under miri.
383     use core::mem::MaybeUninit;
384     let mut vec: Vec<_> = [1i32, 2, 3, 4].map(|v| MaybeUninit::new(vec![v])).into();
385     vec.retain(|x| {
386         // SAFETY: Retain must visit every element of Vec in original order and exactly once.
387         // Our values is initialized at creation of Vec.
388         let v = unsafe { x.assume_init_ref()[0] };
389         if v & 1 == 0 {
390             return true;
391         }
392         // SAFETY: Value is initialized.
393         // Value wouldn't be dropped by `Vec::retain`
394         // because `MaybeUninit` doesn't drop content.
395         drop(unsafe { x.assume_init_read() });
396         false
397     });
398     let vec: Vec<i32> = vec
399         .into_iter()
400         .map(|x| unsafe {
401             // SAFETY: All values dropped in retain predicate must be removed by `Vec::retain`.
402             // Remaining values are initialized.
403             x.assume_init()[0]
404         })
405         .collect();
406     assert_eq!(vec, [2, 4]);
407 }
408 
409 #[test]
test_dedup()410 fn test_dedup() {
411     fn case(a: Vec<i32>, b: Vec<i32>) {
412         let mut v = a;
413         v.dedup();
414         assert_eq!(v, b);
415     }
416     case(vec![], vec![]);
417     case(vec![1], vec![1]);
418     case(vec![1, 1], vec![1]);
419     case(vec![1, 2, 3], vec![1, 2, 3]);
420     case(vec![1, 1, 2, 3], vec![1, 2, 3]);
421     case(vec![1, 2, 2, 3], vec![1, 2, 3]);
422     case(vec![1, 2, 3, 3], vec![1, 2, 3]);
423     case(vec![1, 1, 2, 2, 2, 3, 3], vec![1, 2, 3]);
424 }
425 
426 #[test]
test_dedup_by_key()427 fn test_dedup_by_key() {
428     fn case(a: Vec<i32>, b: Vec<i32>) {
429         let mut v = a;
430         v.dedup_by_key(|i| *i / 10);
431         assert_eq!(v, b);
432     }
433     case(vec![], vec![]);
434     case(vec![10], vec![10]);
435     case(vec![10, 11], vec![10]);
436     case(vec![10, 20, 30], vec![10, 20, 30]);
437     case(vec![10, 11, 20, 30], vec![10, 20, 30]);
438     case(vec![10, 20, 21, 30], vec![10, 20, 30]);
439     case(vec![10, 20, 30, 31], vec![10, 20, 30]);
440     case(vec![10, 11, 20, 21, 22, 30, 31], vec![10, 20, 30]);
441 }
442 
443 #[test]
test_dedup_by()444 fn test_dedup_by() {
445     let mut vec = vec!["foo", "bar", "Bar", "baz", "bar"];
446     vec.dedup_by(|a, b| a.eq_ignore_ascii_case(b));
447 
448     assert_eq!(vec, ["foo", "bar", "baz", "bar"]);
449 
450     let mut vec = vec![("foo", 1), ("foo", 2), ("bar", 3), ("bar", 4), ("bar", 5)];
451     vec.dedup_by(|a, b| {
452         a.0 == b.0 && {
453             b.1 += a.1;
454             true
455         }
456     });
457 
458     assert_eq!(vec, [("foo", 3), ("bar", 12)]);
459 }
460 
461 #[test]
test_dedup_unique()462 fn test_dedup_unique() {
463     let mut v0: Vec<Box<_>> = vec![Box::new(1), Box::new(1), Box::new(2), Box::new(3)];
464     v0.dedup();
465     let mut v1: Vec<Box<_>> = vec![Box::new(1), Box::new(2), Box::new(2), Box::new(3)];
466     v1.dedup();
467     let mut v2: Vec<Box<_>> = vec![Box::new(1), Box::new(2), Box::new(3), Box::new(3)];
468     v2.dedup();
469     // If the boxed pointers were leaked or otherwise misused, valgrind
470     // and/or rt should raise errors.
471 }
472 
473 #[test]
zero_sized_values()474 fn zero_sized_values() {
475     let mut v = Vec::new();
476     assert_eq!(v.len(), 0);
477     v.push(());
478     assert_eq!(v.len(), 1);
479     v.push(());
480     assert_eq!(v.len(), 2);
481     assert_eq!(v.pop(), Some(()));
482     assert_eq!(v.pop(), Some(()));
483     assert_eq!(v.pop(), None);
484 
485     assert_eq!(v.iter().count(), 0);
486     v.push(());
487     assert_eq!(v.iter().count(), 1);
488     v.push(());
489     assert_eq!(v.iter().count(), 2);
490 
491     for &() in &v {}
492 
493     assert_eq!(v.iter_mut().count(), 2);
494     v.push(());
495     assert_eq!(v.iter_mut().count(), 3);
496     v.push(());
497     assert_eq!(v.iter_mut().count(), 4);
498 
499     for &mut () in &mut v {}
500     unsafe {
501         v.set_len(0);
502     }
503     assert_eq!(v.iter_mut().count(), 0);
504 }
505 
506 #[test]
test_partition()507 fn test_partition() {
508     assert_eq!([].into_iter().partition(|x: &i32| *x < 3), (vec![], vec![]));
509     assert_eq!([1, 2, 3].into_iter().partition(|x| *x < 4), (vec![1, 2, 3], vec![]));
510     assert_eq!([1, 2, 3].into_iter().partition(|x| *x < 2), (vec![1], vec![2, 3]));
511     assert_eq!([1, 2, 3].into_iter().partition(|x| *x < 0), (vec![], vec![1, 2, 3]));
512 }
513 
514 #[test]
test_zip_unzip()515 fn test_zip_unzip() {
516     let z1 = vec![(1, 4), (2, 5), (3, 6)];
517 
518     let (left, right): (Vec<_>, Vec<_>) = z1.iter().cloned().unzip();
519 
520     assert_eq!((1, 4), (left[0], right[0]));
521     assert_eq!((2, 5), (left[1], right[1]));
522     assert_eq!((3, 6), (left[2], right[2]));
523 }
524 
525 #[test]
test_cmp()526 fn test_cmp() {
527     let x: &[isize] = &[1, 2, 3, 4, 5];
528     let cmp: &[isize] = &[1, 2, 3, 4, 5];
529     assert_eq!(&x[..], cmp);
530     let cmp: &[isize] = &[3, 4, 5];
531     assert_eq!(&x[2..], cmp);
532     let cmp: &[isize] = &[1, 2, 3];
533     assert_eq!(&x[..3], cmp);
534     let cmp: &[isize] = &[2, 3, 4];
535     assert_eq!(&x[1..4], cmp);
536 
537     let x: Vec<isize> = vec![1, 2, 3, 4, 5];
538     let cmp: &[isize] = &[1, 2, 3, 4, 5];
539     assert_eq!(&x[..], cmp);
540     let cmp: &[isize] = &[3, 4, 5];
541     assert_eq!(&x[2..], cmp);
542     let cmp: &[isize] = &[1, 2, 3];
543     assert_eq!(&x[..3], cmp);
544     let cmp: &[isize] = &[2, 3, 4];
545     assert_eq!(&x[1..4], cmp);
546 }
547 
548 #[test]
test_vec_truncate_drop()549 fn test_vec_truncate_drop() {
550     static mut DROPS: u32 = 0;
551     struct Elem(i32);
552     impl Drop for Elem {
553         fn drop(&mut self) {
554             unsafe {
555                 DROPS += 1;
556             }
557         }
558     }
559 
560     let mut v = vec![Elem(1), Elem(2), Elem(3), Elem(4), Elem(5)];
561     assert_eq!(unsafe { DROPS }, 0);
562     v.truncate(3);
563     assert_eq!(unsafe { DROPS }, 2);
564     v.truncate(0);
565     assert_eq!(unsafe { DROPS }, 5);
566 }
567 
568 #[test]
569 #[should_panic]
test_vec_truncate_fail()570 fn test_vec_truncate_fail() {
571     struct BadElem(i32);
572     impl Drop for BadElem {
573         fn drop(&mut self) {
574             let BadElem(ref mut x) = *self;
575             if *x == 0xbadbeef {
576                 panic!("BadElem panic: 0xbadbeef")
577             }
578         }
579     }
580 
581     let mut v = vec![BadElem(1), BadElem(2), BadElem(0xbadbeef), BadElem(4)];
582     v.truncate(0);
583 }
584 
585 #[test]
test_index()586 fn test_index() {
587     let vec = vec![1, 2, 3];
588     assert!(vec[1] == 2);
589 }
590 
591 #[test]
592 #[should_panic]
test_index_out_of_bounds()593 fn test_index_out_of_bounds() {
594     let vec = vec![1, 2, 3];
595     let _ = vec[3];
596 }
597 
598 #[test]
599 #[should_panic]
test_slice_out_of_bounds_1()600 fn test_slice_out_of_bounds_1() {
601     let x = vec![1, 2, 3, 4, 5];
602     let _ = &x[!0..];
603 }
604 
605 #[test]
606 #[should_panic]
test_slice_out_of_bounds_2()607 fn test_slice_out_of_bounds_2() {
608     let x = vec![1, 2, 3, 4, 5];
609     let _ = &x[..6];
610 }
611 
612 #[test]
613 #[should_panic]
test_slice_out_of_bounds_3()614 fn test_slice_out_of_bounds_3() {
615     let x = vec![1, 2, 3, 4, 5];
616     let _ = &x[!0..4];
617 }
618 
619 #[test]
620 #[should_panic]
test_slice_out_of_bounds_4()621 fn test_slice_out_of_bounds_4() {
622     let x = vec![1, 2, 3, 4, 5];
623     let _ = &x[1..6];
624 }
625 
626 #[test]
627 #[should_panic]
test_slice_out_of_bounds_5()628 fn test_slice_out_of_bounds_5() {
629     let x = vec![1, 2, 3, 4, 5];
630     let _ = &x[3..2];
631 }
632 
633 #[test]
634 #[should_panic]
test_swap_remove_empty()635 fn test_swap_remove_empty() {
636     let mut vec = Vec::<i32>::new();
637     vec.swap_remove(0);
638 }
639 
640 #[test]
test_move_items()641 fn test_move_items() {
642     let vec = vec![1, 2, 3];
643     let mut vec2 = vec![];
644     for i in vec {
645         vec2.push(i);
646     }
647     assert_eq!(vec2, [1, 2, 3]);
648 }
649 
650 #[test]
test_move_items_reverse()651 fn test_move_items_reverse() {
652     let vec = vec![1, 2, 3];
653     let mut vec2 = vec![];
654     for i in vec.into_iter().rev() {
655         vec2.push(i);
656     }
657     assert_eq!(vec2, [3, 2, 1]);
658 }
659 
660 #[test]
test_move_items_zero_sized()661 fn test_move_items_zero_sized() {
662     let vec = vec![(), (), ()];
663     let mut vec2 = vec![];
664     for i in vec {
665         vec2.push(i);
666     }
667     assert_eq!(vec2, [(), (), ()]);
668 }
669 
670 #[test]
test_drain_empty_vec()671 fn test_drain_empty_vec() {
672     let mut vec: Vec<i32> = vec![];
673     let mut vec2: Vec<i32> = vec![];
674     for i in vec.drain(..) {
675         vec2.push(i);
676     }
677     assert!(vec.is_empty());
678     assert!(vec2.is_empty());
679 }
680 
681 #[test]
test_drain_items()682 fn test_drain_items() {
683     let mut vec = vec![1, 2, 3];
684     let mut vec2 = vec![];
685     for i in vec.drain(..) {
686         vec2.push(i);
687     }
688     assert_eq!(vec, []);
689     assert_eq!(vec2, [1, 2, 3]);
690 }
691 
692 #[test]
test_drain_items_reverse()693 fn test_drain_items_reverse() {
694     let mut vec = vec![1, 2, 3];
695     let mut vec2 = vec![];
696     for i in vec.drain(..).rev() {
697         vec2.push(i);
698     }
699     assert_eq!(vec, []);
700     assert_eq!(vec2, [3, 2, 1]);
701 }
702 
703 #[test]
test_drain_items_zero_sized()704 fn test_drain_items_zero_sized() {
705     let mut vec = vec![(), (), ()];
706     let mut vec2 = vec![];
707     for i in vec.drain(..) {
708         vec2.push(i);
709     }
710     assert_eq!(vec, []);
711     assert_eq!(vec2, [(), (), ()]);
712 }
713 
714 #[test]
715 #[should_panic]
test_drain_out_of_bounds()716 fn test_drain_out_of_bounds() {
717     let mut v = vec![1, 2, 3, 4, 5];
718     v.drain(5..6);
719 }
720 
721 #[test]
test_drain_range()722 fn test_drain_range() {
723     let mut v = vec![1, 2, 3, 4, 5];
724     for _ in v.drain(4..) {}
725     assert_eq!(v, &[1, 2, 3, 4]);
726 
727     let mut v: Vec<_> = (1..6).map(|x| x.to_string()).collect();
728     for _ in v.drain(1..4) {}
729     assert_eq!(v, &[1.to_string(), 5.to_string()]);
730 
731     let mut v: Vec<_> = (1..6).map(|x| x.to_string()).collect();
732     for _ in v.drain(1..4).rev() {}
733     assert_eq!(v, &[1.to_string(), 5.to_string()]);
734 
735     let mut v: Vec<_> = vec![(); 5];
736     for _ in v.drain(1..4).rev() {}
737     assert_eq!(v, &[(), ()]);
738 }
739 
740 #[test]
test_drain_inclusive_range()741 fn test_drain_inclusive_range() {
742     let mut v = vec!['a', 'b', 'c', 'd', 'e'];
743     for _ in v.drain(1..=3) {}
744     assert_eq!(v, &['a', 'e']);
745 
746     let mut v: Vec<_> = (0..=5).map(|x| x.to_string()).collect();
747     for _ in v.drain(1..=5) {}
748     assert_eq!(v, &["0".to_string()]);
749 
750     let mut v: Vec<String> = (0..=5).map(|x| x.to_string()).collect();
751     for _ in v.drain(0..=5) {}
752     assert_eq!(v, Vec::<String>::new());
753 
754     let mut v: Vec<_> = (0..=5).map(|x| x.to_string()).collect();
755     for _ in v.drain(0..=3) {}
756     assert_eq!(v, &["4".to_string(), "5".to_string()]);
757 
758     let mut v: Vec<_> = (0..=1).map(|x| x.to_string()).collect();
759     for _ in v.drain(..=0) {}
760     assert_eq!(v, &["1".to_string()]);
761 }
762 
763 #[test]
test_drain_max_vec_size()764 fn test_drain_max_vec_size() {
765     let mut v = Vec::<()>::with_capacity(usize::MAX);
766     unsafe {
767         v.set_len(usize::MAX);
768     }
769     for _ in v.drain(usize::MAX - 1..) {}
770     assert_eq!(v.len(), usize::MAX - 1);
771 
772     let mut v = Vec::<()>::with_capacity(usize::MAX);
773     unsafe {
774         v.set_len(usize::MAX);
775     }
776     for _ in v.drain(usize::MAX - 1..=usize::MAX - 1) {}
777     assert_eq!(v.len(), usize::MAX - 1);
778 }
779 
780 #[test]
781 #[should_panic]
test_drain_index_overflow()782 fn test_drain_index_overflow() {
783     let mut v = Vec::<()>::with_capacity(usize::MAX);
784     unsafe {
785         v.set_len(usize::MAX);
786     }
787     v.drain(0..=usize::MAX);
788 }
789 
790 #[test]
791 #[should_panic]
test_drain_inclusive_out_of_bounds()792 fn test_drain_inclusive_out_of_bounds() {
793     let mut v = vec![1, 2, 3, 4, 5];
794     v.drain(5..=5);
795 }
796 
797 #[test]
798 #[should_panic]
test_drain_start_overflow()799 fn test_drain_start_overflow() {
800     let mut v = vec![1, 2, 3];
801     v.drain((Excluded(usize::MAX), Included(0)));
802 }
803 
804 #[test]
805 #[should_panic]
test_drain_end_overflow()806 fn test_drain_end_overflow() {
807     let mut v = vec![1, 2, 3];
808     v.drain((Included(0), Included(usize::MAX)));
809 }
810 
811 #[test]
812 #[cfg_attr(not(panic = "unwind"), ignore = "test requires unwinding support")]
test_drain_leak()813 fn test_drain_leak() {
814     static mut DROPS: i32 = 0;
815 
816     #[derive(Debug, PartialEq)]
817     struct D(u32, bool);
818 
819     impl Drop for D {
820         fn drop(&mut self) {
821             unsafe {
822                 DROPS += 1;
823             }
824 
825             if self.1 {
826                 panic!("panic in `drop`");
827             }
828         }
829     }
830 
831     let mut v = vec![
832         D(0, false),
833         D(1, false),
834         D(2, false),
835         D(3, false),
836         D(4, true),
837         D(5, false),
838         D(6, false),
839     ];
840 
841     catch_unwind(AssertUnwindSafe(|| {
842         v.drain(2..=5);
843     }))
844     .ok();
845 
846     assert_eq!(unsafe { DROPS }, 4);
847     assert_eq!(v, vec![D(0, false), D(1, false), D(6, false),]);
848 }
849 
850 #[test]
test_drain_keep_rest()851 fn test_drain_keep_rest() {
852     let mut v = vec![0, 1, 2, 3, 4, 5, 6];
853     let mut drain = v.drain(1..6);
854     assert_eq!(drain.next(), Some(1));
855     assert_eq!(drain.next_back(), Some(5));
856     assert_eq!(drain.next(), Some(2));
857 
858     drain.keep_rest();
859     assert_eq!(v, &[0, 3, 4, 6]);
860 }
861 
862 #[test]
test_drain_keep_rest_all()863 fn test_drain_keep_rest_all() {
864     let mut v = vec![0, 1, 2, 3, 4, 5, 6];
865     v.drain(1..6).keep_rest();
866     assert_eq!(v, &[0, 1, 2, 3, 4, 5, 6]);
867 }
868 
869 #[test]
test_drain_keep_rest_none()870 fn test_drain_keep_rest_none() {
871     let mut v = vec![0, 1, 2, 3, 4, 5, 6];
872     let mut drain = v.drain(1..6);
873 
874     drain.by_ref().for_each(drop);
875 
876     drain.keep_rest();
877     assert_eq!(v, &[0, 6]);
878 }
879 
880 #[test]
test_splice()881 fn test_splice() {
882     let mut v = vec![1, 2, 3, 4, 5];
883     let a = [10, 11, 12];
884     v.splice(2..4, a);
885     assert_eq!(v, &[1, 2, 10, 11, 12, 5]);
886     v.splice(1..3, Some(20));
887     assert_eq!(v, &[1, 20, 11, 12, 5]);
888 }
889 
890 #[test]
test_splice_inclusive_range()891 fn test_splice_inclusive_range() {
892     let mut v = vec![1, 2, 3, 4, 5];
893     let a = [10, 11, 12];
894     let t1: Vec<_> = v.splice(2..=3, a).collect();
895     assert_eq!(v, &[1, 2, 10, 11, 12, 5]);
896     assert_eq!(t1, &[3, 4]);
897     let t2: Vec<_> = v.splice(1..=2, Some(20)).collect();
898     assert_eq!(v, &[1, 20, 11, 12, 5]);
899     assert_eq!(t2, &[2, 10]);
900 }
901 
902 #[test]
903 #[should_panic]
test_splice_out_of_bounds()904 fn test_splice_out_of_bounds() {
905     let mut v = vec![1, 2, 3, 4, 5];
906     let a = [10, 11, 12];
907     v.splice(5..6, a);
908 }
909 
910 #[test]
911 #[should_panic]
test_splice_inclusive_out_of_bounds()912 fn test_splice_inclusive_out_of_bounds() {
913     let mut v = vec![1, 2, 3, 4, 5];
914     let a = [10, 11, 12];
915     v.splice(5..=5, a);
916 }
917 
918 #[test]
test_splice_items_zero_sized()919 fn test_splice_items_zero_sized() {
920     let mut vec = vec![(), (), ()];
921     let vec2 = vec![];
922     let t: Vec<_> = vec.splice(1..2, vec2.iter().cloned()).collect();
923     assert_eq!(vec, &[(), ()]);
924     assert_eq!(t, &[()]);
925 }
926 
927 #[test]
test_splice_unbounded()928 fn test_splice_unbounded() {
929     let mut vec = vec![1, 2, 3, 4, 5];
930     let t: Vec<_> = vec.splice(.., None).collect();
931     assert_eq!(vec, &[]);
932     assert_eq!(t, &[1, 2, 3, 4, 5]);
933 }
934 
935 #[test]
test_splice_forget()936 fn test_splice_forget() {
937     let mut v = vec![1, 2, 3, 4, 5];
938     let a = [10, 11, 12];
939     std::mem::forget(v.splice(2..4, a));
940     assert_eq!(v, &[1, 2]);
941 }
942 
943 #[test]
test_into_boxed_slice()944 fn test_into_boxed_slice() {
945     let xs = vec![1, 2, 3];
946     let ys = xs.into_boxed_slice();
947     assert_eq!(&*ys, [1, 2, 3]);
948 }
949 
950 #[test]
test_append()951 fn test_append() {
952     let mut vec = vec![1, 2, 3];
953     let mut vec2 = vec![4, 5, 6];
954     vec.append(&mut vec2);
955     assert_eq!(vec, [1, 2, 3, 4, 5, 6]);
956     assert_eq!(vec2, []);
957 }
958 
959 #[test]
test_split_off()960 fn test_split_off() {
961     let mut vec = vec![1, 2, 3, 4, 5, 6];
962     let orig_capacity = vec.capacity();
963     let vec2 = vec.split_off(4);
964     assert_eq!(vec, [1, 2, 3, 4]);
965     assert_eq!(vec2, [5, 6]);
966     assert_eq!(vec.capacity(), orig_capacity);
967 }
968 
969 #[test]
test_split_off_take_all()970 fn test_split_off_take_all() {
971     let mut vec = vec![1, 2, 3, 4, 5, 6];
972     let orig_ptr = vec.as_ptr();
973     let orig_capacity = vec.capacity();
974     let vec2 = vec.split_off(0);
975     assert_eq!(vec, []);
976     assert_eq!(vec2, [1, 2, 3, 4, 5, 6]);
977     assert_eq!(vec.capacity(), orig_capacity);
978     assert_eq!(vec2.as_ptr(), orig_ptr);
979 }
980 
981 #[test]
test_into_iter_as_slice()982 fn test_into_iter_as_slice() {
983     let vec = vec!['a', 'b', 'c'];
984     let mut into_iter = vec.into_iter();
985     assert_eq!(into_iter.as_slice(), &['a', 'b', 'c']);
986     let _ = into_iter.next().unwrap();
987     assert_eq!(into_iter.as_slice(), &['b', 'c']);
988     let _ = into_iter.next().unwrap();
989     let _ = into_iter.next().unwrap();
990     assert_eq!(into_iter.as_slice(), &[]);
991 }
992 
993 #[test]
test_into_iter_as_mut_slice()994 fn test_into_iter_as_mut_slice() {
995     let vec = vec!['a', 'b', 'c'];
996     let mut into_iter = vec.into_iter();
997     assert_eq!(into_iter.as_slice(), &['a', 'b', 'c']);
998     into_iter.as_mut_slice()[0] = 'x';
999     into_iter.as_mut_slice()[1] = 'y';
1000     assert_eq!(into_iter.next().unwrap(), 'x');
1001     assert_eq!(into_iter.as_slice(), &['y', 'c']);
1002 }
1003 
1004 #[test]
test_into_iter_debug()1005 fn test_into_iter_debug() {
1006     let vec = vec!['a', 'b', 'c'];
1007     let into_iter = vec.into_iter();
1008     let debug = format!("{into_iter:?}");
1009     assert_eq!(debug, "IntoIter(['a', 'b', 'c'])");
1010 }
1011 
1012 #[test]
test_into_iter_count()1013 fn test_into_iter_count() {
1014     assert_eq!([1, 2, 3].into_iter().count(), 3);
1015 }
1016 
1017 #[test]
test_into_iter_next_chunk()1018 fn test_into_iter_next_chunk() {
1019     let mut iter = b"lorem".to_vec().into_iter();
1020 
1021     assert_eq!(iter.next_chunk().unwrap(), [b'l', b'o']); // N is inferred as 2
1022     assert_eq!(iter.next_chunk().unwrap(), [b'r', b'e', b'm']); // N is inferred as 3
1023     assert_eq!(iter.next_chunk::<4>().unwrap_err().as_slice(), &[]); // N is explicitly 4
1024 }
1025 
1026 #[test]
test_into_iter_clone()1027 fn test_into_iter_clone() {
1028     fn iter_equal<I: Iterator<Item = i32>>(it: I, slice: &[i32]) {
1029         let v: Vec<i32> = it.collect();
1030         assert_eq!(&v[..], slice);
1031     }
1032     let mut it = [1, 2, 3].into_iter();
1033     iter_equal(it.clone(), &[1, 2, 3]);
1034     assert_eq!(it.next(), Some(1));
1035     let mut it = it.rev();
1036     iter_equal(it.clone(), &[3, 2]);
1037     assert_eq!(it.next(), Some(3));
1038     iter_equal(it.clone(), &[2]);
1039     assert_eq!(it.next(), Some(2));
1040     iter_equal(it.clone(), &[]);
1041     assert_eq!(it.next(), None);
1042 }
1043 
1044 #[test]
1045 #[cfg_attr(not(panic = "unwind"), ignore = "test requires unwinding support")]
test_into_iter_leak()1046 fn test_into_iter_leak() {
1047     static mut DROPS: i32 = 0;
1048 
1049     struct D(bool);
1050 
1051     impl Drop for D {
1052         fn drop(&mut self) {
1053             unsafe {
1054                 DROPS += 1;
1055             }
1056 
1057             if self.0 {
1058                 panic!("panic in `drop`");
1059             }
1060         }
1061     }
1062 
1063     let v = vec![D(false), D(true), D(false)];
1064 
1065     catch_unwind(move || drop(v.into_iter())).ok();
1066 
1067     assert_eq!(unsafe { DROPS }, 3);
1068 }
1069 
1070 #[test]
test_into_iter_advance_by()1071 fn test_into_iter_advance_by() {
1072     let mut i = vec![1, 2, 3, 4, 5].into_iter();
1073     assert_eq!(i.advance_by(0), Ok(()));
1074     assert_eq!(i.advance_back_by(0), Ok(()));
1075     assert_eq!(i.as_slice(), [1, 2, 3, 4, 5]);
1076 
1077     assert_eq!(i.advance_by(1), Ok(()));
1078     assert_eq!(i.advance_back_by(1), Ok(()));
1079     assert_eq!(i.as_slice(), [2, 3, 4]);
1080 
1081     assert_eq!(i.advance_back_by(usize::MAX), Err(NonZeroUsize::new(usize::MAX - 3).unwrap()));
1082 
1083     assert_eq!(i.advance_by(usize::MAX), Err(NonZeroUsize::new(usize::MAX).unwrap()));
1084 
1085     assert_eq!(i.advance_by(0), Ok(()));
1086     assert_eq!(i.advance_back_by(0), Ok(()));
1087 
1088     assert_eq!(i.len(), 0);
1089 }
1090 
1091 #[test]
test_into_iter_drop_allocator()1092 fn test_into_iter_drop_allocator() {
1093     struct ReferenceCountedAllocator<'a>(DropCounter<'a>);
1094 
1095     unsafe impl Allocator for ReferenceCountedAllocator<'_> {
1096         fn allocate(&self, layout: Layout) -> Result<NonNull<[u8]>, core::alloc::AllocError> {
1097             System.allocate(layout)
1098         }
1099 
1100         unsafe fn deallocate(&self, ptr: NonNull<u8>, layout: Layout) {
1101             // Safety: Invariants passed to caller.
1102             unsafe { System.deallocate(ptr, layout) }
1103         }
1104     }
1105 
1106     let mut drop_count = 0;
1107 
1108     let allocator = ReferenceCountedAllocator(DropCounter { count: &mut drop_count });
1109     let _ = Vec::<u32, _>::new_in(allocator);
1110     assert_eq!(drop_count, 1);
1111 
1112     let allocator = ReferenceCountedAllocator(DropCounter { count: &mut drop_count });
1113     let _ = Vec::<u32, _>::new_in(allocator).into_iter();
1114     assert_eq!(drop_count, 2);
1115 }
1116 
1117 #[test]
test_into_iter_zst()1118 fn test_into_iter_zst() {
1119     #[derive(Debug, Clone)]
1120     struct AlignedZstWithDrop([u64; 0]);
1121     impl Drop for AlignedZstWithDrop {
1122         fn drop(&mut self) {
1123             let addr = self as *mut _ as usize;
1124             assert!(hint::black_box(addr) % mem::align_of::<u64>() == 0);
1125         }
1126     }
1127 
1128     const C: AlignedZstWithDrop = AlignedZstWithDrop([0u64; 0]);
1129 
1130     for _ in vec![C].into_iter() {}
1131     for _ in vec![C; 5].into_iter().rev() {}
1132 
1133     let mut it = vec![C, C].into_iter();
1134     assert_eq!(it.advance_by(1), Ok(()));
1135     drop(it);
1136 
1137     let mut it = vec![C, C].into_iter();
1138     it.next_chunk::<1>().unwrap();
1139     drop(it);
1140 
1141     let mut it = vec![C, C].into_iter();
1142     it.next_chunk::<4>().unwrap_err();
1143     drop(it);
1144 }
1145 
1146 #[test]
test_from_iter_specialization()1147 fn test_from_iter_specialization() {
1148     let src: Vec<usize> = vec![0usize; 1];
1149     let srcptr = src.as_ptr();
1150     let sink = src.into_iter().collect::<Vec<_>>();
1151     let sinkptr = sink.as_ptr();
1152     assert_eq!(srcptr, sinkptr);
1153 }
1154 
1155 #[test]
test_from_iter_partially_drained_in_place_specialization()1156 fn test_from_iter_partially_drained_in_place_specialization() {
1157     let src: Vec<usize> = vec![0usize; 10];
1158     let srcptr = src.as_ptr();
1159     let mut iter = src.into_iter();
1160     iter.next();
1161     iter.next();
1162     let sink = iter.collect::<Vec<_>>();
1163     let sinkptr = sink.as_ptr();
1164     assert_eq!(srcptr, sinkptr);
1165 }
1166 
1167 #[test]
test_from_iter_specialization_with_iterator_adapters()1168 fn test_from_iter_specialization_with_iterator_adapters() {
1169     fn assert_in_place_trait<T: InPlaceIterable>(_: &T) {}
1170     let src: Vec<usize> = vec![0usize; 256];
1171     let srcptr = src.as_ptr();
1172     let iter = src
1173         .into_iter()
1174         .enumerate()
1175         .map(|i| i.0 + i.1)
1176         .zip(std::iter::repeat(1usize))
1177         .map(|(a, b)| a + b)
1178         .map_while(Option::Some)
1179         .skip(1)
1180         .map(|e| if e != usize::MAX { Ok(std::num::NonZeroUsize::new(e)) } else { Err(()) });
1181     assert_in_place_trait(&iter);
1182     let sink = iter.collect::<Result<Vec<_>, _>>().unwrap();
1183     let sinkptr = sink.as_ptr();
1184     assert_eq!(srcptr, sinkptr as *const usize);
1185 }
1186 
1187 #[test]
test_from_iter_specialization_head_tail_drop()1188 fn test_from_iter_specialization_head_tail_drop() {
1189     let drop_count: Vec<_> = (0..=2).map(|_| Rc::new(())).collect();
1190     let src: Vec<_> = drop_count.iter().cloned().collect();
1191     let srcptr = src.as_ptr();
1192     let iter = src.into_iter();
1193     let sink: Vec<_> = iter.skip(1).take(1).collect();
1194     let sinkptr = sink.as_ptr();
1195     assert_eq!(srcptr, sinkptr, "specialization was applied");
1196     assert_eq!(Rc::strong_count(&drop_count[0]), 1, "front was dropped");
1197     assert_eq!(Rc::strong_count(&drop_count[1]), 2, "one element was collected");
1198     assert_eq!(Rc::strong_count(&drop_count[2]), 1, "tail was dropped");
1199     assert_eq!(sink.len(), 1);
1200 }
1201 
1202 #[test]
1203 #[cfg_attr(not(panic = "unwind"), ignore = "test requires unwinding support")]
test_from_iter_specialization_panic_during_iteration_drops()1204 fn test_from_iter_specialization_panic_during_iteration_drops() {
1205     let drop_count: Vec<_> = (0..=2).map(|_| Rc::new(())).collect();
1206     let src: Vec<_> = drop_count.iter().cloned().collect();
1207     let iter = src.into_iter();
1208 
1209     let _ = std::panic::catch_unwind(AssertUnwindSafe(|| {
1210         let _ = iter
1211             .enumerate()
1212             .filter_map(|(i, e)| {
1213                 if i == 1 {
1214                     std::panic!("aborting iteration");
1215                 }
1216                 Some(e)
1217             })
1218             .collect::<Vec<_>>();
1219     }));
1220 
1221     assert!(
1222         drop_count.iter().map(Rc::strong_count).all(|count| count == 1),
1223         "all items were dropped once"
1224     );
1225 }
1226 
1227 #[test]
1228 #[cfg_attr(not(panic = "unwind"), ignore = "test requires unwinding support")]
test_from_iter_specialization_panic_during_drop_doesnt_leak()1229 fn test_from_iter_specialization_panic_during_drop_doesnt_leak() {
1230     static mut DROP_COUNTER_OLD: [usize; 5] = [0; 5];
1231     static mut DROP_COUNTER_NEW: [usize; 2] = [0; 2];
1232 
1233     #[derive(Debug)]
1234     struct Old(usize);
1235 
1236     impl Drop for Old {
1237         fn drop(&mut self) {
1238             unsafe {
1239                 DROP_COUNTER_OLD[self.0] += 1;
1240             }
1241 
1242             if self.0 == 3 {
1243                 panic!();
1244             }
1245 
1246             println!("Dropped Old: {}", self.0);
1247         }
1248     }
1249 
1250     #[derive(Debug)]
1251     struct New(usize);
1252 
1253     impl Drop for New {
1254         fn drop(&mut self) {
1255             unsafe {
1256                 DROP_COUNTER_NEW[self.0] += 1;
1257             }
1258 
1259             println!("Dropped New: {}", self.0);
1260         }
1261     }
1262 
1263     let _ = std::panic::catch_unwind(AssertUnwindSafe(|| {
1264         let v = vec![Old(0), Old(1), Old(2), Old(3), Old(4)];
1265         let _ = v.into_iter().map(|x| New(x.0)).take(2).collect::<Vec<_>>();
1266     }));
1267 
1268     assert_eq!(unsafe { DROP_COUNTER_OLD[0] }, 1);
1269     assert_eq!(unsafe { DROP_COUNTER_OLD[1] }, 1);
1270     assert_eq!(unsafe { DROP_COUNTER_OLD[2] }, 1);
1271     assert_eq!(unsafe { DROP_COUNTER_OLD[3] }, 1);
1272     assert_eq!(unsafe { DROP_COUNTER_OLD[4] }, 1);
1273 
1274     assert_eq!(unsafe { DROP_COUNTER_NEW[0] }, 1);
1275     assert_eq!(unsafe { DROP_COUNTER_NEW[1] }, 1);
1276 }
1277 
1278 // regression test for issue #85322. Peekable previously implemented InPlaceIterable,
1279 // but due to an interaction with IntoIter's current Clone implementation it failed to uphold
1280 // the contract.
1281 #[test]
test_collect_after_iterator_clone()1282 fn test_collect_after_iterator_clone() {
1283     let v = vec![0; 5];
1284     let mut i = v.into_iter().map(|i| i + 1).peekable();
1285     i.peek();
1286     let v = i.clone().collect::<Vec<_>>();
1287     assert_eq!(v, [1, 1, 1, 1, 1]);
1288     assert!(v.len() <= v.capacity());
1289 }
1290 #[test]
test_cow_from()1291 fn test_cow_from() {
1292     let borrowed: &[_] = &["borrowed", "(slice)"];
1293     let owned = vec!["owned", "(vec)"];
1294     match (Cow::from(owned.clone()), Cow::from(borrowed)) {
1295         (Cow::Owned(o), Cow::Borrowed(b)) => assert!(o == owned && b == borrowed),
1296         _ => panic!("invalid `Cow::from`"),
1297     }
1298 }
1299 
1300 #[test]
test_from_cow()1301 fn test_from_cow() {
1302     let borrowed: &[_] = &["borrowed", "(slice)"];
1303     let owned = vec!["owned", "(vec)"];
1304     assert_eq!(Vec::from(Cow::Borrowed(borrowed)), vec!["borrowed", "(slice)"]);
1305     assert_eq!(Vec::from(Cow::Owned(owned)), vec!["owned", "(vec)"]);
1306 }
1307 
1308 #[allow(dead_code)]
assert_covariance()1309 fn assert_covariance() {
1310     fn drain<'new>(d: Drain<'static, &'static str>) -> Drain<'new, &'new str> {
1311         d
1312     }
1313     fn into_iter<'new>(i: IntoIter<&'static str>) -> IntoIter<&'new str> {
1314         i
1315     }
1316 }
1317 
1318 #[test]
from_into_inner()1319 fn from_into_inner() {
1320     let vec = vec![1, 2, 3];
1321     let ptr = vec.as_ptr();
1322     let vec = vec.into_iter().collect::<Vec<_>>();
1323     assert_eq!(vec, [1, 2, 3]);
1324     assert_eq!(vec.as_ptr(), ptr);
1325 
1326     let ptr = &vec[1] as *const _;
1327     let mut it = vec.into_iter();
1328     it.next().unwrap();
1329     let vec = it.collect::<Vec<_>>();
1330     assert_eq!(vec, [2, 3]);
1331     assert!(ptr != vec.as_ptr());
1332 }
1333 
1334 #[test]
overaligned_allocations()1335 fn overaligned_allocations() {
1336     #[repr(align(256))]
1337     struct Foo(usize);
1338     let mut v = vec![Foo(273)];
1339     for i in 0..0x1000 {
1340         v.reserve_exact(i);
1341         assert!(v[0].0 == 273);
1342         assert!(v.as_ptr() as usize & 0xff == 0);
1343         v.shrink_to_fit();
1344         assert!(v[0].0 == 273);
1345         assert!(v.as_ptr() as usize & 0xff == 0);
1346     }
1347 }
1348 
1349 #[test]
extract_if_empty()1350 fn extract_if_empty() {
1351     let mut vec: Vec<i32> = vec![];
1352 
1353     {
1354         let mut iter = vec.extract_if(|_| true);
1355         assert_eq!(iter.size_hint(), (0, Some(0)));
1356         assert_eq!(iter.next(), None);
1357         assert_eq!(iter.size_hint(), (0, Some(0)));
1358         assert_eq!(iter.next(), None);
1359         assert_eq!(iter.size_hint(), (0, Some(0)));
1360     }
1361     assert_eq!(vec.len(), 0);
1362     assert_eq!(vec, vec![]);
1363 }
1364 
1365 #[test]
extract_if_zst()1366 fn extract_if_zst() {
1367     let mut vec = vec![(), (), (), (), ()];
1368     let initial_len = vec.len();
1369     let mut count = 0;
1370     {
1371         let mut iter = vec.extract_if(|_| true);
1372         assert_eq!(iter.size_hint(), (0, Some(initial_len)));
1373         while let Some(_) = iter.next() {
1374             count += 1;
1375             assert_eq!(iter.size_hint(), (0, Some(initial_len - count)));
1376         }
1377         assert_eq!(iter.size_hint(), (0, Some(0)));
1378         assert_eq!(iter.next(), None);
1379         assert_eq!(iter.size_hint(), (0, Some(0)));
1380     }
1381 
1382     assert_eq!(count, initial_len);
1383     assert_eq!(vec.len(), 0);
1384     assert_eq!(vec, vec![]);
1385 }
1386 
1387 #[test]
extract_if_false()1388 fn extract_if_false() {
1389     let mut vec = vec![1, 2, 3, 4, 5, 6, 7, 8, 9, 10];
1390 
1391     let initial_len = vec.len();
1392     let mut count = 0;
1393     {
1394         let mut iter = vec.extract_if(|_| false);
1395         assert_eq!(iter.size_hint(), (0, Some(initial_len)));
1396         for _ in iter.by_ref() {
1397             count += 1;
1398         }
1399         assert_eq!(iter.size_hint(), (0, Some(0)));
1400         assert_eq!(iter.next(), None);
1401         assert_eq!(iter.size_hint(), (0, Some(0)));
1402     }
1403 
1404     assert_eq!(count, 0);
1405     assert_eq!(vec.len(), initial_len);
1406     assert_eq!(vec, vec![1, 2, 3, 4, 5, 6, 7, 8, 9, 10]);
1407 }
1408 
1409 #[test]
extract_if_true()1410 fn extract_if_true() {
1411     let mut vec = vec![1, 2, 3, 4, 5, 6, 7, 8, 9, 10];
1412 
1413     let initial_len = vec.len();
1414     let mut count = 0;
1415     {
1416         let mut iter = vec.extract_if(|_| true);
1417         assert_eq!(iter.size_hint(), (0, Some(initial_len)));
1418         while let Some(_) = iter.next() {
1419             count += 1;
1420             assert_eq!(iter.size_hint(), (0, Some(initial_len - count)));
1421         }
1422         assert_eq!(iter.size_hint(), (0, Some(0)));
1423         assert_eq!(iter.next(), None);
1424         assert_eq!(iter.size_hint(), (0, Some(0)));
1425     }
1426 
1427     assert_eq!(count, initial_len);
1428     assert_eq!(vec.len(), 0);
1429     assert_eq!(vec, vec![]);
1430 }
1431 
1432 #[test]
extract_if_complex()1433 fn extract_if_complex() {
1434     {
1435         //                [+xxx++++++xxxxx++++x+x++]
1436         let mut vec = vec![
1437             1, 2, 4, 6, 7, 9, 11, 13, 15, 17, 18, 20, 22, 24, 26, 27, 29, 31, 33, 34, 35, 36, 37,
1438             39,
1439         ];
1440 
1441         let removed = vec.extract_if(|x| *x % 2 == 0).collect::<Vec<_>>();
1442         assert_eq!(removed.len(), 10);
1443         assert_eq!(removed, vec![2, 4, 6, 18, 20, 22, 24, 26, 34, 36]);
1444 
1445         assert_eq!(vec.len(), 14);
1446         assert_eq!(vec, vec![1, 7, 9, 11, 13, 15, 17, 27, 29, 31, 33, 35, 37, 39]);
1447     }
1448 
1449     {
1450         //                [xxx++++++xxxxx++++x+x++]
1451         let mut vec = vec![
1452             2, 4, 6, 7, 9, 11, 13, 15, 17, 18, 20, 22, 24, 26, 27, 29, 31, 33, 34, 35, 36, 37, 39,
1453         ];
1454 
1455         let removed = vec.extract_if(|x| *x % 2 == 0).collect::<Vec<_>>();
1456         assert_eq!(removed.len(), 10);
1457         assert_eq!(removed, vec![2, 4, 6, 18, 20, 22, 24, 26, 34, 36]);
1458 
1459         assert_eq!(vec.len(), 13);
1460         assert_eq!(vec, vec![7, 9, 11, 13, 15, 17, 27, 29, 31, 33, 35, 37, 39]);
1461     }
1462 
1463     {
1464         //                [xxx++++++xxxxx++++x+x]
1465         let mut vec =
1466             vec![2, 4, 6, 7, 9, 11, 13, 15, 17, 18, 20, 22, 24, 26, 27, 29, 31, 33, 34, 35, 36];
1467 
1468         let removed = vec.extract_if(|x| *x % 2 == 0).collect::<Vec<_>>();
1469         assert_eq!(removed.len(), 10);
1470         assert_eq!(removed, vec![2, 4, 6, 18, 20, 22, 24, 26, 34, 36]);
1471 
1472         assert_eq!(vec.len(), 11);
1473         assert_eq!(vec, vec![7, 9, 11, 13, 15, 17, 27, 29, 31, 33, 35]);
1474     }
1475 
1476     {
1477         //                [xxxxxxxxxx+++++++++++]
1478         let mut vec = vec![2, 4, 6, 8, 10, 12, 14, 16, 18, 20, 1, 3, 5, 7, 9, 11, 13, 15, 17, 19];
1479 
1480         let removed = vec.extract_if(|x| *x % 2 == 0).collect::<Vec<_>>();
1481         assert_eq!(removed.len(), 10);
1482         assert_eq!(removed, vec![2, 4, 6, 8, 10, 12, 14, 16, 18, 20]);
1483 
1484         assert_eq!(vec.len(), 10);
1485         assert_eq!(vec, vec![1, 3, 5, 7, 9, 11, 13, 15, 17, 19]);
1486     }
1487 
1488     {
1489         //                [+++++++++++xxxxxxxxxx]
1490         let mut vec = vec![1, 3, 5, 7, 9, 11, 13, 15, 17, 19, 2, 4, 6, 8, 10, 12, 14, 16, 18, 20];
1491 
1492         let removed = vec.extract_if(|x| *x % 2 == 0).collect::<Vec<_>>();
1493         assert_eq!(removed.len(), 10);
1494         assert_eq!(removed, vec![2, 4, 6, 8, 10, 12, 14, 16, 18, 20]);
1495 
1496         assert_eq!(vec.len(), 10);
1497         assert_eq!(vec, vec![1, 3, 5, 7, 9, 11, 13, 15, 17, 19]);
1498     }
1499 }
1500 
1501 // FIXME: re-enable emscripten once it can unwind again
1502 #[test]
1503 #[cfg(not(target_os = "emscripten"))]
1504 #[cfg_attr(not(panic = "unwind"), ignore = "test requires unwinding support")]
extract_if_consumed_panic()1505 fn extract_if_consumed_panic() {
1506     use std::rc::Rc;
1507     use std::sync::Mutex;
1508 
1509     struct Check {
1510         index: usize,
1511         drop_counts: Rc<Mutex<Vec<usize>>>,
1512     }
1513 
1514     impl Drop for Check {
1515         fn drop(&mut self) {
1516             self.drop_counts.lock().unwrap()[self.index] += 1;
1517             println!("drop: {}", self.index);
1518         }
1519     }
1520 
1521     let check_count = 10;
1522     let drop_counts = Rc::new(Mutex::new(vec![0_usize; check_count]));
1523     let mut data: Vec<Check> = (0..check_count)
1524         .map(|index| Check { index, drop_counts: Rc::clone(&drop_counts) })
1525         .collect();
1526 
1527     let _ = std::panic::catch_unwind(move || {
1528         let filter = |c: &mut Check| {
1529             if c.index == 2 {
1530                 panic!("panic at index: {}", c.index);
1531             }
1532             // Verify that if the filter could panic again on another element
1533             // that it would not cause a double panic and all elements of the
1534             // vec would still be dropped exactly once.
1535             if c.index == 4 {
1536                 panic!("panic at index: {}", c.index);
1537             }
1538             c.index < 6
1539         };
1540         let drain = data.extract_if(filter);
1541 
1542         // NOTE: The ExtractIf is explicitly consumed
1543         drain.for_each(drop);
1544     });
1545 
1546     let drop_counts = drop_counts.lock().unwrap();
1547     assert_eq!(check_count, drop_counts.len());
1548 
1549     for (index, count) in drop_counts.iter().cloned().enumerate() {
1550         assert_eq!(1, count, "unexpected drop count at index: {} (count: {})", index, count);
1551     }
1552 }
1553 
1554 // FIXME: Re-enable emscripten once it can catch panics
1555 #[test]
1556 #[cfg(not(target_os = "emscripten"))]
1557 #[cfg_attr(not(panic = "unwind"), ignore = "test requires unwinding support")]
extract_if_unconsumed_panic()1558 fn extract_if_unconsumed_panic() {
1559     use std::rc::Rc;
1560     use std::sync::Mutex;
1561 
1562     struct Check {
1563         index: usize,
1564         drop_counts: Rc<Mutex<Vec<usize>>>,
1565     }
1566 
1567     impl Drop for Check {
1568         fn drop(&mut self) {
1569             self.drop_counts.lock().unwrap()[self.index] += 1;
1570             println!("drop: {}", self.index);
1571         }
1572     }
1573 
1574     let check_count = 10;
1575     let drop_counts = Rc::new(Mutex::new(vec![0_usize; check_count]));
1576     let mut data: Vec<Check> = (0..check_count)
1577         .map(|index| Check { index, drop_counts: Rc::clone(&drop_counts) })
1578         .collect();
1579 
1580     let _ = std::panic::catch_unwind(move || {
1581         let filter = |c: &mut Check| {
1582             if c.index == 2 {
1583                 panic!("panic at index: {}", c.index);
1584             }
1585             // Verify that if the filter could panic again on another element
1586             // that it would not cause a double panic and all elements of the
1587             // vec would still be dropped exactly once.
1588             if c.index == 4 {
1589                 panic!("panic at index: {}", c.index);
1590             }
1591             c.index < 6
1592         };
1593         let _drain = data.extract_if(filter);
1594 
1595         // NOTE: The ExtractIf is dropped without being consumed
1596     });
1597 
1598     let drop_counts = drop_counts.lock().unwrap();
1599     assert_eq!(check_count, drop_counts.len());
1600 
1601     for (index, count) in drop_counts.iter().cloned().enumerate() {
1602         assert_eq!(1, count, "unexpected drop count at index: {} (count: {})", index, count);
1603     }
1604 }
1605 
1606 #[test]
extract_if_unconsumed()1607 fn extract_if_unconsumed() {
1608     let mut vec = vec![1, 2, 3, 4];
1609     let drain = vec.extract_if(|&mut x| x % 2 != 0);
1610     drop(drain);
1611     assert_eq!(vec, [1, 2, 3, 4]);
1612 }
1613 
1614 #[test]
test_reserve_exact()1615 fn test_reserve_exact() {
1616     // This is all the same as test_reserve
1617 
1618     let mut v = Vec::new();
1619     assert_eq!(v.capacity(), 0);
1620 
1621     v.reserve_exact(2);
1622     assert!(v.capacity() >= 2);
1623 
1624     for i in 0..16 {
1625         v.push(i);
1626     }
1627 
1628     assert!(v.capacity() >= 16);
1629     v.reserve_exact(16);
1630     assert!(v.capacity() >= 32);
1631 
1632     v.push(16);
1633 
1634     v.reserve_exact(16);
1635     assert!(v.capacity() >= 33)
1636 }
1637 
1638 #[test]
1639 #[cfg_attr(miri, ignore)] // Miri does not support signalling OOM
1640 #[cfg_attr(target_os = "android", ignore)] // Android used in CI has a broken dlmalloc
test_try_reserve()1641 fn test_try_reserve() {
1642     // These are the interesting cases:
1643     // * exactly isize::MAX should never trigger a CapacityOverflow (can be OOM)
1644     // * > isize::MAX should always fail
1645     //    * On 16/32-bit should CapacityOverflow
1646     //    * On 64-bit should OOM
1647     // * overflow may trigger when adding `len` to `cap` (in number of elements)
1648     // * overflow may trigger when multiplying `new_cap` by size_of::<T> (to get bytes)
1649 
1650     const MAX_CAP: usize = isize::MAX as usize;
1651     const MAX_USIZE: usize = usize::MAX;
1652 
1653     {
1654         // Note: basic stuff is checked by test_reserve
1655         let mut empty_bytes: Vec<u8> = Vec::new();
1656 
1657         // Check isize::MAX doesn't count as an overflow
1658         if let Err(CapacityOverflow) = empty_bytes.try_reserve(MAX_CAP).map_err(|e| e.kind()) {
1659             panic!("isize::MAX shouldn't trigger an overflow!");
1660         }
1661         // Play it again, frank! (just to be sure)
1662         if let Err(CapacityOverflow) = empty_bytes.try_reserve(MAX_CAP).map_err(|e| e.kind()) {
1663             panic!("isize::MAX shouldn't trigger an overflow!");
1664         }
1665 
1666         // Check isize::MAX + 1 does count as overflow
1667         assert_matches!(
1668             empty_bytes.try_reserve(MAX_CAP + 1).map_err(|e| e.kind()),
1669             Err(CapacityOverflow),
1670             "isize::MAX + 1 should trigger an overflow!"
1671         );
1672 
1673         // Check usize::MAX does count as overflow
1674         assert_matches!(
1675             empty_bytes.try_reserve(MAX_USIZE).map_err(|e| e.kind()),
1676             Err(CapacityOverflow),
1677             "usize::MAX should trigger an overflow!"
1678         );
1679     }
1680 
1681     {
1682         // Same basic idea, but with non-zero len
1683         let mut ten_bytes: Vec<u8> = vec![1, 2, 3, 4, 5, 6, 7, 8, 9, 10];
1684 
1685         if let Err(CapacityOverflow) = ten_bytes.try_reserve(MAX_CAP - 10).map_err(|e| e.kind()) {
1686             panic!("isize::MAX shouldn't trigger an overflow!");
1687         }
1688         if let Err(CapacityOverflow) = ten_bytes.try_reserve(MAX_CAP - 10).map_err(|e| e.kind()) {
1689             panic!("isize::MAX shouldn't trigger an overflow!");
1690         }
1691 
1692         assert_matches!(
1693             ten_bytes.try_reserve(MAX_CAP - 9).map_err(|e| e.kind()),
1694             Err(CapacityOverflow),
1695             "isize::MAX + 1 should trigger an overflow!"
1696         );
1697 
1698         // Should always overflow in the add-to-len
1699         assert_matches!(
1700             ten_bytes.try_reserve(MAX_USIZE).map_err(|e| e.kind()),
1701             Err(CapacityOverflow),
1702             "usize::MAX should trigger an overflow!"
1703         );
1704     }
1705 
1706     {
1707         // Same basic idea, but with interesting type size
1708         let mut ten_u32s: Vec<u32> = vec![1, 2, 3, 4, 5, 6, 7, 8, 9, 10];
1709 
1710         if let Err(CapacityOverflow) = ten_u32s.try_reserve(MAX_CAP / 4 - 10).map_err(|e| e.kind())
1711         {
1712             panic!("isize::MAX shouldn't trigger an overflow!");
1713         }
1714         if let Err(CapacityOverflow) = ten_u32s.try_reserve(MAX_CAP / 4 - 10).map_err(|e| e.kind())
1715         {
1716             panic!("isize::MAX shouldn't trigger an overflow!");
1717         }
1718 
1719         assert_matches!(
1720             ten_u32s.try_reserve(MAX_CAP / 4 - 9).map_err(|e| e.kind()),
1721             Err(CapacityOverflow),
1722             "isize::MAX + 1 should trigger an overflow!"
1723         );
1724 
1725         // Should fail in the mul-by-size
1726         assert_matches!(
1727             ten_u32s.try_reserve(MAX_USIZE - 20).map_err(|e| e.kind()),
1728             Err(CapacityOverflow),
1729             "usize::MAX should trigger an overflow!"
1730         );
1731     }
1732 }
1733 
1734 #[test]
1735 #[cfg_attr(miri, ignore)] // Miri does not support signalling OOM
1736 #[cfg_attr(target_os = "android", ignore)] // Android used in CI has a broken dlmalloc
test_try_reserve_exact()1737 fn test_try_reserve_exact() {
1738     // This is exactly the same as test_try_reserve with the method changed.
1739     // See that test for comments.
1740 
1741     const MAX_CAP: usize = isize::MAX as usize;
1742     const MAX_USIZE: usize = usize::MAX;
1743 
1744     {
1745         let mut empty_bytes: Vec<u8> = Vec::new();
1746 
1747         if let Err(CapacityOverflow) = empty_bytes.try_reserve_exact(MAX_CAP).map_err(|e| e.kind())
1748         {
1749             panic!("isize::MAX shouldn't trigger an overflow!");
1750         }
1751         if let Err(CapacityOverflow) = empty_bytes.try_reserve_exact(MAX_CAP).map_err(|e| e.kind())
1752         {
1753             panic!("isize::MAX shouldn't trigger an overflow!");
1754         }
1755 
1756         assert_matches!(
1757             empty_bytes.try_reserve_exact(MAX_CAP + 1).map_err(|e| e.kind()),
1758             Err(CapacityOverflow),
1759             "isize::MAX + 1 should trigger an overflow!"
1760         );
1761 
1762         assert_matches!(
1763             empty_bytes.try_reserve_exact(MAX_USIZE).map_err(|e| e.kind()),
1764             Err(CapacityOverflow),
1765             "usize::MAX should trigger an overflow!"
1766         );
1767     }
1768 
1769     {
1770         let mut ten_bytes: Vec<u8> = vec![1, 2, 3, 4, 5, 6, 7, 8, 9, 10];
1771 
1772         if let Err(CapacityOverflow) =
1773             ten_bytes.try_reserve_exact(MAX_CAP - 10).map_err(|e| e.kind())
1774         {
1775             panic!("isize::MAX shouldn't trigger an overflow!");
1776         }
1777         if let Err(CapacityOverflow) =
1778             ten_bytes.try_reserve_exact(MAX_CAP - 10).map_err(|e| e.kind())
1779         {
1780             panic!("isize::MAX shouldn't trigger an overflow!");
1781         }
1782 
1783         assert_matches!(
1784             ten_bytes.try_reserve_exact(MAX_CAP - 9).map_err(|e| e.kind()),
1785             Err(CapacityOverflow),
1786             "isize::MAX + 1 should trigger an overflow!"
1787         );
1788 
1789         assert_matches!(
1790             ten_bytes.try_reserve_exact(MAX_USIZE).map_err(|e| e.kind()),
1791             Err(CapacityOverflow),
1792             "usize::MAX should trigger an overflow!"
1793         );
1794     }
1795 
1796     {
1797         let mut ten_u32s: Vec<u32> = vec![1, 2, 3, 4, 5, 6, 7, 8, 9, 10];
1798 
1799         if let Err(CapacityOverflow) =
1800             ten_u32s.try_reserve_exact(MAX_CAP / 4 - 10).map_err(|e| e.kind())
1801         {
1802             panic!("isize::MAX shouldn't trigger an overflow!");
1803         }
1804         if let Err(CapacityOverflow) =
1805             ten_u32s.try_reserve_exact(MAX_CAP / 4 - 10).map_err(|e| e.kind())
1806         {
1807             panic!("isize::MAX shouldn't trigger an overflow!");
1808         }
1809 
1810         assert_matches!(
1811             ten_u32s.try_reserve_exact(MAX_CAP / 4 - 9).map_err(|e| e.kind()),
1812             Err(CapacityOverflow),
1813             "isize::MAX + 1 should trigger an overflow!"
1814         );
1815 
1816         assert_matches!(
1817             ten_u32s.try_reserve_exact(MAX_USIZE - 20).map_err(|e| e.kind()),
1818             Err(CapacityOverflow),
1819             "usize::MAX should trigger an overflow!"
1820         );
1821     }
1822 }
1823 
1824 #[test]
test_stable_pointers()1825 fn test_stable_pointers() {
1826     /// Pull an element from the iterator, then drop it.
1827     /// Useful to cover both the `next` and `drop` paths of an iterator.
1828     fn next_then_drop<I: Iterator>(mut i: I) {
1829         i.next().unwrap();
1830         drop(i);
1831     }
1832 
1833     // Test that, if we reserved enough space, adding and removing elements does not
1834     // invalidate references into the vector (such as `v0`). This test also
1835     // runs in Miri, which would detect such problems.
1836     // Note that this test does *not* constitute a stable guarantee that all these functions do not
1837     // reallocate! Only what is explicitly documented at
1838     // <https://doc.rust-lang.org/nightly/std/vec/struct.Vec.html#guarantees> is stably guaranteed.
1839     let mut v = Vec::with_capacity(128);
1840     v.push(13);
1841 
1842     // Laundering the lifetime -- we take care that `v` does not reallocate, so that's okay.
1843     let v0 = &mut v[0];
1844     let v0 = unsafe { &mut *(v0 as *mut _) };
1845     // Now do a bunch of things and occasionally use `v0` again to assert it is still valid.
1846 
1847     // Pushing/inserting and popping/removing
1848     v.push(1);
1849     v.push(2);
1850     v.insert(1, 1);
1851     assert_eq!(*v0, 13);
1852     v.remove(1);
1853     v.pop().unwrap();
1854     assert_eq!(*v0, 13);
1855     v.push(1);
1856     v.swap_remove(1);
1857     assert_eq!(v.len(), 2);
1858     v.swap_remove(1); // swap_remove the last element
1859     assert_eq!(*v0, 13);
1860 
1861     // Appending
1862     v.append(&mut vec![27, 19]);
1863     assert_eq!(*v0, 13);
1864 
1865     // Extending
1866     v.extend_from_slice(&[1, 2]);
1867     v.extend(&[1, 2]); // `slice::Iter` (with `T: Copy`) specialization
1868     v.extend(vec![2, 3]); // `vec::IntoIter` specialization
1869     v.extend(std::iter::once(3)); // `TrustedLen` specialization
1870     v.extend(std::iter::empty::<i32>()); // `TrustedLen` specialization with empty iterator
1871     v.extend(std::iter::once(3).filter(|_| true)); // base case
1872     v.extend(std::iter::once(&3)); // `cloned` specialization
1873     assert_eq!(*v0, 13);
1874 
1875     // Truncation
1876     v.truncate(2);
1877     assert_eq!(*v0, 13);
1878 
1879     // Resizing
1880     v.resize_with(v.len() + 10, || 42);
1881     assert_eq!(*v0, 13);
1882     v.resize_with(2, || panic!());
1883     assert_eq!(*v0, 13);
1884 
1885     // No-op reservation
1886     v.reserve(32);
1887     v.reserve_exact(32);
1888     assert_eq!(*v0, 13);
1889 
1890     // Partial draining
1891     v.resize_with(10, || 42);
1892     next_then_drop(v.drain(5..));
1893     assert_eq!(*v0, 13);
1894 
1895     // Splicing
1896     v.resize_with(10, || 42);
1897     next_then_drop(v.splice(5.., vec![1, 2, 3, 4, 5])); // empty tail after range
1898     assert_eq!(*v0, 13);
1899     next_then_drop(v.splice(5..8, vec![1])); // replacement is smaller than original range
1900     assert_eq!(*v0, 13);
1901     next_then_drop(v.splice(5..6, [1; 10].into_iter().filter(|_| true))); // lower bound not exact
1902     assert_eq!(*v0, 13);
1903 
1904     // spare_capacity_mut
1905     v.spare_capacity_mut();
1906     assert_eq!(*v0, 13);
1907 
1908     // Smoke test that would fire even outside Miri if an actual relocation happened.
1909     // Also ensures the pointer is still writeable after all this.
1910     *v0 -= 13;
1911     assert_eq!(v[0], 0);
1912 }
1913 
1914 // https://github.com/rust-lang/rust/pull/49496 introduced specialization based on:
1915 //
1916 // ```
1917 // unsafe impl<T: ?Sized> IsZero for *mut T {
1918 //     fn is_zero(&self) -> bool {
1919 //         (*self).is_null()
1920 //     }
1921 // }
1922 // ```
1923 //
1924 // … to call `RawVec::with_capacity_zeroed` for creating `Vec<*mut T>`,
1925 // which is incorrect for fat pointers since `<*mut T>::is_null` only looks at the data component.
1926 // That is, a fat pointer can be “null” without being made entirely of zero bits.
1927 #[test]
vec_macro_repeating_null_raw_fat_pointer()1928 fn vec_macro_repeating_null_raw_fat_pointer() {
1929     let raw_dyn = &mut (|| ()) as &mut dyn Fn() as *mut dyn Fn();
1930     let vtable = dbg!(ptr_metadata(raw_dyn));
1931     let null_raw_dyn = ptr_from_raw_parts(std::ptr::null_mut(), vtable);
1932     assert!(null_raw_dyn.is_null());
1933 
1934     let vec = vec![null_raw_dyn; 1];
1935     dbg!(ptr_metadata(vec[0]));
1936     assert!(vec[0] == null_raw_dyn);
1937 
1938     // Polyfill for https://github.com/rust-lang/rfcs/pull/2580
1939 
1940     fn ptr_metadata(ptr: *mut dyn Fn()) -> *mut () {
1941         unsafe { std::mem::transmute::<*mut dyn Fn(), DynRepr>(ptr).vtable }
1942     }
1943 
1944     fn ptr_from_raw_parts(data: *mut (), vtable: *mut ()) -> *mut dyn Fn() {
1945         unsafe { std::mem::transmute::<DynRepr, *mut dyn Fn()>(DynRepr { data, vtable }) }
1946     }
1947 
1948     #[repr(C)]
1949     struct DynRepr {
1950         data: *mut (),
1951         vtable: *mut (),
1952     }
1953 }
1954 
1955 // This test will likely fail if you change the capacities used in
1956 // `RawVec::grow_amortized`.
1957 #[test]
test_push_growth_strategy()1958 fn test_push_growth_strategy() {
1959     // If the element size is 1, we jump from 0 to 8, then double.
1960     {
1961         let mut v1: Vec<u8> = vec![];
1962         assert_eq!(v1.capacity(), 0);
1963 
1964         for _ in 0..8 {
1965             v1.push(0);
1966             assert_eq!(v1.capacity(), 8);
1967         }
1968 
1969         for _ in 8..16 {
1970             v1.push(0);
1971             assert_eq!(v1.capacity(), 16);
1972         }
1973 
1974         for _ in 16..32 {
1975             v1.push(0);
1976             assert_eq!(v1.capacity(), 32);
1977         }
1978 
1979         for _ in 32..64 {
1980             v1.push(0);
1981             assert_eq!(v1.capacity(), 64);
1982         }
1983     }
1984 
1985     // If the element size is 2..=1024, we jump from 0 to 4, then double.
1986     {
1987         let mut v2: Vec<u16> = vec![];
1988         let mut v1024: Vec<[u8; 1024]> = vec![];
1989         assert_eq!(v2.capacity(), 0);
1990         assert_eq!(v1024.capacity(), 0);
1991 
1992         for _ in 0..4 {
1993             v2.push(0);
1994             v1024.push([0; 1024]);
1995             assert_eq!(v2.capacity(), 4);
1996             assert_eq!(v1024.capacity(), 4);
1997         }
1998 
1999         for _ in 4..8 {
2000             v2.push(0);
2001             v1024.push([0; 1024]);
2002             assert_eq!(v2.capacity(), 8);
2003             assert_eq!(v1024.capacity(), 8);
2004         }
2005 
2006         for _ in 8..16 {
2007             v2.push(0);
2008             v1024.push([0; 1024]);
2009             assert_eq!(v2.capacity(), 16);
2010             assert_eq!(v1024.capacity(), 16);
2011         }
2012 
2013         for _ in 16..32 {
2014             v2.push(0);
2015             v1024.push([0; 1024]);
2016             assert_eq!(v2.capacity(), 32);
2017             assert_eq!(v1024.capacity(), 32);
2018         }
2019 
2020         for _ in 32..64 {
2021             v2.push(0);
2022             v1024.push([0; 1024]);
2023             assert_eq!(v2.capacity(), 64);
2024             assert_eq!(v1024.capacity(), 64);
2025         }
2026     }
2027 
2028     // If the element size is > 1024, we jump from 0 to 1, then double.
2029     {
2030         let mut v1025: Vec<[u8; 1025]> = vec![];
2031         assert_eq!(v1025.capacity(), 0);
2032 
2033         for _ in 0..1 {
2034             v1025.push([0; 1025]);
2035             assert_eq!(v1025.capacity(), 1);
2036         }
2037 
2038         for _ in 1..2 {
2039             v1025.push([0; 1025]);
2040             assert_eq!(v1025.capacity(), 2);
2041         }
2042 
2043         for _ in 2..4 {
2044             v1025.push([0; 1025]);
2045             assert_eq!(v1025.capacity(), 4);
2046         }
2047 
2048         for _ in 4..8 {
2049             v1025.push([0; 1025]);
2050             assert_eq!(v1025.capacity(), 8);
2051         }
2052 
2053         for _ in 8..16 {
2054             v1025.push([0; 1025]);
2055             assert_eq!(v1025.capacity(), 16);
2056         }
2057 
2058         for _ in 16..32 {
2059             v1025.push([0; 1025]);
2060             assert_eq!(v1025.capacity(), 32);
2061         }
2062 
2063         for _ in 32..64 {
2064             v1025.push([0; 1025]);
2065             assert_eq!(v1025.capacity(), 64);
2066         }
2067     }
2068 }
2069 
2070 macro_rules! generate_assert_eq_vec_and_prim {
2071     ($name:ident<$B:ident>($type:ty)) => {
2072         fn $name<A: PartialEq<$B> + Debug, $B: Debug>(a: Vec<A>, b: $type) {
2073             assert!(a == b);
2074             assert_eq!(a, b);
2075         }
2076     };
2077 }
2078 
2079 generate_assert_eq_vec_and_prim! { assert_eq_vec_and_slice  <B>(&[B])   }
2080 generate_assert_eq_vec_and_prim! { assert_eq_vec_and_array_3<B>([B; 3]) }
2081 
2082 #[test]
partialeq_vec_and_prim()2083 fn partialeq_vec_and_prim() {
2084     assert_eq_vec_and_slice(vec![1, 2, 3], &[1, 2, 3]);
2085     assert_eq_vec_and_array_3(vec![1, 2, 3], [1, 2, 3]);
2086 }
2087 
2088 macro_rules! assert_partial_eq_valid {
2089     ($a2:expr, $a3:expr; $b2:expr, $b3: expr) => {
2090         assert!($a2 == $b2);
2091         assert!($a2 != $b3);
2092         assert!($a3 != $b2);
2093         assert!($a3 == $b3);
2094         assert_eq!($a2, $b2);
2095         assert_ne!($a2, $b3);
2096         assert_ne!($a3, $b2);
2097         assert_eq!($a3, $b3);
2098     };
2099 }
2100 
2101 #[test]
partialeq_vec_full()2102 fn partialeq_vec_full() {
2103     let vec2: Vec<_> = vec![1, 2];
2104     let vec3: Vec<_> = vec![1, 2, 3];
2105     let slice2: &[_] = &[1, 2];
2106     let slice3: &[_] = &[1, 2, 3];
2107     let slicemut2: &[_] = &mut [1, 2];
2108     let slicemut3: &[_] = &mut [1, 2, 3];
2109     let array2: [_; 2] = [1, 2];
2110     let array3: [_; 3] = [1, 2, 3];
2111     let arrayref2: &[_; 2] = &[1, 2];
2112     let arrayref3: &[_; 3] = &[1, 2, 3];
2113 
2114     assert_partial_eq_valid!(vec2,vec3; vec2,vec3);
2115     assert_partial_eq_valid!(vec2,vec3; slice2,slice3);
2116     assert_partial_eq_valid!(vec2,vec3; slicemut2,slicemut3);
2117     assert_partial_eq_valid!(slice2,slice3; vec2,vec3);
2118     assert_partial_eq_valid!(slicemut2,slicemut3; vec2,vec3);
2119     assert_partial_eq_valid!(vec2,vec3; array2,array3);
2120     assert_partial_eq_valid!(vec2,vec3; arrayref2,arrayref3);
2121     assert_partial_eq_valid!(vec2,vec3; arrayref2[..],arrayref3[..]);
2122 }
2123 
2124 #[test]
test_vec_cycle()2125 fn test_vec_cycle() {
2126     #[derive(Debug)]
2127     struct C<'a> {
2128         v: Vec<Cell<Option<&'a C<'a>>>>,
2129     }
2130 
2131     impl<'a> C<'a> {
2132         fn new() -> C<'a> {
2133             C { v: Vec::new() }
2134         }
2135     }
2136 
2137     let mut c1 = C::new();
2138     let mut c2 = C::new();
2139     let mut c3 = C::new();
2140 
2141     // Push
2142     c1.v.push(Cell::new(None));
2143     c1.v.push(Cell::new(None));
2144 
2145     c2.v.push(Cell::new(None));
2146     c2.v.push(Cell::new(None));
2147 
2148     c3.v.push(Cell::new(None));
2149     c3.v.push(Cell::new(None));
2150 
2151     // Set
2152     c1.v[0].set(Some(&c2));
2153     c1.v[1].set(Some(&c3));
2154 
2155     c2.v[0].set(Some(&c2));
2156     c2.v[1].set(Some(&c3));
2157 
2158     c3.v[0].set(Some(&c1));
2159     c3.v[1].set(Some(&c2));
2160 }
2161 
2162 #[test]
test_vec_cycle_wrapped()2163 fn test_vec_cycle_wrapped() {
2164     struct Refs<'a> {
2165         v: Vec<Cell<Option<&'a C<'a>>>>,
2166     }
2167 
2168     struct C<'a> {
2169         refs: Refs<'a>,
2170     }
2171 
2172     impl<'a> Refs<'a> {
2173         fn new() -> Refs<'a> {
2174             Refs { v: Vec::new() }
2175         }
2176     }
2177 
2178     impl<'a> C<'a> {
2179         fn new() -> C<'a> {
2180             C { refs: Refs::new() }
2181         }
2182     }
2183 
2184     let mut c1 = C::new();
2185     let mut c2 = C::new();
2186     let mut c3 = C::new();
2187 
2188     c1.refs.v.push(Cell::new(None));
2189     c1.refs.v.push(Cell::new(None));
2190     c2.refs.v.push(Cell::new(None));
2191     c2.refs.v.push(Cell::new(None));
2192     c3.refs.v.push(Cell::new(None));
2193     c3.refs.v.push(Cell::new(None));
2194 
2195     c1.refs.v[0].set(Some(&c2));
2196     c1.refs.v[1].set(Some(&c3));
2197     c2.refs.v[0].set(Some(&c2));
2198     c2.refs.v[1].set(Some(&c3));
2199     c3.refs.v[0].set(Some(&c1));
2200     c3.refs.v[1].set(Some(&c2));
2201 }
2202 
2203 #[test]
test_zero_sized_capacity()2204 fn test_zero_sized_capacity() {
2205     for len in [0, 1, 2, 4, 8, 16, 32, 64, 128, 256] {
2206         let v = Vec::<()>::with_capacity(len);
2207         assert_eq!(v.len(), 0);
2208         assert_eq!(v.capacity(), usize::MAX);
2209     }
2210 }
2211 
2212 #[test]
test_zero_sized_vec_push()2213 fn test_zero_sized_vec_push() {
2214     const N: usize = 8;
2215 
2216     for len in 0..N {
2217         let mut tester = Vec::with_capacity(len);
2218         assert_eq!(tester.len(), 0);
2219         assert!(tester.capacity() >= len);
2220         for _ in 0..len {
2221             tester.push(());
2222         }
2223         assert_eq!(tester.len(), len);
2224         assert_eq!(tester.iter().count(), len);
2225         tester.clear();
2226     }
2227 }
2228 
2229 #[test]
test_vec_macro_repeat()2230 fn test_vec_macro_repeat() {
2231     assert_eq!(vec![1; 3], vec![1, 1, 1]);
2232     assert_eq!(vec![1; 2], vec![1, 1]);
2233     assert_eq!(vec![1; 1], vec![1]);
2234     assert_eq!(vec![1; 0], vec![]);
2235 
2236     // from_elem syntax (see RFC 832)
2237     let el = Box::new(1);
2238     let n = 3;
2239     assert_eq!(vec![el; n], vec![Box::new(1), Box::new(1), Box::new(1)]);
2240 }
2241 
2242 #[test]
test_vec_swap()2243 fn test_vec_swap() {
2244     let mut a: Vec<isize> = vec![0, 1, 2, 3, 4, 5, 6];
2245     a.swap(2, 4);
2246     assert_eq!(a[2], 4);
2247     assert_eq!(a[4], 2);
2248     let mut n = 42;
2249     swap(&mut n, &mut a[0]);
2250     assert_eq!(a[0], 42);
2251     assert_eq!(n, 0);
2252 }
2253 
2254 #[test]
test_extend_from_within_spec()2255 fn test_extend_from_within_spec() {
2256     #[derive(Copy)]
2257     struct CopyOnly;
2258 
2259     impl Clone for CopyOnly {
2260         fn clone(&self) -> Self {
2261             panic!("extend_from_within must use specialization on copy");
2262         }
2263     }
2264 
2265     vec![CopyOnly, CopyOnly].extend_from_within(..);
2266 }
2267 
2268 #[test]
test_extend_from_within_clone()2269 fn test_extend_from_within_clone() {
2270     let mut v = vec![String::from("sssss"), String::from("12334567890"), String::from("c")];
2271     v.extend_from_within(1..);
2272 
2273     assert_eq!(v, ["sssss", "12334567890", "c", "12334567890", "c"]);
2274 }
2275 
2276 #[test]
test_extend_from_within_complete_rande()2277 fn test_extend_from_within_complete_rande() {
2278     let mut v = vec![0, 1, 2, 3];
2279     v.extend_from_within(..);
2280 
2281     assert_eq!(v, [0, 1, 2, 3, 0, 1, 2, 3]);
2282 }
2283 
2284 #[test]
test_extend_from_within_empty_rande()2285 fn test_extend_from_within_empty_rande() {
2286     let mut v = vec![0, 1, 2, 3];
2287     v.extend_from_within(1..1);
2288 
2289     assert_eq!(v, [0, 1, 2, 3]);
2290 }
2291 
2292 #[test]
2293 #[should_panic]
test_extend_from_within_out_of_rande()2294 fn test_extend_from_within_out_of_rande() {
2295     let mut v = vec![0, 1];
2296     v.extend_from_within(..3);
2297 }
2298 
2299 #[test]
test_extend_from_within_zst()2300 fn test_extend_from_within_zst() {
2301     let mut v = vec![(); 8];
2302     v.extend_from_within(3..7);
2303 
2304     assert_eq!(v, [(); 12]);
2305 }
2306 
2307 #[test]
test_extend_from_within_empty_vec()2308 fn test_extend_from_within_empty_vec() {
2309     let mut v = Vec::<i32>::new();
2310     v.extend_from_within(..);
2311 
2312     assert_eq!(v, []);
2313 }
2314 
2315 #[test]
test_extend_from_within()2316 fn test_extend_from_within() {
2317     let mut v = vec![String::from("a"), String::from("b"), String::from("c")];
2318     v.extend_from_within(1..=2);
2319     v.extend_from_within(..=1);
2320 
2321     assert_eq!(v, ["a", "b", "c", "b", "c", "a", "b"]);
2322 }
2323 
2324 #[test]
test_vec_dedup_by()2325 fn test_vec_dedup_by() {
2326     let mut vec: Vec<i32> = vec![1, -1, 2, 3, 1, -5, 5, -2, 2];
2327 
2328     vec.dedup_by(|a, b| a.abs() == b.abs());
2329 
2330     assert_eq!(vec, [1, 2, 3, 1, -5, -2]);
2331 }
2332 
2333 #[test]
test_vec_dedup_empty()2334 fn test_vec_dedup_empty() {
2335     let mut vec: Vec<i32> = Vec::new();
2336 
2337     vec.dedup();
2338 
2339     assert_eq!(vec, []);
2340 }
2341 
2342 #[test]
test_vec_dedup_one()2343 fn test_vec_dedup_one() {
2344     let mut vec = vec![12i32];
2345 
2346     vec.dedup();
2347 
2348     assert_eq!(vec, [12]);
2349 }
2350 
2351 #[test]
test_vec_dedup_multiple_ident()2352 fn test_vec_dedup_multiple_ident() {
2353     let mut vec = vec![12, 12, 12, 12, 12, 11, 11, 11, 11, 11, 11];
2354 
2355     vec.dedup();
2356 
2357     assert_eq!(vec, [12, 11]);
2358 }
2359 
2360 #[test]
test_vec_dedup_partialeq()2361 fn test_vec_dedup_partialeq() {
2362     #[derive(Debug)]
2363     struct Foo(i32, i32);
2364 
2365     impl PartialEq for Foo {
2366         fn eq(&self, other: &Foo) -> bool {
2367             self.0 == other.0
2368         }
2369     }
2370 
2371     let mut vec = vec![Foo(0, 1), Foo(0, 5), Foo(1, 7), Foo(1, 9)];
2372 
2373     vec.dedup();
2374     assert_eq!(vec, [Foo(0, 1), Foo(1, 7)]);
2375 }
2376 
2377 #[test]
test_vec_dedup()2378 fn test_vec_dedup() {
2379     let mut vec: Vec<bool> = Vec::with_capacity(8);
2380     let mut template = vec.clone();
2381 
2382     for x in 0u8..255u8 {
2383         vec.clear();
2384         template.clear();
2385 
2386         let iter = (0..8).map(move |bit| (x >> bit) & 1 == 1);
2387         vec.extend(iter);
2388         template.extend_from_slice(&vec);
2389 
2390         let (dedup, _) = template.partition_dedup();
2391         vec.dedup();
2392 
2393         assert_eq!(vec, dedup);
2394     }
2395 }
2396 
2397 #[test]
2398 #[cfg_attr(not(panic = "unwind"), ignore = "test requires unwinding support")]
test_vec_dedup_panicking()2399 fn test_vec_dedup_panicking() {
2400     #[derive(Debug)]
2401     struct Panic<'a> {
2402         drop_counter: &'a Cell<u32>,
2403         value: bool,
2404         index: usize,
2405     }
2406 
2407     impl<'a> PartialEq for Panic<'a> {
2408         fn eq(&self, other: &Self) -> bool {
2409             self.value == other.value
2410         }
2411     }
2412 
2413     impl<'a> Drop for Panic<'a> {
2414         fn drop(&mut self) {
2415             self.drop_counter.set(self.drop_counter.get() + 1);
2416             if !std::thread::panicking() {
2417                 assert!(self.index != 4);
2418             }
2419         }
2420     }
2421 
2422     let drop_counter = &Cell::new(0);
2423     let expected = [
2424         Panic { drop_counter, value: false, index: 0 },
2425         Panic { drop_counter, value: false, index: 5 },
2426         Panic { drop_counter, value: true, index: 6 },
2427         Panic { drop_counter, value: true, index: 7 },
2428     ];
2429     let mut vec = vec![
2430         Panic { drop_counter, value: false, index: 0 },
2431         // these elements get deduplicated
2432         Panic { drop_counter, value: false, index: 1 },
2433         Panic { drop_counter, value: false, index: 2 },
2434         Panic { drop_counter, value: false, index: 3 },
2435         Panic { drop_counter, value: false, index: 4 },
2436         // here it panics while dropping the item with index==4
2437         Panic { drop_counter, value: false, index: 5 },
2438         Panic { drop_counter, value: true, index: 6 },
2439         Panic { drop_counter, value: true, index: 7 },
2440     ];
2441 
2442     let _ = catch_unwind(AssertUnwindSafe(|| vec.dedup())).unwrap_err();
2443 
2444     assert_eq!(drop_counter.get(), 4);
2445 
2446     let ok = vec.iter().zip(expected.iter()).all(|(x, y)| x.index == y.index);
2447 
2448     if !ok {
2449         panic!("expected: {expected:?}\ngot: {vec:?}\n");
2450     }
2451 }
2452 
2453 // Regression test for issue #82533
2454 #[test]
2455 #[cfg_attr(not(panic = "unwind"), ignore = "test requires unwinding support")]
test_extend_from_within_panicking_clone()2456 fn test_extend_from_within_panicking_clone() {
2457     struct Panic<'dc> {
2458         drop_count: &'dc AtomicU32,
2459         aaaaa: bool,
2460     }
2461 
2462     impl Clone for Panic<'_> {
2463         fn clone(&self) -> Self {
2464             if self.aaaaa {
2465                 panic!("panic! at the clone");
2466             }
2467 
2468             Self { ..*self }
2469         }
2470     }
2471 
2472     impl Drop for Panic<'_> {
2473         fn drop(&mut self) {
2474             self.drop_count.fetch_add(1, Ordering::SeqCst);
2475         }
2476     }
2477 
2478     let count = core::sync::atomic::AtomicU32::new(0);
2479     let mut vec = vec![
2480         Panic { drop_count: &count, aaaaa: false },
2481         Panic { drop_count: &count, aaaaa: true },
2482         Panic { drop_count: &count, aaaaa: false },
2483     ];
2484 
2485     // This should clone&append one Panic{..} at the end, and then panic while
2486     // cloning second Panic{..}. This means that `Panic::drop` should be called
2487     // 4 times (3 for items already in vector, 1 for just appended).
2488     //
2489     // Previously just appended item was leaked, making drop_count = 3, instead of 4.
2490     std::panic::catch_unwind(move || vec.extend_from_within(..)).unwrap_err();
2491 
2492     assert_eq!(count.load(Ordering::SeqCst), 4);
2493 }
2494 
2495 #[test]
2496 #[should_panic = "vec len overflow"]
test_into_flattened_size_overflow()2497 fn test_into_flattened_size_overflow() {
2498     let v = vec![[(); usize::MAX]; 2];
2499     let _ = v.into_flattened();
2500 }
2501