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 8624130 : void local_check_equal(A a, B b)
35 : {
36 8624130 : local_check(a == b);
37 : }
38 19012100 : void local_check(bool b)
39 : {
40 3982850 : passed &= b;
41 : }
42 213888 : void test() {
43 213888 : const pretype& const_pre_vector = pre_vector;
44 398272 : local_check_equal(real_vector.size(), pre_vector.size());
45 213888 : local_check_equal(real_vector.empty(), pre_vector.empty());
46 2205310 : for (Size s = 0; s < real_vector.size(); s++) {
47 1991420 : local_check(real_vector[s] == pre_vector[s]);
48 3982850 : local_check(&(pre_vector[s]) == &(pre_vector.begin()[s]));
49 1991420 : local_check(&(pre_vector[s]) == &*(pre_vector.begin() + s));
50 3982850 : local_check(&(pre_vector[s]) == &*((pre_vector.end() + s) - real_vector.size()));
51 : }
52 : // local_check(realtype(pre_vector) == real_vector);
53 213888 : local_check(pretype(real_vector.begin(), real_vector.end()) == pre_vector);
54 641664 : local_check(pretype(pre_vector.begin(), pre_vector.end()) == pre_vector);
55 213888 : size_t pos = 0;
56 :
57 2633090 : for (const T& v : pre_vector) {
58 1991420 : local_check(v == real_vector[pos++]);
59 : }
60 213888 : pos = 0;
61 2205310 : for (const T& v : const_pre_vector) {
62 1991420 : local_check(v == real_vector[pos++]);
63 : }
64 :
65 427776 : CDataStream ss1(SER_DISK, 0);
66 213888 : CDataStream ss2(SER_DISK, 0);
67 213888 : ss1 << real_vector;
68 213888 : ss2 << pre_vector;
69 213888 : local_check_equal(ss1.size(), ss2.size());
70 8393470 : for (Size s = 0; s < ss1.size(); s++) {
71 8179580 : local_check_equal(ss1[s], ss2[s]);
72 : }
73 213888 : }
74 :
75 : public:
76 8384 : void resize(Size s) {
77 8384 : real_vector.resize(s);
78 8384 : local_check_equal(real_vector.size(), s);
79 8384 : pre_vector.resize(s);
80 8384 : local_check_equal(pre_vector.size(), s);
81 8384 : test();
82 8384 : }
83 :
84 1536 : void reserve(Size s) {
85 1536 : real_vector.reserve(s);
86 1536 : local_check(real_vector.capacity() >= s);
87 1536 : pre_vector.reserve(s);
88 1536 : local_check(pre_vector.capacity() >= s);
89 1536 : test();
90 1536 : }
91 :
92 31424 : void insert(Size position, const T& value) {
93 31424 : real_vector.insert(real_vector.begin() + position, value);
94 62848 : pre_vector.insert(pre_vector.begin() + position, value);
95 31424 : test();
96 31424 : }
97 :
98 8832 : void insert(Size position, Size count, const T& value) {
99 8832 : real_vector.insert(real_vector.begin() + position, count, value);
100 17664 : pre_vector.insert(pre_vector.begin() + position, count, value);
101 8832 : test();
102 8832 : }
103 :
104 : template<typename I>
105 1856 : void insert_range(Size position, I first, I last) {
106 1856 : real_vector.insert(real_vector.begin() + position, first, last);
107 3712 : pre_vector.insert(pre_vector.begin() + position, first, last);
108 1856 : test();
109 1856 : }
110 :
111 14016 : void erase(Size position) {
112 14016 : real_vector.erase(real_vector.begin() + position);
113 28032 : pre_vector.erase(pre_vector.begin() + position);
114 14016 : test();
115 14016 : }
116 :
117 10688 : void erase(Size first, Size last) {
118 10688 : real_vector.erase(real_vector.begin() + first, real_vector.begin() + last);
119 21376 : pre_vector.erase(pre_vector.begin() + first, pre_vector.begin() + last);
120 10688 : test();
121 10688 : }
122 :
123 116160 : void update(Size pos, const T& value) {
124 116160 : real_vector[pos] = value;
125 116160 : pre_vector[pos] = value;
126 116160 : test();
127 116160 : }
128 :
129 3584 : void push_back(const T& value) {
130 3584 : real_vector.push_back(value);
131 3584 : pre_vector.push_back(value);
132 3584 : test();
133 3584 : }
134 :
135 3072 : void pop_back() {
136 3072 : real_vector.pop_back();
137 3072 : pre_vector.pop_back();
138 3072 : test();
139 3072 : }
140 :
141 64 : void clear() {
142 128 : real_vector.clear();
143 64 : pre_vector.clear();
144 64 : }
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 595264 : Size size() {
152 595264 : return real_vector.size();
153 : }
154 :
155 : Size capacity() {
156 : return pre_vector.capacity();
157 : }
158 :
159 960 : void shrink_to_fit() {
160 960 : pre_vector.shrink_to_fit();
161 960 : test();
162 960 : }
163 :
164 8768 : void swap() {
165 8768 : real_vector.swap(real_vector_alt);
166 8768 : pre_vector.swap(pre_vector_alt);
167 8768 : test();
168 8768 : }
169 :
170 2624 : void move() {
171 2624 : real_vector = std::move(real_vector_alt);
172 2624 : real_vector_alt.clear();
173 2624 : pre_vector = std::move(pre_vector_alt);
174 2624 : pre_vector_alt.clear();
175 2624 : }
176 :
177 3648 : void copy() {
178 3648 : real_vector = real_vector_alt;
179 3648 : pre_vector = pre_vector_alt;
180 3648 : }
181 :
182 4608 : void resize_uninitialized(realtype values) {
183 4608 : size_t r = values.size();
184 4608 : size_t s = real_vector.size() / 2;
185 4608 : if (real_vector.capacity() < s + r) {
186 1344 : real_vector.reserve(s + r);
187 : }
188 4608 : real_vector.resize(s);
189 4608 : pre_vector.resize_uninitialized(s);
190 42240 : for (auto v : values) {
191 37632 : real_vector.push_back(v);
192 : }
193 4608 : auto p = pre_vector.size();
194 4608 : pre_vector.resize_uninitialized(p + r);
195 42240 : for (auto v : values) {
196 37632 : pre_vector[p] = v;
197 37632 : ++p;
198 : }
199 4608 : test();
200 4608 : }
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 31424 : test.insert(InsecureRand32() % (test.size() + 1), InsecureRand32());
220 : }
221 131072 : if (test.size() > 0 && ((r >> 2) % 4) == 1) {
222 14016 : test.erase(InsecureRand32() % test.size());
223 : }
224 131072 : if (((r >> 4) % 8) == 2) {
225 16576 : int new_size = std::max<int>(0, std::min<int>(30, test.size() + (InsecureRand32() % 5) - 2));
226 8384 : test.resize(new_size);
227 : }
228 131072 : if (((r >> 7) % 8) == 3) {
229 8832 : test.insert(InsecureRand32() % (test.size() + 1), 1 + (InsecureRand32() % 2), InsecureRand32());
230 : }
231 131072 : if (((r >> 10) % 8) == 4) {
232 8960 : int del = std::min<int>(test.size(), 1 + (InsecureRand32() % 2));
233 8960 : int beg = InsecureRand32() % (test.size() + 1 - del);
234 8960 : test.erase(beg, beg + del);
235 : }
236 131072 : if (((r >> 13) % 16) == 5) {
237 3584 : test.push_back(InsecureRand32());
238 : }
239 131072 : if (test.size() > 0 && ((r >> 17) % 16) == 6) {
240 3072 : test.pop_back();
241 : }
242 131072 : if (((r >> 21) % 32) == 7) {
243 1856 : int values[4];
244 1856 : int num = 1 + (InsecureRand32() % 4);
245 6912 : for (int i = 0; i < num; i++) {
246 5056 : values[i] = InsecureRand32();
247 : }
248 1856 : test.insert_range(InsecureRand32() % (test.size() + 1), values, values + num);
249 : }
250 131072 : if (((r >> 26) % 32) == 8) {
251 1728 : int del = std::min<int>(test.size(), 1 + (InsecureRand32() % 4));
252 1728 : int beg = InsecureRand32() % (test.size() + 1 - del);
253 1728 : test.erase(beg, beg + del);
254 : }
255 131072 : r = InsecureRand32();
256 131072 : if (r % 32 == 9) {
257 1536 : test.reserve(InsecureRand32() % 32);
258 : }
259 131072 : if ((r >> 5) % 64 == 10) {
260 960 : test.shrink_to_fit();
261 : }
262 131072 : if (test.size() > 0) {
263 116160 : test.update(InsecureRand32() % test.size(), InsecureRand32());
264 : }
265 131072 : if (((r >> 11) % 1024) == 11) {
266 64 : 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 8768 : test.swap();
273 : }
274 131072 : if (((r >> 15) % 16) == 8) {
275 3648 : test.copy();
276 : }
277 131072 : if (((r >> 15) % 32) == 18) {
278 2624 : test.move();
279 : }
280 131072 : if (InsecureRandBits(5) == 19) {
281 4608 : unsigned int num = 1 + (InsecureRandBits(4));
282 9216 : std::vector<int> values(num);
283 42240 : for (auto &v : values) {
284 37632 : v = InsecureRand32();
285 : }
286 9216 : test.resize_uninitialized(values);
287 : }
288 : }
289 : }
290 1 : }
291 :
292 : BOOST_AUTO_TEST_SUITE_END()
|
