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 // <string>
10 
11 // template<class InputIterator>
12 //   basic_string& assign(InputIterator first, InputIterator last); // constexpr since C++20
13 
14 #include <string>
15 #include <cassert>
16 
17 #include "test_macros.h"
18 #include "test_iterators.h"
19 #include "min_allocator.h"
20 
21 template <class S, class It>
22 TEST_CONSTEXPR_CXX20 void
test(S s,It first,It last,S expected)23 test(S s, It first, It last, S expected)
24 {
25     s.assign(first, last);
26     LIBCPP_ASSERT(s.__invariants());
27     assert(s == expected);
28 }
29 
30 #ifndef TEST_HAS_NO_EXCEPTIONS
operator charWidget31 struct Widget { operator char() const { throw 42; } };
32 
33 template <class S, class It>
34 void
test_exceptions(S s,It first,It last)35 test_exceptions(S s, It first, It last)
36 {
37     S original = s;
38     typename S::iterator begin = s.begin();
39     typename S::iterator end = s.end();
40 
41     try {
42         s.assign(first, last);
43         assert(false);
44     } catch (...) {}
45 
46     // Part of "no effects" is that iterators and pointers
47     // into the string must not have been invalidated.
48     LIBCPP_ASSERT(s.__invariants());
49     assert(s == original);
50     assert(s.begin() == begin);
51     assert(s.end() == end);
52 }
53 #endif
54 
test()55 TEST_CONSTEXPR_CXX20 bool test() {
56   {
57     typedef std::string S;
58     const char* s = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz";
59     test(S(), s, s, S());
60     test(S(), s, s+1, S("A"));
61     test(S(), s, s+10, S("ABCDEFGHIJ"));
62     test(S(), s, s+52, S("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"));
63 
64     test(S("12345"), s, s, S());
65     test(S("12345"), s, s+1, S("A"));
66     test(S("12345"), s, s+10, S("ABCDEFGHIJ"));
67     test(S("12345"), s, s+52, S("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"));
68 
69     test(S("1234567890"), s, s, S());
70     test(S("1234567890"), s, s+1, S("A"));
71     test(S("1234567890"), s, s+10, S("ABCDEFGHIJ"));
72     test(S("1234567890"), s, s+52, S("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"));
73 
74     test(S("12345678901234567890"), s, s, S());
75     test(S("12345678901234567890"), s, s+1, S("A"));
76     test(S("12345678901234567890"), s, s+10, S("ABCDEFGHIJ"));
77     test(S("12345678901234567890"), s, s+52,
78          S("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"));
79 
80     test(S(), cpp17_input_iterator<const char*>(s), cpp17_input_iterator<const char*>(s), S());
81     test(S(), cpp17_input_iterator<const char*>(s), cpp17_input_iterator<const char*>(s+1), S("A"));
82     test(S(), cpp17_input_iterator<const char*>(s), cpp17_input_iterator<const char*>(s+10),
83          S("ABCDEFGHIJ"));
84     test(S(), cpp17_input_iterator<const char*>(s), cpp17_input_iterator<const char*>(s+52),
85          S("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"));
86 
87     test(S("12345"), cpp17_input_iterator<const char*>(s), cpp17_input_iterator<const char*>(s),
88          S());
89     test(S("12345"), cpp17_input_iterator<const char*>(s), cpp17_input_iterator<const char*>(s+1),
90          S("A"));
91     test(S("12345"), cpp17_input_iterator<const char*>(s), cpp17_input_iterator<const char*>(s+10),
92          S("ABCDEFGHIJ"));
93     test(S("12345"), cpp17_input_iterator<const char*>(s), cpp17_input_iterator<const char*>(s+52),
94          S("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"));
95 
96     test(S("1234567890"), cpp17_input_iterator<const char*>(s), cpp17_input_iterator<const char*>(s),
97          S());
98     test(S("1234567890"), cpp17_input_iterator<const char*>(s), cpp17_input_iterator<const char*>(s+1),
99          S("A"));
100     test(S("1234567890"), cpp17_input_iterator<const char*>(s), cpp17_input_iterator<const char*>(s+10),
101          S("ABCDEFGHIJ"));
102     test(S("1234567890"), cpp17_input_iterator<const char*>(s), cpp17_input_iterator<const char*>(s+52),
103          S("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"));
104 
105     test(S("12345678901234567890"), cpp17_input_iterator<const char*>(s), cpp17_input_iterator<const char*>(s),
106          S());
107     test(S("12345678901234567890"), cpp17_input_iterator<const char*>(s), cpp17_input_iterator<const char*>(s+1),
108          S("A"));
109     test(S("12345678901234567890"), cpp17_input_iterator<const char*>(s), cpp17_input_iterator<const char*>(s+10),
110          S("ABCDEFGHIJ"));
111     test(S("12345678901234567890"), cpp17_input_iterator<const char*>(s), cpp17_input_iterator<const char*>(s+52),
112          S("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"));
113   }
114 #if TEST_STD_VER >= 11
115   {
116     typedef std::basic_string<char, std::char_traits<char>, min_allocator<char>> S;
117     const char* s = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz";
118     test(S(), s, s, S());
119     test(S(), s, s+1, S("A"));
120     test(S(), s, s+10, S("ABCDEFGHIJ"));
121     test(S(), s, s+52, S("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"));
122 
123     test(S("12345"), s, s, S());
124     test(S("12345"), s, s+1, S("A"));
125     test(S("12345"), s, s+10, S("ABCDEFGHIJ"));
126     test(S("12345"), s, s+52, S("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"));
127 
128     test(S("1234567890"), s, s, S());
129     test(S("1234567890"), s, s+1, S("A"));
130     test(S("1234567890"), s, s+10, S("ABCDEFGHIJ"));
131     test(S("1234567890"), s, s+52, S("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"));
132 
133     test(S("12345678901234567890"), s, s, S());
134     test(S("12345678901234567890"), s, s+1, S("A"));
135     test(S("12345678901234567890"), s, s+10, S("ABCDEFGHIJ"));
136     test(S("12345678901234567890"), s, s+52,
137          S("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"));
138 
139     test(S(), cpp17_input_iterator<const char*>(s), cpp17_input_iterator<const char*>(s), S());
140     test(S(), cpp17_input_iterator<const char*>(s), cpp17_input_iterator<const char*>(s+1), S("A"));
141     test(S(), cpp17_input_iterator<const char*>(s), cpp17_input_iterator<const char*>(s+10),
142          S("ABCDEFGHIJ"));
143     test(S(), cpp17_input_iterator<const char*>(s), cpp17_input_iterator<const char*>(s+52),
144          S("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"));
145 
146     test(S("12345"), cpp17_input_iterator<const char*>(s), cpp17_input_iterator<const char*>(s),
147          S());
148     test(S("12345"), cpp17_input_iterator<const char*>(s), cpp17_input_iterator<const char*>(s+1),
149          S("A"));
150     test(S("12345"), cpp17_input_iterator<const char*>(s), cpp17_input_iterator<const char*>(s+10),
151          S("ABCDEFGHIJ"));
152     test(S("12345"), cpp17_input_iterator<const char*>(s), cpp17_input_iterator<const char*>(s+52),
153          S("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"));
154 
155     test(S("1234567890"), cpp17_input_iterator<const char*>(s), cpp17_input_iterator<const char*>(s),
156          S());
157     test(S("1234567890"), cpp17_input_iterator<const char*>(s), cpp17_input_iterator<const char*>(s+1),
158          S("A"));
159     test(S("1234567890"), cpp17_input_iterator<const char*>(s), cpp17_input_iterator<const char*>(s+10),
160          S("ABCDEFGHIJ"));
161     test(S("1234567890"), cpp17_input_iterator<const char*>(s), cpp17_input_iterator<const char*>(s+52),
162          S("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"));
163 
164     test(S("12345678901234567890"), cpp17_input_iterator<const char*>(s), cpp17_input_iterator<const char*>(s),
165          S());
166     test(S("12345678901234567890"), cpp17_input_iterator<const char*>(s), cpp17_input_iterator<const char*>(s+1),
167          S("A"));
168     test(S("12345678901234567890"), cpp17_input_iterator<const char*>(s), cpp17_input_iterator<const char*>(s+10),
169          S("ABCDEFGHIJ"));
170     test(S("12345678901234567890"), cpp17_input_iterator<const char*>(s), cpp17_input_iterator<const char*>(s+52),
171          S("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"));
172   }
173 #endif
174 #ifndef TEST_HAS_NO_EXCEPTIONS
175   if (!TEST_IS_CONSTANT_EVALUATED) { // test iterator operations that throw
176     typedef std::string S;
177     typedef ThrowingIterator<char> TIter;
178     typedef cpp17_input_iterator<TIter> IIter;
179     const char* s = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz";
180     test_exceptions(S(), IIter(TIter(s, s+10, 4, TIter::TAIncrement)), IIter(TIter()));
181     test_exceptions(S(), IIter(TIter(s, s+10, 5, TIter::TADereference)), IIter(TIter()));
182     test_exceptions(S(), IIter(TIter(s, s+10, 6, TIter::TAComparison)), IIter(TIter()));
183 
184     test_exceptions(S(), TIter(s, s+10, 4, TIter::TAIncrement), TIter());
185     test_exceptions(S(), TIter(s, s+10, 5, TIter::TADereference), TIter());
186     test_exceptions(S(), TIter(s, s+10, 6, TIter::TAComparison), TIter());
187 
188     Widget w[100];
189     test_exceptions(S(), w, w+100);
190   }
191 #endif
192 
193   { // test assigning to self
194     typedef std::string S;
195     S s_short = "123/";
196     S s_long  = "Lorem ipsum dolor sit amet, consectetur/";
197 
198     s_short.assign(s_short.begin(), s_short.end());
199     assert(s_short == "123/");
200     s_short.assign(s_short.begin() + 2, s_short.end());
201     assert(s_short == "3/");
202 
203     s_long.assign(s_long.begin(), s_long.end());
204     assert(s_long == "Lorem ipsum dolor sit amet, consectetur/");
205 
206     s_long.assign(s_long.begin() + 30, s_long.end());
207     assert(s_long == "nsectetur/");
208   }
209 
210   { // test assigning a different type
211     typedef std::string S;
212     const uint8_t p[] = "ABCD";
213 
214     S s;
215     s.assign(p, p + 4);
216     assert(s == "ABCD");
217   }
218 
219   { // regression-test assigning to self in sneaky ways
220     std::string sneaky = "hello";
221     sneaky.resize(sneaky.capacity(), 'x');
222     std::string expected = sneaky + std::string(1, '\0');
223     test(sneaky, sneaky.data(), sneaky.data() + sneaky.size() + 1, expected);
224   }
225 
226   return true;
227 }
228 
main(int,char **)229 int main(int, char**)
230 {
231   test();
232 #if TEST_STD_VER > 17
233   static_assert(test());
234 #endif
235 
236     return 0;
237 }
238