Line data Source code
1 : // Copyright (c) 2015 The Bitcoin Core developers
2 : // Distributed under the MIT software license, see the accompanying
3 : // file COPYING or http://www.opensource.org/licenses/mit-license.php.
4 :
5 : #include <vector>
6 : #include "prevector.h"
7 : #include "random.h"
8 :
9 : #include "serialize.h"
10 : #include "streams.h"
11 :
12 : #include "test/test_pivx.h"
13 :
14 : #include <boost/test/unit_test.hpp>
15 :
16 : BOOST_FIXTURE_TEST_SUITE(PrevectorTests, TestingSetup)
17 :
18 : template<unsigned int N, typename T>
19 : class prevector_tester {
20 : typedef std::vector<T> realtype;
21 : realtype real_vector;
22 : realtype real_vector_alt;
23 :
24 : typedef prevector<N, T> pretype;
25 : pretype pre_vector;
26 : pretype pre_vector_alt;
27 :
28 : typedef typename pretype::size_type Size;
29 : bool passed = true;
30 : FastRandomContext rand_cache;
31 : uint256 rand_seed;
32 :
33 : template <typename A, typename B>
34 11867600 : void local_check_equal(A a, B b)
35 : {
36 11867600 : local_check(a == b);
37 : }
38 26292400 : void local_check(bool b)
39 : {
40 5588610 : passed &= b;
41 : }
42 224512 : void test() {
43 224512 : const pretype& const_pre_vector = pre_vector;
44 435264 : local_check_equal(real_vector.size(), pre_vector.size());
45 224512 : local_check_equal(real_vector.empty(), pre_vector.empty());
46 3018820 : for (Size s = 0; s < real_vector.size(); s++) {
47 2794300 : local_check(real_vector[s] == pre_vector[s]);
48 5588610 : local_check(&(pre_vector[s]) == &(pre_vector.begin()[s]));
49 2794300 : local_check(&(pre_vector[s]) == &*(pre_vector.begin() + s));
50 5588610 : local_check(&(pre_vector[s]) == &*((pre_vector.end() + s) - real_vector.size()));
51 : }
52 : // local_check(realtype(pre_vector) == real_vector);
53 224512 : local_check(pretype(real_vector.begin(), real_vector.end()) == pre_vector);
54 673536 : local_check(pretype(pre_vector.begin(), pre_vector.end()) == pre_vector);
55 224512 : size_t pos = 0;
56 :
57 3467840 : for (const T& v : pre_vector) {
58 2794300 : local_check(v == real_vector[pos++]);
59 : }
60 224512 : pos = 0;
61 3018820 : for (const T& v : const_pre_vector) {
62 2794300 : local_check(v == real_vector[pos++]);
63 : }
64 :
65 449024 : CDataStream ss1(SER_DISK, 0);
66 224512 : CDataStream ss2(SER_DISK, 0);
67 224512 : ss1 << real_vector;
68 224512 : ss2 << pre_vector;
69 224512 : local_check_equal(ss1.size(), ss2.size());
70 11626200 : for (Size s = 0; s < ss1.size(); s++) {
71 11401700 : local_check_equal(ss1[s], ss2[s]);
72 : }
73 224512 : }
74 :
75 : public:
76 8448 : void resize(Size s) {
77 8448 : real_vector.resize(s);
78 8448 : local_check_equal(real_vector.size(), s);
79 8448 : pre_vector.resize(s);
80 8448 : local_check_equal(pre_vector.size(), s);
81 8448 : test();
82 8448 : }
83 :
84 2112 : void reserve(Size s) {
85 2112 : real_vector.reserve(s);
86 2112 : local_check(real_vector.capacity() >= s);
87 2112 : pre_vector.reserve(s);
88 2112 : local_check(pre_vector.capacity() >= s);
89 2112 : test();
90 2112 : }
91 :
92 34880 : void insert(Size position, const T& value) {
93 34880 : real_vector.insert(real_vector.begin() + position, value);
94 69760 : pre_vector.insert(pre_vector.begin() + position, value);
95 34880 : test();
96 34880 : }
97 :
98 8704 : void insert(Size position, Size count, const T& value) {
99 8704 : real_vector.insert(real_vector.begin() + position, count, value);
100 17408 : pre_vector.insert(pre_vector.begin() + position, count, value);
101 8704 : test();
102 8704 : }
103 :
104 : template<typename I>
105 1984 : void insert_range(Size position, I first, I last) {
106 1984 : real_vector.insert(real_vector.begin() + position, first, last);
107 3968 : pre_vector.insert(pre_vector.begin() + position, first, last);
108 1984 : test();
109 1984 : }
110 :
111 15040 : void erase(Size position) {
112 15040 : real_vector.erase(real_vector.begin() + position);
113 30080 : pre_vector.erase(pre_vector.begin() + position);
114 15040 : test();
115 15040 : }
116 :
117 10816 : void erase(Size first, Size last) {
118 10816 : real_vector.erase(real_vector.begin() + first, real_vector.begin() + last);
119 21632 : pre_vector.erase(pre_vector.begin() + first, pre_vector.begin() + last);
120 10816 : test();
121 10816 : }
122 :
123 119424 : void update(Size pos, const T& value) {
124 119424 : real_vector[pos] = value;
125 119424 : pre_vector[pos] = value;
126 119424 : test();
127 119424 : }
128 :
129 4160 : void push_back(const T& value) {
130 4160 : real_vector.push_back(value);
131 4160 : pre_vector.push_back(value);
132 4160 : test();
133 4160 : }
134 :
135 3968 : void pop_back() {
136 3968 : real_vector.pop_back();
137 3968 : pre_vector.pop_back();
138 3968 : test();
139 3968 : }
140 :
141 128 : void clear() {
142 256 : real_vector.clear();
143 128 : pre_vector.clear();
144 128 : }
145 :
146 64 : void assign(Size n, const T& value) {
147 128 : real_vector.assign(n, value);
148 64 : pre_vector.assign(n, value);
149 : }
150 :
151 603328 : Size size() {
152 603328 : return real_vector.size();
153 : }
154 :
155 : Size capacity() {
156 : return pre_vector.capacity();
157 : }
158 :
159 1408 : void shrink_to_fit() {
160 1408 : pre_vector.shrink_to_fit();
161 1408 : test();
162 1408 : }
163 :
164 9728 : void swap() {
165 9728 : real_vector.swap(real_vector_alt);
166 9728 : pre_vector.swap(pre_vector_alt);
167 9728 : test();
168 9728 : }
169 :
170 1664 : void move() {
171 1664 : real_vector = std::move(real_vector_alt);
172 1664 : real_vector_alt.clear();
173 1664 : pre_vector = std::move(pre_vector_alt);
174 1664 : pre_vector_alt.clear();
175 1664 : }
176 :
177 4608 : void copy() {
178 4608 : real_vector = real_vector_alt;
179 4608 : pre_vector = pre_vector_alt;
180 4608 : }
181 :
182 3840 : void resize_uninitialized(realtype values) {
183 3840 : size_t r = values.size();
184 3840 : size_t s = real_vector.size() / 2;
185 3840 : if (real_vector.capacity() < s + r) {
186 832 : real_vector.reserve(s + r);
187 : }
188 3840 : real_vector.resize(s);
189 3840 : pre_vector.resize_uninitialized(s);
190 34432 : for (auto v : values) {
191 30592 : real_vector.push_back(v);
192 : }
193 3840 : auto p = pre_vector.size();
194 3840 : pre_vector.resize_uninitialized(p + r);
195 34432 : for (auto v : values) {
196 30592 : pre_vector[p] = v;
197 30592 : ++p;
198 : }
199 3840 : test();
200 3840 : }
201 :
202 64 : ~prevector_tester() {
203 128 : BOOST_CHECK_MESSAGE(passed, "insecure_rand: " + rand_seed.ToString());
204 256 : }
205 128 : prevector_tester() {
206 64 : SeedInsecureRand();
207 64 : rand_seed = InsecureRand256();
208 64 : rand_cache = FastRandomContext(rand_seed);
209 64 : }
210 : };
211 :
212 2 : BOOST_AUTO_TEST_CASE(PrevectorTestInt)
213 : {
214 65 : for (int j = 0; j < 64; j++) {
215 128 : prevector_tester<8, int> test;
216 131136 : for (int i = 0; i < 2048; i++) {
217 131072 : int r = InsecureRand32();
218 131072 : if ((r % 4) == 0) {
219 34880 : test.insert(InsecureRand32() % (test.size() + 1), InsecureRand32());
220 : }
221 131072 : if (test.size() > 0 && ((r >> 2) % 4) == 1) {
222 15040 : test.erase(InsecureRand32() % test.size());
223 : }
224 131072 : if (((r >> 4) % 8) == 2) {
225 16768 : int new_size = std::max<int>(0, std::min<int>(30, test.size() + (InsecureRand32() % 5) - 2));
226 8448 : test.resize(new_size);
227 : }
228 131072 : if (((r >> 7) % 8) == 3) {
229 8704 : test.insert(InsecureRand32() % (test.size() + 1), 1 + (InsecureRand32() % 2), InsecureRand32());
230 : }
231 131072 : if (((r >> 10) % 8) == 4) {
232 8192 : int del = std::min<int>(test.size(), 1 + (InsecureRand32() % 2));
233 8192 : int beg = InsecureRand32() % (test.size() + 1 - del);
234 8192 : test.erase(beg, beg + del);
235 : }
236 131072 : if (((r >> 13) % 16) == 5) {
237 4160 : test.push_back(InsecureRand32());
238 : }
239 131072 : if (test.size() > 0 && ((r >> 17) % 16) == 6) {
240 3968 : test.pop_back();
241 : }
242 131072 : if (((r >> 21) % 32) == 7) {
243 1984 : int values[4];
244 1984 : int num = 1 + (InsecureRand32() % 4);
245 6400 : for (int i = 0; i < num; i++) {
246 4416 : values[i] = InsecureRand32();
247 : }
248 1984 : test.insert_range(InsecureRand32() % (test.size() + 1), values, values + num);
249 : }
250 131072 : if (((r >> 26) % 32) == 8) {
251 2624 : int del = std::min<int>(test.size(), 1 + (InsecureRand32() % 4));
252 2624 : int beg = InsecureRand32() % (test.size() + 1 - del);
253 2624 : test.erase(beg, beg + del);
254 : }
255 131072 : r = InsecureRand32();
256 131072 : if (r % 32 == 9) {
257 2112 : test.reserve(InsecureRand32() % 32);
258 : }
259 131072 : if ((r >> 5) % 64 == 10) {
260 1408 : test.shrink_to_fit();
261 : }
262 131072 : if (test.size() > 0) {
263 119424 : test.update(InsecureRand32() % test.size(), InsecureRand32());
264 : }
265 131072 : if (((r >> 11) % 1024) == 11) {
266 128 : test.clear();
267 : }
268 131072 : if (((r >> 21) % 512) == 12) {
269 128 : test.assign(InsecureRand32() % 32, InsecureRand32());
270 : }
271 131072 : if (((r >> 15) % 8) == 3) {
272 9728 : test.swap();
273 : }
274 131072 : if (((r >> 15) % 16) == 8) {
275 4608 : test.copy();
276 : }
277 131072 : if (((r >> 15) % 32) == 18) {
278 1664 : test.move();
279 : }
280 131072 : if (InsecureRandBits(5) == 19) {
281 3840 : unsigned int num = 1 + (InsecureRandBits(4));
282 7680 : std::vector<int> values(num);
283 34432 : for (auto &v : values) {
284 30592 : v = InsecureRand32();
285 : }
286 7680 : test.resize_uninitialized(values);
287 : }
288 : }
289 : }
290 1 : }
291 :
292 : BOOST_AUTO_TEST_SUITE_END()
|
