1 //===----------------------------------------------------------------------===//
2 //
3 // The LLVM Compiler Infrastructure
4 //
5 // This file is dual licensed under the MIT and the University of Illinois Open
6 // Source Licenses. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9
10 // <string>
11
12 // template<class InputIterator>
13 // basic_string& assign(InputIterator first, InputIterator last);
14
15 #include <string>
16 #include <cassert>
17
18 #include "test_macros.h"
19 #include "test_iterators.h"
20 #include "min_allocator.h"
21
22 template <class S, class It>
23 void
test(S s,It first,It last,S expected)24 test(S s, It first, It last, S expected)
25 {
26 s.assign(first, last);
27 LIBCPP_ASSERT(s.__invariants());
28 assert(s == expected);
29 }
30
31 #ifndef TEST_HAS_NO_EXCEPTIONS
32 template <class S, class It>
33 void
test_exceptions(S s,It first,It last)34 test_exceptions(S s, It first, It last)
35 {
36 S aCopy = s;
37 try {
38 s.assign(first, last);
39 assert(false);
40 }
41 catch (...) {}
42 LIBCPP_ASSERT(s.__invariants());
43 assert(s == aCopy);
44 }
45 #endif
46
main()47 int main()
48 {
49 {
50 typedef std::string S;
51 const char* s = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz";
52 test(S(), s, s, S());
53 test(S(), s, s+1, S("A"));
54 test(S(), s, s+10, S("ABCDEFGHIJ"));
55 test(S(), s, s+52, S("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"));
56
57 test(S("12345"), s, s, S());
58 test(S("12345"), s, s+1, S("A"));
59 test(S("12345"), s, s+10, S("ABCDEFGHIJ"));
60 test(S("12345"), s, s+52, S("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"));
61
62 test(S("1234567890"), s, s, S());
63 test(S("1234567890"), s, s+1, S("A"));
64 test(S("1234567890"), s, s+10, S("ABCDEFGHIJ"));
65 test(S("1234567890"), s, s+52, S("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"));
66
67 test(S("12345678901234567890"), s, s, S());
68 test(S("12345678901234567890"), s, s+1, S("A"));
69 test(S("12345678901234567890"), s, s+10, S("ABCDEFGHIJ"));
70 test(S("12345678901234567890"), s, s+52,
71 S("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"));
72
73 test(S(), input_iterator<const char*>(s), input_iterator<const char*>(s), S());
74 test(S(), input_iterator<const char*>(s), input_iterator<const char*>(s+1), S("A"));
75 test(S(), input_iterator<const char*>(s), input_iterator<const char*>(s+10),
76 S("ABCDEFGHIJ"));
77 test(S(), input_iterator<const char*>(s), input_iterator<const char*>(s+52),
78 S("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"));
79
80 test(S("12345"), input_iterator<const char*>(s), input_iterator<const char*>(s),
81 S());
82 test(S("12345"), input_iterator<const char*>(s), input_iterator<const char*>(s+1),
83 S("A"));
84 test(S("12345"), input_iterator<const char*>(s), input_iterator<const char*>(s+10),
85 S("ABCDEFGHIJ"));
86 test(S("12345"), input_iterator<const char*>(s), input_iterator<const char*>(s+52),
87 S("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"));
88
89 test(S("1234567890"), input_iterator<const char*>(s), input_iterator<const char*>(s),
90 S());
91 test(S("1234567890"), input_iterator<const char*>(s), input_iterator<const char*>(s+1),
92 S("A"));
93 test(S("1234567890"), input_iterator<const char*>(s), input_iterator<const char*>(s+10),
94 S("ABCDEFGHIJ"));
95 test(S("1234567890"), input_iterator<const char*>(s), input_iterator<const char*>(s+52),
96 S("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"));
97
98 test(S("12345678901234567890"), input_iterator<const char*>(s), input_iterator<const char*>(s),
99 S());
100 test(S("12345678901234567890"), input_iterator<const char*>(s), input_iterator<const char*>(s+1),
101 S("A"));
102 test(S("12345678901234567890"), input_iterator<const char*>(s), input_iterator<const char*>(s+10),
103 S("ABCDEFGHIJ"));
104 test(S("12345678901234567890"), input_iterator<const char*>(s), input_iterator<const char*>(s+52),
105 S("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"));
106 }
107 #if TEST_STD_VER >= 11
108 {
109 typedef std::basic_string<char, std::char_traits<char>, min_allocator<char>> S;
110 const char* s = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz";
111 test(S(), s, s, S());
112 test(S(), s, s+1, S("A"));
113 test(S(), s, s+10, S("ABCDEFGHIJ"));
114 test(S(), s, s+52, S("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"));
115
116 test(S("12345"), s, s, S());
117 test(S("12345"), s, s+1, S("A"));
118 test(S("12345"), s, s+10, S("ABCDEFGHIJ"));
119 test(S("12345"), s, s+52, S("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"));
120
121 test(S("1234567890"), s, s, S());
122 test(S("1234567890"), s, s+1, S("A"));
123 test(S("1234567890"), s, s+10, S("ABCDEFGHIJ"));
124 test(S("1234567890"), s, s+52, S("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"));
125
126 test(S("12345678901234567890"), s, s, S());
127 test(S("12345678901234567890"), s, s+1, S("A"));
128 test(S("12345678901234567890"), s, s+10, S("ABCDEFGHIJ"));
129 test(S("12345678901234567890"), s, s+52,
130 S("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"));
131
132 test(S(), input_iterator<const char*>(s), input_iterator<const char*>(s), S());
133 test(S(), input_iterator<const char*>(s), input_iterator<const char*>(s+1), S("A"));
134 test(S(), input_iterator<const char*>(s), input_iterator<const char*>(s+10),
135 S("ABCDEFGHIJ"));
136 test(S(), input_iterator<const char*>(s), input_iterator<const char*>(s+52),
137 S("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"));
138
139 test(S("12345"), input_iterator<const char*>(s), input_iterator<const char*>(s),
140 S());
141 test(S("12345"), input_iterator<const char*>(s), input_iterator<const char*>(s+1),
142 S("A"));
143 test(S("12345"), input_iterator<const char*>(s), input_iterator<const char*>(s+10),
144 S("ABCDEFGHIJ"));
145 test(S("12345"), input_iterator<const char*>(s), input_iterator<const char*>(s+52),
146 S("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"));
147
148 test(S("1234567890"), input_iterator<const char*>(s), input_iterator<const char*>(s),
149 S());
150 test(S("1234567890"), input_iterator<const char*>(s), input_iterator<const char*>(s+1),
151 S("A"));
152 test(S("1234567890"), input_iterator<const char*>(s), input_iterator<const char*>(s+10),
153 S("ABCDEFGHIJ"));
154 test(S("1234567890"), input_iterator<const char*>(s), input_iterator<const char*>(s+52),
155 S("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"));
156
157 test(S("12345678901234567890"), input_iterator<const char*>(s), input_iterator<const char*>(s),
158 S());
159 test(S("12345678901234567890"), input_iterator<const char*>(s), input_iterator<const char*>(s+1),
160 S("A"));
161 test(S("12345678901234567890"), input_iterator<const char*>(s), input_iterator<const char*>(s+10),
162 S("ABCDEFGHIJ"));
163 test(S("12345678901234567890"), input_iterator<const char*>(s), input_iterator<const char*>(s+52),
164 S("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"));
165 }
166 #endif
167 #ifndef TEST_HAS_NO_EXCEPTIONS
168 { // test iterator operations that throw
169 typedef std::string S;
170 typedef ThrowingIterator<char> TIter;
171 typedef input_iterator<TIter> IIter;
172 const char* s = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz";
173 test_exceptions(S(), IIter(TIter(s, s+10, 4, TIter::TAIncrement)), IIter());
174 test_exceptions(S(), IIter(TIter(s, s+10, 5, TIter::TADereference)), IIter());
175 test_exceptions(S(), IIter(TIter(s, s+10, 6, TIter::TAComparison)), IIter());
176
177 test_exceptions(S(), TIter(s, s+10, 4, TIter::TAIncrement), TIter());
178 test_exceptions(S(), TIter(s, s+10, 5, TIter::TADereference), TIter());
179 test_exceptions(S(), TIter(s, s+10, 6, TIter::TAComparison), TIter());
180 }
181 #endif
182
183 { // test assigning to self
184 typedef std::string S;
185 S s_short = "123/";
186 S s_long = "Lorem ipsum dolor sit amet, consectetur/";
187
188 s_short.assign(s_short.begin(), s_short.end());
189 assert(s_short == "123/");
190 s_short.assign(s_short.begin() + 2, s_short.end());
191 assert(s_short == "3/");
192
193 s_long.assign(s_long.begin(), s_long.end());
194 assert(s_long == "Lorem ipsum dolor sit amet, consectetur/");
195
196 s_long.assign(s_long.begin() + 30, s_long.end());
197 assert(s_long == "nsectetur/");
198 }
199
200 { // test assigning a different type
201 typedef std::string S;
202 const uint8_t p[] = "ABCD";
203
204 S s;
205 s.assign(p, p + 4);
206 assert(s == "ABCD");
207 }
208 }
209