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