]> git.proxmox.com Git - rustc.git/blob - library/alloc/tests/vec.rs
New upstream version 1.52.0+dfsg1
[rustc.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_during_iteration_drops() {
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_from_iter_specialization_panic_during_drop_leaks() {
1055 static mut DROP_COUNTER: usize = 0;
1056
1057 #[derive(Debug)]
1058 enum Droppable {
1059 DroppedTwice(Box<i32>),
1060 PanicOnDrop,
1061 }
1062
1063 impl Drop for Droppable {
1064 fn drop(&mut self) {
1065 match self {
1066 Droppable::DroppedTwice(_) => {
1067 unsafe {
1068 DROP_COUNTER += 1;
1069 }
1070 println!("Dropping!")
1071 }
1072 Droppable::PanicOnDrop => {
1073 if !std::thread::panicking() {
1074 panic!();
1075 }
1076 }
1077 }
1078 }
1079 }
1080
1081 let _ = std::panic::catch_unwind(AssertUnwindSafe(|| {
1082 let v = vec![Droppable::DroppedTwice(Box::new(123)), Droppable::PanicOnDrop];
1083 let _ = v.into_iter().take(0).collect::<Vec<_>>();
1084 }));
1085
1086 assert_eq!(unsafe { DROP_COUNTER }, 1);
1087 }
1088
1089 #[test]
1090 fn test_cow_from() {
1091 let borrowed: &[_] = &["borrowed", "(slice)"];
1092 let owned = vec!["owned", "(vec)"];
1093 match (Cow::from(owned.clone()), Cow::from(borrowed)) {
1094 (Cow::Owned(o), Cow::Borrowed(b)) => assert!(o == owned && b == borrowed),
1095 _ => panic!("invalid `Cow::from`"),
1096 }
1097 }
1098
1099 #[test]
1100 fn test_from_cow() {
1101 let borrowed: &[_] = &["borrowed", "(slice)"];
1102 let owned = vec!["owned", "(vec)"];
1103 assert_eq!(Vec::from(Cow::Borrowed(borrowed)), vec!["borrowed", "(slice)"]);
1104 assert_eq!(Vec::from(Cow::Owned(owned)), vec!["owned", "(vec)"]);
1105 }
1106
1107 #[allow(dead_code)]
1108 fn assert_covariance() {
1109 fn drain<'new>(d: Drain<'static, &'static str>) -> Drain<'new, &'new str> {
1110 d
1111 }
1112 fn into_iter<'new>(i: IntoIter<&'static str>) -> IntoIter<&'new str> {
1113 i
1114 }
1115 }
1116
1117 #[test]
1118 fn from_into_inner() {
1119 let vec = vec![1, 2, 3];
1120 let ptr = vec.as_ptr();
1121 let vec = vec.into_iter().collect::<Vec<_>>();
1122 assert_eq!(vec, [1, 2, 3]);
1123 assert_eq!(vec.as_ptr(), ptr);
1124
1125 let ptr = &vec[1] as *const _;
1126 let mut it = vec.into_iter();
1127 it.next().unwrap();
1128 let vec = it.collect::<Vec<_>>();
1129 assert_eq!(vec, [2, 3]);
1130 assert!(ptr != vec.as_ptr());
1131 }
1132
1133 #[test]
1134 fn overaligned_allocations() {
1135 #[repr(align(256))]
1136 struct Foo(usize);
1137 let mut v = vec![Foo(273)];
1138 for i in 0..0x1000 {
1139 v.reserve_exact(i);
1140 assert!(v[0].0 == 273);
1141 assert!(v.as_ptr() as usize & 0xff == 0);
1142 v.shrink_to_fit();
1143 assert!(v[0].0 == 273);
1144 assert!(v.as_ptr() as usize & 0xff == 0);
1145 }
1146 }
1147
1148 #[test]
1149 fn drain_filter_empty() {
1150 let mut vec: Vec<i32> = vec![];
1151
1152 {
1153 let mut iter = vec.drain_filter(|_| true);
1154 assert_eq!(iter.size_hint(), (0, Some(0)));
1155 assert_eq!(iter.next(), None);
1156 assert_eq!(iter.size_hint(), (0, Some(0)));
1157 assert_eq!(iter.next(), None);
1158 assert_eq!(iter.size_hint(), (0, Some(0)));
1159 }
1160 assert_eq!(vec.len(), 0);
1161 assert_eq!(vec, vec![]);
1162 }
1163
1164 #[test]
1165 fn drain_filter_zst() {
1166 let mut vec = vec![(), (), (), (), ()];
1167 let initial_len = vec.len();
1168 let mut count = 0;
1169 {
1170 let mut iter = vec.drain_filter(|_| true);
1171 assert_eq!(iter.size_hint(), (0, Some(initial_len)));
1172 while let Some(_) = iter.next() {
1173 count += 1;
1174 assert_eq!(iter.size_hint(), (0, Some(initial_len - count)));
1175 }
1176 assert_eq!(iter.size_hint(), (0, Some(0)));
1177 assert_eq!(iter.next(), None);
1178 assert_eq!(iter.size_hint(), (0, Some(0)));
1179 }
1180
1181 assert_eq!(count, initial_len);
1182 assert_eq!(vec.len(), 0);
1183 assert_eq!(vec, vec![]);
1184 }
1185
1186 #[test]
1187 fn drain_filter_false() {
1188 let mut vec = vec![1, 2, 3, 4, 5, 6, 7, 8, 9, 10];
1189
1190 let initial_len = vec.len();
1191 let mut count = 0;
1192 {
1193 let mut iter = vec.drain_filter(|_| false);
1194 assert_eq!(iter.size_hint(), (0, Some(initial_len)));
1195 for _ in iter.by_ref() {
1196 count += 1;
1197 }
1198 assert_eq!(iter.size_hint(), (0, Some(0)));
1199 assert_eq!(iter.next(), None);
1200 assert_eq!(iter.size_hint(), (0, Some(0)));
1201 }
1202
1203 assert_eq!(count, 0);
1204 assert_eq!(vec.len(), initial_len);
1205 assert_eq!(vec, vec![1, 2, 3, 4, 5, 6, 7, 8, 9, 10]);
1206 }
1207
1208 #[test]
1209 fn drain_filter_true() {
1210 let mut vec = vec![1, 2, 3, 4, 5, 6, 7, 8, 9, 10];
1211
1212 let initial_len = vec.len();
1213 let mut count = 0;
1214 {
1215 let mut iter = vec.drain_filter(|_| true);
1216 assert_eq!(iter.size_hint(), (0, Some(initial_len)));
1217 while let Some(_) = iter.next() {
1218 count += 1;
1219 assert_eq!(iter.size_hint(), (0, Some(initial_len - count)));
1220 }
1221 assert_eq!(iter.size_hint(), (0, Some(0)));
1222 assert_eq!(iter.next(), None);
1223 assert_eq!(iter.size_hint(), (0, Some(0)));
1224 }
1225
1226 assert_eq!(count, initial_len);
1227 assert_eq!(vec.len(), 0);
1228 assert_eq!(vec, vec![]);
1229 }
1230
1231 #[test]
1232 fn drain_filter_complex() {
1233 {
1234 // [+xxx++++++xxxxx++++x+x++]
1235 let mut vec = vec![
1236 1, 2, 4, 6, 7, 9, 11, 13, 15, 17, 18, 20, 22, 24, 26, 27, 29, 31, 33, 34, 35, 36, 37,
1237 39,
1238 ];
1239
1240 let removed = vec.drain_filter(|x| *x % 2 == 0).collect::<Vec<_>>();
1241 assert_eq!(removed.len(), 10);
1242 assert_eq!(removed, vec![2, 4, 6, 18, 20, 22, 24, 26, 34, 36]);
1243
1244 assert_eq!(vec.len(), 14);
1245 assert_eq!(vec, vec![1, 7, 9, 11, 13, 15, 17, 27, 29, 31, 33, 35, 37, 39]);
1246 }
1247
1248 {
1249 // [xxx++++++xxxxx++++x+x++]
1250 let mut vec = vec![
1251 2, 4, 6, 7, 9, 11, 13, 15, 17, 18, 20, 22, 24, 26, 27, 29, 31, 33, 34, 35, 36, 37, 39,
1252 ];
1253
1254 let removed = vec.drain_filter(|x| *x % 2 == 0).collect::<Vec<_>>();
1255 assert_eq!(removed.len(), 10);
1256 assert_eq!(removed, vec![2, 4, 6, 18, 20, 22, 24, 26, 34, 36]);
1257
1258 assert_eq!(vec.len(), 13);
1259 assert_eq!(vec, vec![7, 9, 11, 13, 15, 17, 27, 29, 31, 33, 35, 37, 39]);
1260 }
1261
1262 {
1263 // [xxx++++++xxxxx++++x+x]
1264 let mut vec =
1265 vec![2, 4, 6, 7, 9, 11, 13, 15, 17, 18, 20, 22, 24, 26, 27, 29, 31, 33, 34, 35, 36];
1266
1267 let removed = vec.drain_filter(|x| *x % 2 == 0).collect::<Vec<_>>();
1268 assert_eq!(removed.len(), 10);
1269 assert_eq!(removed, vec![2, 4, 6, 18, 20, 22, 24, 26, 34, 36]);
1270
1271 assert_eq!(vec.len(), 11);
1272 assert_eq!(vec, vec![7, 9, 11, 13, 15, 17, 27, 29, 31, 33, 35]);
1273 }
1274
1275 {
1276 // [xxxxxxxxxx+++++++++++]
1277 let mut vec = vec![2, 4, 6, 8, 10, 12, 14, 16, 18, 20, 1, 3, 5, 7, 9, 11, 13, 15, 17, 19];
1278
1279 let removed = vec.drain_filter(|x| *x % 2 == 0).collect::<Vec<_>>();
1280 assert_eq!(removed.len(), 10);
1281 assert_eq!(removed, vec![2, 4, 6, 8, 10, 12, 14, 16, 18, 20]);
1282
1283 assert_eq!(vec.len(), 10);
1284 assert_eq!(vec, vec![1, 3, 5, 7, 9, 11, 13, 15, 17, 19]);
1285 }
1286
1287 {
1288 // [+++++++++++xxxxxxxxxx]
1289 let mut vec = vec![1, 3, 5, 7, 9, 11, 13, 15, 17, 19, 2, 4, 6, 8, 10, 12, 14, 16, 18, 20];
1290
1291 let removed = vec.drain_filter(|x| *x % 2 == 0).collect::<Vec<_>>();
1292 assert_eq!(removed.len(), 10);
1293 assert_eq!(removed, vec![2, 4, 6, 8, 10, 12, 14, 16, 18, 20]);
1294
1295 assert_eq!(vec.len(), 10);
1296 assert_eq!(vec, vec![1, 3, 5, 7, 9, 11, 13, 15, 17, 19]);
1297 }
1298 }
1299
1300 // FIXME: re-enable emscripten once it can unwind again
1301 #[test]
1302 #[cfg(not(target_os = "emscripten"))]
1303 fn drain_filter_consumed_panic() {
1304 use std::rc::Rc;
1305 use std::sync::Mutex;
1306
1307 struct Check {
1308 index: usize,
1309 drop_counts: Rc<Mutex<Vec<usize>>>,
1310 }
1311
1312 impl Drop for Check {
1313 fn drop(&mut self) {
1314 self.drop_counts.lock().unwrap()[self.index] += 1;
1315 println!("drop: {}", self.index);
1316 }
1317 }
1318
1319 let check_count = 10;
1320 let drop_counts = Rc::new(Mutex::new(vec![0_usize; check_count]));
1321 let mut data: Vec<Check> = (0..check_count)
1322 .map(|index| Check { index, drop_counts: Rc::clone(&drop_counts) })
1323 .collect();
1324
1325 let _ = std::panic::catch_unwind(move || {
1326 let filter = |c: &mut Check| {
1327 if c.index == 2 {
1328 panic!("panic at index: {}", c.index);
1329 }
1330 // Verify that if the filter could panic again on another element
1331 // that it would not cause a double panic and all elements of the
1332 // vec would still be dropped exactly once.
1333 if c.index == 4 {
1334 panic!("panic at index: {}", c.index);
1335 }
1336 c.index < 6
1337 };
1338 let drain = data.drain_filter(filter);
1339
1340 // NOTE: The DrainFilter is explicitly consumed
1341 drain.for_each(drop);
1342 });
1343
1344 let drop_counts = drop_counts.lock().unwrap();
1345 assert_eq!(check_count, drop_counts.len());
1346
1347 for (index, count) in drop_counts.iter().cloned().enumerate() {
1348 assert_eq!(1, count, "unexpected drop count at index: {} (count: {})", index, count);
1349 }
1350 }
1351
1352 // FIXME: Re-enable emscripten once it can catch panics
1353 #[test]
1354 #[cfg(not(target_os = "emscripten"))]
1355 fn drain_filter_unconsumed_panic() {
1356 use std::rc::Rc;
1357 use std::sync::Mutex;
1358
1359 struct Check {
1360 index: usize,
1361 drop_counts: Rc<Mutex<Vec<usize>>>,
1362 }
1363
1364 impl Drop for Check {
1365 fn drop(&mut self) {
1366 self.drop_counts.lock().unwrap()[self.index] += 1;
1367 println!("drop: {}", self.index);
1368 }
1369 }
1370
1371 let check_count = 10;
1372 let drop_counts = Rc::new(Mutex::new(vec![0_usize; check_count]));
1373 let mut data: Vec<Check> = (0..check_count)
1374 .map(|index| Check { index, drop_counts: Rc::clone(&drop_counts) })
1375 .collect();
1376
1377 let _ = std::panic::catch_unwind(move || {
1378 let filter = |c: &mut Check| {
1379 if c.index == 2 {
1380 panic!("panic at index: {}", c.index);
1381 }
1382 // Verify that if the filter could panic again on another element
1383 // that it would not cause a double panic and all elements of the
1384 // vec would still be dropped exactly once.
1385 if c.index == 4 {
1386 panic!("panic at index: {}", c.index);
1387 }
1388 c.index < 6
1389 };
1390 let _drain = data.drain_filter(filter);
1391
1392 // NOTE: The DrainFilter is dropped without being consumed
1393 });
1394
1395 let drop_counts = drop_counts.lock().unwrap();
1396 assert_eq!(check_count, drop_counts.len());
1397
1398 for (index, count) in drop_counts.iter().cloned().enumerate() {
1399 assert_eq!(1, count, "unexpected drop count at index: {} (count: {})", index, count);
1400 }
1401 }
1402
1403 #[test]
1404 fn drain_filter_unconsumed() {
1405 let mut vec = vec![1, 2, 3, 4];
1406 let drain = vec.drain_filter(|&mut x| x % 2 != 0);
1407 drop(drain);
1408 assert_eq!(vec, [2, 4]);
1409 }
1410
1411 #[test]
1412 fn test_reserve_exact() {
1413 // This is all the same as test_reserve
1414
1415 let mut v = Vec::new();
1416 assert_eq!(v.capacity(), 0);
1417
1418 v.reserve_exact(2);
1419 assert!(v.capacity() >= 2);
1420
1421 for i in 0..16 {
1422 v.push(i);
1423 }
1424
1425 assert!(v.capacity() >= 16);
1426 v.reserve_exact(16);
1427 assert!(v.capacity() >= 32);
1428
1429 v.push(16);
1430
1431 v.reserve_exact(16);
1432 assert!(v.capacity() >= 33)
1433 }
1434
1435 #[test]
1436 #[cfg_attr(miri, ignore)] // Miri does not support signalling OOM
1437 #[cfg_attr(target_os = "android", ignore)] // Android used in CI has a broken dlmalloc
1438 fn test_try_reserve() {
1439 // These are the interesting cases:
1440 // * exactly isize::MAX should never trigger a CapacityOverflow (can be OOM)
1441 // * > isize::MAX should always fail
1442 // * On 16/32-bit should CapacityOverflow
1443 // * On 64-bit should OOM
1444 // * overflow may trigger when adding `len` to `cap` (in number of elements)
1445 // * overflow may trigger when multiplying `new_cap` by size_of::<T> (to get bytes)
1446
1447 const MAX_CAP: usize = isize::MAX as usize;
1448 const MAX_USIZE: usize = usize::MAX;
1449
1450 // On 16/32-bit, we check that allocations don't exceed isize::MAX,
1451 // on 64-bit, we assume the OS will give an OOM for such a ridiculous size.
1452 // Any platform that succeeds for these requests is technically broken with
1453 // ptr::offset because LLVM is the worst.
1454 let guards_against_isize = usize::BITS < 64;
1455
1456 {
1457 // Note: basic stuff is checked by test_reserve
1458 let mut empty_bytes: Vec<u8> = Vec::new();
1459
1460 // Check isize::MAX doesn't count as an overflow
1461 if let Err(CapacityOverflow) = empty_bytes.try_reserve(MAX_CAP) {
1462 panic!("isize::MAX shouldn't trigger an overflow!");
1463 }
1464 // Play it again, frank! (just to be sure)
1465 if let Err(CapacityOverflow) = empty_bytes.try_reserve(MAX_CAP) {
1466 panic!("isize::MAX shouldn't trigger an overflow!");
1467 }
1468
1469 if guards_against_isize {
1470 // Check isize::MAX + 1 does count as overflow
1471 if let Err(CapacityOverflow) = empty_bytes.try_reserve(MAX_CAP + 1) {
1472 } else {
1473 panic!("isize::MAX + 1 should trigger an overflow!")
1474 }
1475
1476 // Check usize::MAX does count as overflow
1477 if let Err(CapacityOverflow) = empty_bytes.try_reserve(MAX_USIZE) {
1478 } else {
1479 panic!("usize::MAX should trigger an overflow!")
1480 }
1481 } else {
1482 // Check isize::MAX + 1 is an OOM
1483 if let Err(AllocError { .. }) = empty_bytes.try_reserve(MAX_CAP + 1) {
1484 } else {
1485 panic!("isize::MAX + 1 should trigger an OOM!")
1486 }
1487
1488 // Check usize::MAX is an OOM
1489 if let Err(AllocError { .. }) = empty_bytes.try_reserve(MAX_USIZE) {
1490 } else {
1491 panic!("usize::MAX should trigger an OOM!")
1492 }
1493 }
1494 }
1495
1496 {
1497 // Same basic idea, but with non-zero len
1498 let mut ten_bytes: Vec<u8> = vec![1, 2, 3, 4, 5, 6, 7, 8, 9, 10];
1499
1500 if let Err(CapacityOverflow) = ten_bytes.try_reserve(MAX_CAP - 10) {
1501 panic!("isize::MAX shouldn't trigger an overflow!");
1502 }
1503 if let Err(CapacityOverflow) = ten_bytes.try_reserve(MAX_CAP - 10) {
1504 panic!("isize::MAX shouldn't trigger an overflow!");
1505 }
1506 if guards_against_isize {
1507 if let Err(CapacityOverflow) = ten_bytes.try_reserve(MAX_CAP - 9) {
1508 } else {
1509 panic!("isize::MAX + 1 should trigger an overflow!");
1510 }
1511 } else {
1512 if let Err(AllocError { .. }) = ten_bytes.try_reserve(MAX_CAP - 9) {
1513 } else {
1514 panic!("isize::MAX + 1 should trigger an OOM!")
1515 }
1516 }
1517 // Should always overflow in the add-to-len
1518 if let Err(CapacityOverflow) = ten_bytes.try_reserve(MAX_USIZE) {
1519 } else {
1520 panic!("usize::MAX should trigger an overflow!")
1521 }
1522 }
1523
1524 {
1525 // Same basic idea, but with interesting type size
1526 let mut ten_u32s: Vec<u32> = vec![1, 2, 3, 4, 5, 6, 7, 8, 9, 10];
1527
1528 if let Err(CapacityOverflow) = ten_u32s.try_reserve(MAX_CAP / 4 - 10) {
1529 panic!("isize::MAX shouldn't trigger an overflow!");
1530 }
1531 if let Err(CapacityOverflow) = ten_u32s.try_reserve(MAX_CAP / 4 - 10) {
1532 panic!("isize::MAX shouldn't trigger an overflow!");
1533 }
1534 if guards_against_isize {
1535 if let Err(CapacityOverflow) = ten_u32s.try_reserve(MAX_CAP / 4 - 9) {
1536 } else {
1537 panic!("isize::MAX + 1 should trigger an overflow!");
1538 }
1539 } else {
1540 if let Err(AllocError { .. }) = ten_u32s.try_reserve(MAX_CAP / 4 - 9) {
1541 } else {
1542 panic!("isize::MAX + 1 should trigger an OOM!")
1543 }
1544 }
1545 // Should fail in the mul-by-size
1546 if let Err(CapacityOverflow) = ten_u32s.try_reserve(MAX_USIZE - 20) {
1547 } else {
1548 panic!("usize::MAX should trigger an overflow!");
1549 }
1550 }
1551 }
1552
1553 #[test]
1554 #[cfg_attr(miri, ignore)] // Miri does not support signalling OOM
1555 #[cfg_attr(target_os = "android", ignore)] // Android used in CI has a broken dlmalloc
1556 fn test_try_reserve_exact() {
1557 // This is exactly the same as test_try_reserve with the method changed.
1558 // See that test for comments.
1559
1560 const MAX_CAP: usize = isize::MAX as usize;
1561 const MAX_USIZE: usize = usize::MAX;
1562
1563 let guards_against_isize = size_of::<usize>() < 8;
1564
1565 {
1566 let mut empty_bytes: Vec<u8> = Vec::new();
1567
1568 if let Err(CapacityOverflow) = empty_bytes.try_reserve_exact(MAX_CAP) {
1569 panic!("isize::MAX shouldn't trigger an overflow!");
1570 }
1571 if let Err(CapacityOverflow) = empty_bytes.try_reserve_exact(MAX_CAP) {
1572 panic!("isize::MAX shouldn't trigger an overflow!");
1573 }
1574
1575 if guards_against_isize {
1576 if let Err(CapacityOverflow) = empty_bytes.try_reserve_exact(MAX_CAP + 1) {
1577 } else {
1578 panic!("isize::MAX + 1 should trigger an overflow!")
1579 }
1580
1581 if let Err(CapacityOverflow) = empty_bytes.try_reserve_exact(MAX_USIZE) {
1582 } else {
1583 panic!("usize::MAX should trigger an overflow!")
1584 }
1585 } else {
1586 if let Err(AllocError { .. }) = empty_bytes.try_reserve_exact(MAX_CAP + 1) {
1587 } else {
1588 panic!("isize::MAX + 1 should trigger an OOM!")
1589 }
1590
1591 if let Err(AllocError { .. }) = empty_bytes.try_reserve_exact(MAX_USIZE) {
1592 } else {
1593 panic!("usize::MAX should trigger an OOM!")
1594 }
1595 }
1596 }
1597
1598 {
1599 let mut ten_bytes: Vec<u8> = vec![1, 2, 3, 4, 5, 6, 7, 8, 9, 10];
1600
1601 if let Err(CapacityOverflow) = ten_bytes.try_reserve_exact(MAX_CAP - 10) {
1602 panic!("isize::MAX shouldn't trigger an overflow!");
1603 }
1604 if let Err(CapacityOverflow) = ten_bytes.try_reserve_exact(MAX_CAP - 10) {
1605 panic!("isize::MAX shouldn't trigger an overflow!");
1606 }
1607 if guards_against_isize {
1608 if let Err(CapacityOverflow) = ten_bytes.try_reserve_exact(MAX_CAP - 9) {
1609 } else {
1610 panic!("isize::MAX + 1 should trigger an overflow!");
1611 }
1612 } else {
1613 if let Err(AllocError { .. }) = ten_bytes.try_reserve_exact(MAX_CAP - 9) {
1614 } else {
1615 panic!("isize::MAX + 1 should trigger an OOM!")
1616 }
1617 }
1618 if let Err(CapacityOverflow) = ten_bytes.try_reserve_exact(MAX_USIZE) {
1619 } else {
1620 panic!("usize::MAX should trigger an overflow!")
1621 }
1622 }
1623
1624 {
1625 let mut ten_u32s: Vec<u32> = vec![1, 2, 3, 4, 5, 6, 7, 8, 9, 10];
1626
1627 if let Err(CapacityOverflow) = ten_u32s.try_reserve_exact(MAX_CAP / 4 - 10) {
1628 panic!("isize::MAX shouldn't trigger an overflow!");
1629 }
1630 if let Err(CapacityOverflow) = ten_u32s.try_reserve_exact(MAX_CAP / 4 - 10) {
1631 panic!("isize::MAX shouldn't trigger an overflow!");
1632 }
1633 if guards_against_isize {
1634 if let Err(CapacityOverflow) = ten_u32s.try_reserve_exact(MAX_CAP / 4 - 9) {
1635 } else {
1636 panic!("isize::MAX + 1 should trigger an overflow!");
1637 }
1638 } else {
1639 if let Err(AllocError { .. }) = ten_u32s.try_reserve_exact(MAX_CAP / 4 - 9) {
1640 } else {
1641 panic!("isize::MAX + 1 should trigger an OOM!")
1642 }
1643 }
1644 if let Err(CapacityOverflow) = ten_u32s.try_reserve_exact(MAX_USIZE - 20) {
1645 } else {
1646 panic!("usize::MAX should trigger an overflow!")
1647 }
1648 }
1649 }
1650
1651 #[test]
1652 fn test_stable_pointers() {
1653 /// Pull an element from the iterator, then drop it.
1654 /// Useful to cover both the `next` and `drop` paths of an iterator.
1655 fn next_then_drop<I: Iterator>(mut i: I) {
1656 i.next().unwrap();
1657 drop(i);
1658 }
1659
1660 // Test that, if we reserved enough space, adding and removing elements does not
1661 // invalidate references into the vector (such as `v0`). This test also
1662 // runs in Miri, which would detect such problems.
1663 // Note that this test does *not* constitute a stable guarantee that all these functions do not
1664 // reallocate! Only what is explicitly documented at
1665 // <https://doc.rust-lang.org/nightly/std/vec/struct.Vec.html#guarantees> is stably guaranteed.
1666 let mut v = Vec::with_capacity(128);
1667 v.push(13);
1668
1669 // Laundering the lifetime -- we take care that `v` does not reallocate, so that's okay.
1670 let v0 = &mut v[0];
1671 let v0 = unsafe { &mut *(v0 as *mut _) };
1672 // Now do a bunch of things and occasionally use `v0` again to assert it is still valid.
1673
1674 // Pushing/inserting and popping/removing
1675 v.push(1);
1676 v.push(2);
1677 v.insert(1, 1);
1678 assert_eq!(*v0, 13);
1679 v.remove(1);
1680 v.pop().unwrap();
1681 assert_eq!(*v0, 13);
1682 v.push(1);
1683 v.swap_remove(1);
1684 assert_eq!(v.len(), 2);
1685 v.swap_remove(1); // swap_remove the last element
1686 assert_eq!(*v0, 13);
1687
1688 // Appending
1689 v.append(&mut vec![27, 19]);
1690 assert_eq!(*v0, 13);
1691
1692 // Extending
1693 v.extend_from_slice(&[1, 2]);
1694 v.extend(&[1, 2]); // `slice::Iter` (with `T: Copy`) specialization
1695 v.extend(vec![2, 3]); // `vec::IntoIter` specialization
1696 v.extend(std::iter::once(3)); // `TrustedLen` specialization
1697 v.extend(std::iter::empty::<i32>()); // `TrustedLen` specialization with empty iterator
1698 v.extend(std::iter::once(3).filter(|_| true)); // base case
1699 v.extend(std::iter::once(&3)); // `cloned` specialization
1700 assert_eq!(*v0, 13);
1701
1702 // Truncation
1703 v.truncate(2);
1704 assert_eq!(*v0, 13);
1705
1706 // Resizing
1707 v.resize_with(v.len() + 10, || 42);
1708 assert_eq!(*v0, 13);
1709 v.resize_with(2, || panic!());
1710 assert_eq!(*v0, 13);
1711
1712 // No-op reservation
1713 v.reserve(32);
1714 v.reserve_exact(32);
1715 assert_eq!(*v0, 13);
1716
1717 // Partial draining
1718 v.resize_with(10, || 42);
1719 next_then_drop(v.drain(5..));
1720 assert_eq!(*v0, 13);
1721
1722 // Splicing
1723 v.resize_with(10, || 42);
1724 next_then_drop(v.splice(5.., vec![1, 2, 3, 4, 5])); // empty tail after range
1725 assert_eq!(*v0, 13);
1726 next_then_drop(v.splice(5..8, vec![1])); // replacement is smaller than original range
1727 assert_eq!(*v0, 13);
1728 next_then_drop(v.splice(5..6, vec![1; 10].into_iter().filter(|_| true))); // lower bound not exact
1729 assert_eq!(*v0, 13);
1730
1731 // spare_capacity_mut
1732 v.spare_capacity_mut();
1733 assert_eq!(*v0, 13);
1734
1735 // Smoke test that would fire even outside Miri if an actual relocation happened.
1736 *v0 -= 13;
1737 assert_eq!(v[0], 0);
1738 }
1739
1740 // https://github.com/rust-lang/rust/pull/49496 introduced specialization based on:
1741 //
1742 // ```
1743 // unsafe impl<T: ?Sized> IsZero for *mut T {
1744 // fn is_zero(&self) -> bool {
1745 // (*self).is_null()
1746 // }
1747 // }
1748 // ```
1749 //
1750 // … to call `RawVec::with_capacity_zeroed` for creating `Vec<*mut T>`,
1751 // which is incorrect for fat pointers since `<*mut T>::is_null` only looks at the data component.
1752 // That is, a fat pointer can be “null” without being made entirely of zero bits.
1753 #[test]
1754 fn vec_macro_repeating_null_raw_fat_pointer() {
1755 let raw_dyn = &mut (|| ()) as &mut dyn Fn() as *mut dyn Fn();
1756 let vtable = dbg!(ptr_metadata(raw_dyn));
1757 let null_raw_dyn = ptr_from_raw_parts(std::ptr::null_mut(), vtable);
1758 assert!(null_raw_dyn.is_null());
1759
1760 let vec = vec![null_raw_dyn; 1];
1761 dbg!(ptr_metadata(vec[0]));
1762 assert!(vec[0] == null_raw_dyn);
1763
1764 // Polyfill for https://github.com/rust-lang/rfcs/pull/2580
1765
1766 fn ptr_metadata(ptr: *mut dyn Fn()) -> *mut () {
1767 unsafe { std::mem::transmute::<*mut dyn Fn(), DynRepr>(ptr).vtable }
1768 }
1769
1770 fn ptr_from_raw_parts(data: *mut (), vtable: *mut ()) -> *mut dyn Fn() {
1771 unsafe { std::mem::transmute::<DynRepr, *mut dyn Fn()>(DynRepr { data, vtable }) }
1772 }
1773
1774 #[repr(C)]
1775 struct DynRepr {
1776 data: *mut (),
1777 vtable: *mut (),
1778 }
1779 }
1780
1781 // This test will likely fail if you change the capacities used in
1782 // `RawVec::grow_amortized`.
1783 #[test]
1784 fn test_push_growth_strategy() {
1785 // If the element size is 1, we jump from 0 to 8, then double.
1786 {
1787 let mut v1: Vec<u8> = vec![];
1788 assert_eq!(v1.capacity(), 0);
1789
1790 for _ in 0..8 {
1791 v1.push(0);
1792 assert_eq!(v1.capacity(), 8);
1793 }
1794
1795 for _ in 8..16 {
1796 v1.push(0);
1797 assert_eq!(v1.capacity(), 16);
1798 }
1799
1800 for _ in 16..32 {
1801 v1.push(0);
1802 assert_eq!(v1.capacity(), 32);
1803 }
1804
1805 for _ in 32..64 {
1806 v1.push(0);
1807 assert_eq!(v1.capacity(), 64);
1808 }
1809 }
1810
1811 // If the element size is 2..=1024, we jump from 0 to 4, then double.
1812 {
1813 let mut v2: Vec<u16> = vec![];
1814 let mut v1024: Vec<[u8; 1024]> = vec![];
1815 assert_eq!(v2.capacity(), 0);
1816 assert_eq!(v1024.capacity(), 0);
1817
1818 for _ in 0..4 {
1819 v2.push(0);
1820 v1024.push([0; 1024]);
1821 assert_eq!(v2.capacity(), 4);
1822 assert_eq!(v1024.capacity(), 4);
1823 }
1824
1825 for _ in 4..8 {
1826 v2.push(0);
1827 v1024.push([0; 1024]);
1828 assert_eq!(v2.capacity(), 8);
1829 assert_eq!(v1024.capacity(), 8);
1830 }
1831
1832 for _ in 8..16 {
1833 v2.push(0);
1834 v1024.push([0; 1024]);
1835 assert_eq!(v2.capacity(), 16);
1836 assert_eq!(v1024.capacity(), 16);
1837 }
1838
1839 for _ in 16..32 {
1840 v2.push(0);
1841 v1024.push([0; 1024]);
1842 assert_eq!(v2.capacity(), 32);
1843 assert_eq!(v1024.capacity(), 32);
1844 }
1845
1846 for _ in 32..64 {
1847 v2.push(0);
1848 v1024.push([0; 1024]);
1849 assert_eq!(v2.capacity(), 64);
1850 assert_eq!(v1024.capacity(), 64);
1851 }
1852 }
1853
1854 // If the element size is > 1024, we jump from 0 to 1, then double.
1855 {
1856 let mut v1025: Vec<[u8; 1025]> = vec![];
1857 assert_eq!(v1025.capacity(), 0);
1858
1859 for _ in 0..1 {
1860 v1025.push([0; 1025]);
1861 assert_eq!(v1025.capacity(), 1);
1862 }
1863
1864 for _ in 1..2 {
1865 v1025.push([0; 1025]);
1866 assert_eq!(v1025.capacity(), 2);
1867 }
1868
1869 for _ in 2..4 {
1870 v1025.push([0; 1025]);
1871 assert_eq!(v1025.capacity(), 4);
1872 }
1873
1874 for _ in 4..8 {
1875 v1025.push([0; 1025]);
1876 assert_eq!(v1025.capacity(), 8);
1877 }
1878
1879 for _ in 8..16 {
1880 v1025.push([0; 1025]);
1881 assert_eq!(v1025.capacity(), 16);
1882 }
1883
1884 for _ in 16..32 {
1885 v1025.push([0; 1025]);
1886 assert_eq!(v1025.capacity(), 32);
1887 }
1888
1889 for _ in 32..64 {
1890 v1025.push([0; 1025]);
1891 assert_eq!(v1025.capacity(), 64);
1892 }
1893 }
1894 }
1895
1896 macro_rules! generate_assert_eq_vec_and_prim {
1897 ($name:ident<$B:ident>($type:ty)) => {
1898 fn $name<A: PartialEq<$B> + Debug, $B: Debug>(a: Vec<A>, b: $type) {
1899 assert!(a == b);
1900 assert_eq!(a, b);
1901 }
1902 };
1903 }
1904
1905 generate_assert_eq_vec_and_prim! { assert_eq_vec_and_slice <B>(&[B]) }
1906 generate_assert_eq_vec_and_prim! { assert_eq_vec_and_array_3<B>([B; 3]) }
1907
1908 #[test]
1909 fn partialeq_vec_and_prim() {
1910 assert_eq_vec_and_slice(vec![1, 2, 3], &[1, 2, 3]);
1911 assert_eq_vec_and_array_3(vec![1, 2, 3], [1, 2, 3]);
1912 }
1913
1914 macro_rules! assert_partial_eq_valid {
1915 ($a2:expr, $a3:expr; $b2:expr, $b3: expr) => {
1916 assert!($a2 == $b2);
1917 assert!($a2 != $b3);
1918 assert!($a3 != $b2);
1919 assert!($a3 == $b3);
1920 assert_eq!($a2, $b2);
1921 assert_ne!($a2, $b3);
1922 assert_ne!($a3, $b2);
1923 assert_eq!($a3, $b3);
1924 };
1925 }
1926
1927 #[test]
1928 fn partialeq_vec_full() {
1929 let vec2: Vec<_> = vec![1, 2];
1930 let vec3: Vec<_> = vec![1, 2, 3];
1931 let slice2: &[_] = &[1, 2];
1932 let slice3: &[_] = &[1, 2, 3];
1933 let slicemut2: &[_] = &mut [1, 2];
1934 let slicemut3: &[_] = &mut [1, 2, 3];
1935 let array2: [_; 2] = [1, 2];
1936 let array3: [_; 3] = [1, 2, 3];
1937 let arrayref2: &[_; 2] = &[1, 2];
1938 let arrayref3: &[_; 3] = &[1, 2, 3];
1939
1940 assert_partial_eq_valid!(vec2,vec3; vec2,vec3);
1941 assert_partial_eq_valid!(vec2,vec3; slice2,slice3);
1942 assert_partial_eq_valid!(vec2,vec3; slicemut2,slicemut3);
1943 assert_partial_eq_valid!(slice2,slice3; vec2,vec3);
1944 assert_partial_eq_valid!(slicemut2,slicemut3; vec2,vec3);
1945 assert_partial_eq_valid!(vec2,vec3; array2,array3);
1946 assert_partial_eq_valid!(vec2,vec3; arrayref2,arrayref3);
1947 assert_partial_eq_valid!(vec2,vec3; arrayref2[..],arrayref3[..]);
1948 }
1949
1950 #[test]
1951 fn test_vec_cycle() {
1952 #[derive(Debug)]
1953 struct C<'a> {
1954 v: Vec<Cell<Option<&'a C<'a>>>>,
1955 }
1956
1957 impl<'a> C<'a> {
1958 fn new() -> C<'a> {
1959 C { v: Vec::new() }
1960 }
1961 }
1962
1963 let mut c1 = C::new();
1964 let mut c2 = C::new();
1965 let mut c3 = C::new();
1966
1967 // Push
1968 c1.v.push(Cell::new(None));
1969 c1.v.push(Cell::new(None));
1970
1971 c2.v.push(Cell::new(None));
1972 c2.v.push(Cell::new(None));
1973
1974 c3.v.push(Cell::new(None));
1975 c3.v.push(Cell::new(None));
1976
1977 // Set
1978 c1.v[0].set(Some(&c2));
1979 c1.v[1].set(Some(&c3));
1980
1981 c2.v[0].set(Some(&c2));
1982 c2.v[1].set(Some(&c3));
1983
1984 c3.v[0].set(Some(&c1));
1985 c3.v[1].set(Some(&c2));
1986 }
1987
1988 #[test]
1989 fn test_vec_cycle_wrapped() {
1990 struct Refs<'a> {
1991 v: Vec<Cell<Option<&'a C<'a>>>>,
1992 }
1993
1994 struct C<'a> {
1995 refs: Refs<'a>,
1996 }
1997
1998 impl<'a> Refs<'a> {
1999 fn new() -> Refs<'a> {
2000 Refs { v: Vec::new() }
2001 }
2002 }
2003
2004 impl<'a> C<'a> {
2005 fn new() -> C<'a> {
2006 C { refs: Refs::new() }
2007 }
2008 }
2009
2010 let mut c1 = C::new();
2011 let mut c2 = C::new();
2012 let mut c3 = C::new();
2013
2014 c1.refs.v.push(Cell::new(None));
2015 c1.refs.v.push(Cell::new(None));
2016 c2.refs.v.push(Cell::new(None));
2017 c2.refs.v.push(Cell::new(None));
2018 c3.refs.v.push(Cell::new(None));
2019 c3.refs.v.push(Cell::new(None));
2020
2021 c1.refs.v[0].set(Some(&c2));
2022 c1.refs.v[1].set(Some(&c3));
2023 c2.refs.v[0].set(Some(&c2));
2024 c2.refs.v[1].set(Some(&c3));
2025 c3.refs.v[0].set(Some(&c1));
2026 c3.refs.v[1].set(Some(&c2));
2027 }
2028
2029 #[test]
2030 fn test_zero_sized_vec_push() {
2031 const N: usize = 8;
2032
2033 for len in 0..N {
2034 let mut tester = Vec::with_capacity(len);
2035 assert_eq!(tester.len(), 0);
2036 assert!(tester.capacity() >= len);
2037 for _ in 0..len {
2038 tester.push(());
2039 }
2040 assert_eq!(tester.len(), len);
2041 assert_eq!(tester.iter().count(), len);
2042 tester.clear();
2043 }
2044 }
2045
2046 #[test]
2047 fn test_vec_macro_repeat() {
2048 assert_eq!(vec![1; 3], vec![1, 1, 1]);
2049 assert_eq!(vec![1; 2], vec![1, 1]);
2050 assert_eq!(vec![1; 1], vec![1]);
2051 assert_eq!(vec![1; 0], vec![]);
2052
2053 // from_elem syntax (see RFC 832)
2054 let el = Box::new(1);
2055 let n = 3;
2056 assert_eq!(vec![el; n], vec![Box::new(1), Box::new(1), Box::new(1)]);
2057 }
2058
2059 #[test]
2060 fn test_vec_swap() {
2061 let mut a: Vec<isize> = vec![0, 1, 2, 3, 4, 5, 6];
2062 a.swap(2, 4);
2063 assert_eq!(a[2], 4);
2064 assert_eq!(a[4], 2);
2065 let mut n = 42;
2066 swap(&mut n, &mut a[0]);
2067 assert_eq!(a[0], 42);
2068 assert_eq!(n, 0);
2069 }
2070
2071 #[test]
2072 fn test_extend_from_within_spec() {
2073 #[derive(Copy)]
2074 struct CopyOnly;
2075
2076 impl Clone for CopyOnly {
2077 fn clone(&self) -> Self {
2078 panic!("extend_from_within must use specialization on copy");
2079 }
2080 }
2081
2082 vec![CopyOnly, CopyOnly].extend_from_within(..);
2083 }
2084
2085 #[test]
2086 fn test_extend_from_within_clone() {
2087 let mut v = vec![String::from("sssss"), String::from("12334567890"), String::from("c")];
2088 v.extend_from_within(1..);
2089
2090 assert_eq!(v, ["sssss", "12334567890", "c", "12334567890", "c"]);
2091 }
2092
2093 #[test]
2094 fn test_extend_from_within_complete_rande() {
2095 let mut v = vec![0, 1, 2, 3];
2096 v.extend_from_within(..);
2097
2098 assert_eq!(v, [0, 1, 2, 3, 0, 1, 2, 3]);
2099 }
2100
2101 #[test]
2102 fn test_extend_from_within_empty_rande() {
2103 let mut v = vec![0, 1, 2, 3];
2104 v.extend_from_within(1..1);
2105
2106 assert_eq!(v, [0, 1, 2, 3]);
2107 }
2108
2109 #[test]
2110 #[should_panic]
2111 fn test_extend_from_within_out_of_rande() {
2112 let mut v = vec![0, 1];
2113 v.extend_from_within(..3);
2114 }
2115
2116 #[test]
2117 fn test_extend_from_within_zst() {
2118 let mut v = vec![(); 8];
2119 v.extend_from_within(3..7);
2120
2121 assert_eq!(v, [(); 12]);
2122 }
2123
2124 #[test]
2125 fn test_extend_from_within_empty_vec() {
2126 let mut v = Vec::<i32>::new();
2127 v.extend_from_within(..);
2128
2129 assert_eq!(v, []);
2130 }
2131
2132 #[test]
2133 fn test_extend_from_within() {
2134 let mut v = vec![String::from("a"), String::from("b"), String::from("c")];
2135 v.extend_from_within(1..=2);
2136 v.extend_from_within(..=1);
2137
2138 assert_eq!(v, ["a", "b", "c", "b", "c", "a", "b"]);
2139 }
2140
2141 #[test]
2142 fn test_vec_dedup_by() {
2143 let mut vec: Vec<i32> = vec![1, -1, 2, 3, 1, -5, 5, -2, 2];
2144
2145 vec.dedup_by(|a, b| a.abs() == b.abs());
2146
2147 assert_eq!(vec, [1, 2, 3, 1, -5, -2]);
2148 }
2149
2150 #[test]
2151 fn test_vec_dedup_empty() {
2152 let mut vec: Vec<i32> = Vec::new();
2153
2154 vec.dedup();
2155
2156 assert_eq!(vec, []);
2157 }
2158
2159 #[test]
2160 fn test_vec_dedup_one() {
2161 let mut vec = vec![12i32];
2162
2163 vec.dedup();
2164
2165 assert_eq!(vec, [12]);
2166 }
2167
2168 #[test]
2169 fn test_vec_dedup_multiple_ident() {
2170 let mut vec = vec![12, 12, 12, 12, 12, 11, 11, 11, 11, 11, 11];
2171
2172 vec.dedup();
2173
2174 assert_eq!(vec, [12, 11]);
2175 }
2176
2177 #[test]
2178 fn test_vec_dedup_partialeq() {
2179 #[derive(Debug)]
2180 struct Foo(i32, i32);
2181
2182 impl PartialEq for Foo {
2183 fn eq(&self, other: &Foo) -> bool {
2184 self.0 == other.0
2185 }
2186 }
2187
2188 let mut vec = vec![Foo(0, 1), Foo(0, 5), Foo(1, 7), Foo(1, 9)];
2189
2190 vec.dedup();
2191 assert_eq!(vec, [Foo(0, 1), Foo(1, 7)]);
2192 }
2193
2194 #[test]
2195 fn test_vec_dedup() {
2196 let mut vec: Vec<bool> = Vec::with_capacity(8);
2197 let mut template = vec.clone();
2198
2199 for x in 0u8..255u8 {
2200 vec.clear();
2201 template.clear();
2202
2203 let iter = (0..8).map(move |bit| (x >> bit) & 1 == 1);
2204 vec.extend(iter);
2205 template.extend_from_slice(&vec);
2206
2207 let (dedup, _) = template.partition_dedup();
2208 vec.dedup();
2209
2210 assert_eq!(vec, dedup);
2211 }
2212 }
2213
2214 #[test]
2215 fn test_vec_dedup_panicking() {
2216 #[derive(Debug)]
2217 struct Panic {
2218 drop_counter: &'static AtomicU32,
2219 value: bool,
2220 index: usize,
2221 }
2222
2223 impl PartialEq for Panic {
2224 fn eq(&self, other: &Self) -> bool {
2225 self.value == other.value
2226 }
2227 }
2228
2229 impl Drop for Panic {
2230 fn drop(&mut self) {
2231 let x = self.drop_counter.fetch_add(1, Ordering::SeqCst);
2232 assert!(x != 4);
2233 }
2234 }
2235
2236 static DROP_COUNTER: AtomicU32 = AtomicU32::new(0);
2237 let expected = [
2238 Panic { drop_counter: &DROP_COUNTER, value: false, index: 0 },
2239 Panic { drop_counter: &DROP_COUNTER, value: false, index: 5 },
2240 Panic { drop_counter: &DROP_COUNTER, value: true, index: 6 },
2241 Panic { drop_counter: &DROP_COUNTER, value: true, index: 7 },
2242 ];
2243 let mut vec = vec![
2244 Panic { drop_counter: &DROP_COUNTER, value: false, index: 0 },
2245 // these elements get deduplicated
2246 Panic { drop_counter: &DROP_COUNTER, value: false, index: 1 },
2247 Panic { drop_counter: &DROP_COUNTER, value: false, index: 2 },
2248 Panic { drop_counter: &DROP_COUNTER, value: false, index: 3 },
2249 Panic { drop_counter: &DROP_COUNTER, value: false, index: 4 },
2250 // here it panics
2251 Panic { drop_counter: &DROP_COUNTER, value: false, index: 5 },
2252 Panic { drop_counter: &DROP_COUNTER, value: true, index: 6 },
2253 Panic { drop_counter: &DROP_COUNTER, value: true, index: 7 },
2254 ];
2255
2256 let _ = std::panic::catch_unwind(std::panic::AssertUnwindSafe(|| {
2257 vec.dedup();
2258 }));
2259
2260 let ok = vec.iter().zip(expected.iter()).all(|(x, y)| x.index == y.index);
2261
2262 if !ok {
2263 panic!("expected: {:?}\ngot: {:?}\n", expected, vec);
2264 }
2265 }
2266
2267 // Regression test for issue #82533
2268 #[test]
2269 fn test_extend_from_within_panicing_clone() {
2270 struct Panic<'dc> {
2271 drop_count: &'dc AtomicU32,
2272 aaaaa: bool,
2273 }
2274
2275 impl Clone for Panic<'_> {
2276 fn clone(&self) -> Self {
2277 if self.aaaaa {
2278 panic!("panic! at the clone");
2279 }
2280
2281 Self { ..*self }
2282 }
2283 }
2284
2285 impl Drop for Panic<'_> {
2286 fn drop(&mut self) {
2287 self.drop_count.fetch_add(1, Ordering::SeqCst);
2288 }
2289 }
2290
2291 let count = core::sync::atomic::AtomicU32::new(0);
2292 let mut vec = vec![
2293 Panic { drop_count: &count, aaaaa: false },
2294 Panic { drop_count: &count, aaaaa: true },
2295 Panic { drop_count: &count, aaaaa: false },
2296 ];
2297
2298 // This should clone&append one Panic{..} at the end, and then panic while
2299 // cloning second Panic{..}. This means that `Panic::drop` should be called
2300 // 4 times (3 for items already in vector, 1 for just appended).
2301 //
2302 // Previously just appended item was leaked, making drop_count = 3, instead of 4.
2303 std::panic::catch_unwind(move || vec.extend_from_within(..)).unwrap_err();
2304
2305 assert_eq!(count.load(Ordering::SeqCst), 4);
2306 }