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