xref: /llvm-project/libcxx/test/std/containers/sequences/vector/vector.modifiers/insert_range.pass.cpp (revision 733a98db4a264f474564cc2064b8862dedd8458f)
1 //===----------------------------------------------------------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 
9 // UNSUPPORTED: c++03, c++11, c++14, c++17, c++20
10 // ADDITIONAL_COMPILE_FLAGS(has-fconstexpr-steps): -fconstexpr-steps=2000000
11 
12 // template<container-compatible-range<T> R>
13 //   constexpr iterator insert_range(const_iterator position, R&& rg); // C++23
14 
15 #include <sstream>
16 #include <vector>
17 
18 #include "../../insert_range_sequence_containers.h"
19 #include "asan_testing.h"
20 #include "test_macros.h"
21 
22 // Tested cases:
23 // - different kinds of insertions (inserting an {empty/one-element/mid-sized/long range} into an
24 //   {empty/one-element/full} container at the {beginning/middle/end});
25 // - inserting move-only elements;
26 // - an exception is thrown when copying the elements or when allocating new elements.
27 
28 constexpr bool test() {
29   for_all_iterators_and_allocators<int, const int*>([]<class Iter, class Sent, class Alloc>() {
30     test_sequence_insert_range<std::vector<int, Alloc>, Iter, Sent>([]([[maybe_unused]] auto&& c) {
31       LIBCPP_ASSERT(c.__invariants());
32       LIBCPP_ASSERT(is_contiguous_container_asan_correct(c));
33     });
34   });
35   test_sequence_insert_range_move_only<std::vector>();
36 
37   { // Vector may or may not need to reallocate because of the insertion -- make sure to test both cases.
38     { // Ensure reallocation happens.
39       int in[] = {-1, -2, -3, -4, -5, -6, -7, -8, -9, -10};
40       std::vector<int> v = {1, 2, 3, 4, 5, 6, 7, 8};
41       v.shrink_to_fit();
42       assert(v.capacity() < v.size() + std::ranges::size(in));
43 
44       v.insert_range(v.end(), in);
45       assert(std::ranges::equal(v, std::array{1, 2, 3, 4, 5, 6, 7, 8, -1, -2, -3, -4, -5, -6, -7, -8, -9, -10}));
46     }
47 
48     { // Ensure no reallocation happens.
49       int in[] = {-1, -2, -3, -4, -5, -6, -7, -8, -9, -10};
50       std::vector<int> v = {1, 2, 3, 4, 5, 6, 7, 8};
51       v.reserve(v.size() + std::ranges::size(in));
52       assert(v.capacity() >= v.size() + std::ranges::size(in));
53 
54       v.insert_range(v.end(), in);
55       assert(std::ranges::equal(v, std::array{1, 2, 3, 4, 5, 6, 7, 8, -1, -2, -3, -4, -5, -6, -7, -8, -9, -10}));
56     }
57 
58     { // Ensure input-only sized ranges are accepted.
59       using input_iter = cpp20_input_iterator<const int*>;
60       const int in[]{1, 2, 3, 4};
61       std::vector<int> v{-5, -6};
62       v.insert_range(v.begin(), std::views::counted(input_iter{std::ranges::begin(in)}, std::ranges::ssize(in)));
63       assert(std::ranges::equal(v, std::vector<int>{1, 2, 3, 4, -5, -6}));
64     }
65   }
66 
67   return true;
68 }
69 
70 #ifndef TEST_HAS_NO_LOCALIZATION
71 void test_counted_istream_view() {
72   std::istringstream is{"1 2 3 4"};
73   auto vals = std::views::istream<int>(is);
74   std::vector<int> v;
75   v.insert_range(v.end(), std::views::counted(vals.begin(), 3));
76   assert(v == (std::vector{1, 2, 3}));
77 }
78 #endif
79 
80 int main(int, char**) {
81   test();
82   static_assert(test());
83 
84   static_assert(test_constraints_insert_range<std::vector, int, double>());
85 
86   test_insert_range_exception_safety_throwing_copy<std::vector>();
87   test_insert_range_exception_safety_throwing_allocator<std::vector, int>();
88 
89 #ifndef TEST_HAS_NO_LOCALIZATION
90   test_counted_istream_view();
91 #endif
92 
93   return 0;
94 }
95