1 | // Copyright (c) 2011 The LevelDB Authors. All rights reserved. |
2 | // Use of this source code is governed by a BSD-style license that can be |
3 | // found in the LICENSE file. See the AUTHORS file for names of contributors. |
4 | |
5 | #include "util/testutil.h" |
6 | |
7 | #include <string> |
8 | |
9 | #include "util/random.h" |
10 | |
11 | namespace leveldb { |
12 | namespace test { |
13 | |
14 | Slice RandomString(Random* rnd, int len, std::string* dst) { |
15 | dst->resize(len); |
16 | for (int i = 0; i < len; i++) { |
17 | (*dst)[i] = static_cast<char>(' ' + rnd->Uniform(95)); // ' ' .. '~' |
18 | } |
19 | return Slice(*dst); |
20 | } |
21 | |
22 | std::string RandomKey(Random* rnd, int len) { |
23 | // Make sure to generate a wide variety of characters so we |
24 | // test the boundary conditions for short-key optimizations. |
25 | static const char kTestChars[] = {'\0', '\1', 'a', 'b', 'c', |
26 | 'd', 'e', '\xfd', '\xfe', '\xff'}; |
27 | std::string result; |
28 | for (int i = 0; i < len; i++) { |
29 | result += kTestChars[rnd->Uniform(sizeof(kTestChars))]; |
30 | } |
31 | return result; |
32 | } |
33 | |
34 | Slice CompressibleString(Random* rnd, double compressed_fraction, size_t len, |
35 | std::string* dst) { |
36 | int raw = static_cast<int>(len * compressed_fraction); |
37 | if (raw < 1) raw = 1; |
38 | std::string raw_data; |
39 | RandomString(rnd, raw, &raw_data); |
40 | |
41 | // Duplicate the random data until we have filled "len" bytes |
42 | dst->clear(); |
43 | while (dst->size() < len) { |
44 | dst->append(raw_data); |
45 | } |
46 | dst->resize(len); |
47 | return Slice(*dst); |
48 | } |
49 | |
50 | } // namespace test |
51 | } // namespace leveldb |
52 | |