Page MenuHomePhabricator
Differential D2200 Diff 6365 src/test/dbwrapper_tests.cpp

Changeset View

Standalone View

View Options

src/test/dbwrapper_tests.cpp

Show All 17 Linesbool is_null_key(const std::vector<uint8_t> &key) {
return isnull; return isnull;
} }
BOOST_FIXTURE_TEST_SUITE(dbwrapper_tests, BasicTestingSetup) BOOST_FIXTURE_TEST_SUITE(dbwrapper_tests, BasicTestingSetup)
BOOST_AUTO_TEST_CASE(dbwrapper) { BOOST_AUTO_TEST_CASE(dbwrapper) {
// Perform tests both obfuscated and non-obfuscated. // Perform tests both obfuscated and non-obfuscated.
for (int i = 0; i < 2; i++) { for (bool obfuscate : {false, true}) {
bool obfuscate = (bool)i;
fs::path ph = fs::temp_directory_path() / fs::unique_path(); fs::path ph = fs::temp_directory_path() / fs::unique_path();
CDBWrapper dbw(ph, (1 << 20), true, false, obfuscate); CDBWrapper dbw(ph, (1 << 20), true, false, obfuscate);
char key = 'k'; char key = 'k';
uint256 in = InsecureRand256(); uint256 in = InsecureRand256();
uint256 res; uint256 res;
// Ensure that we're doing real obfuscation when obfuscate=true // Ensure that we're doing real obfuscation when obfuscate=true
BOOST_CHECK(obfuscate != BOOST_CHECK(obfuscate !=
is_null_key(dbwrapper_private::GetObfuscateKey(dbw))); is_null_key(dbwrapper_private::GetObfuscateKey(dbw)));
BOOST_CHECK(dbw.Write(key, in)); BOOST_CHECK(dbw.Write(key, in));
BOOST_CHECK(dbw.Read(key, res)); BOOST_CHECK(dbw.Read(key, res));
BOOST_CHECK_EQUAL(res.ToString(), in.ToString()); BOOST_CHECK_EQUAL(res.ToString(), in.ToString());
} }
} }
// Test batch operations // Test batch operations
BOOST_AUTO_TEST_CASE(dbwrapper_batch) { BOOST_AUTO_TEST_CASE(dbwrapper_batch) {
// Perform tests both obfuscated and non-obfuscated. // Perform tests both obfuscated and non-obfuscated.
for (int i = 0; i < 2; i++) { for (bool obfuscate : {false, true}) {
bool obfuscate = (bool)i;
fs::path ph = fs::temp_directory_path() / fs::unique_path(); fs::path ph = fs::temp_directory_path() / fs::unique_path();
CDBWrapper dbw(ph, (1 << 20), true, false, obfuscate); CDBWrapper dbw(ph, (1 << 20), true, false, obfuscate);
char key = 'i'; char key = 'i';
uint256 in = InsecureRand256(); uint256 in = InsecureRand256();
char key2 = 'j'; char key2 = 'j';
uint256 in2 = InsecureRand256(); uint256 in2 = InsecureRand256();
char key3 = 'k'; char key3 = 'k';
Show All 18 Lines for (bool obfuscate : {false, true}) {
// key3 should've never been written // key3 should've never been written
BOOST_CHECK(dbw.Read(key3, res) == false); BOOST_CHECK(dbw.Read(key3, res) == false);
} }
} }
BOOST_AUTO_TEST_CASE(dbwrapper_iterator) { BOOST_AUTO_TEST_CASE(dbwrapper_iterator) {
// Perform tests both obfuscated and non-obfuscated. // Perform tests both obfuscated and non-obfuscated.
for (int i = 0; i < 2; i++) { for (bool obfuscate : {false, true}) {
bool obfuscate = (bool)i;
fs::path ph = fs::temp_directory_path() / fs::unique_path(); fs::path ph = fs::temp_directory_path() / fs::unique_path();
CDBWrapper dbw(ph, (1 << 20), true, false, obfuscate); CDBWrapper dbw(ph, (1 << 20), true, false, obfuscate);
// The two keys are intentionally chosen for ordering // The two keys are intentionally chosen for ordering
char key = 'j'; char key = 'j';
uint256 in = InsecureRand256(); uint256 in = InsecureRand256();
BOOST_CHECK(dbw.Write(key, in)); BOOST_CHECK(dbw.Write(key, in));
char key2 = 'k'; char key2 = 'k';
▲ Show 20 Lines • Show All 110 Lines • ▼ Show 20 LinesBOOST_AUTO_TEST_CASE(iterator_ordering) {
for (int x = 0x00; x < 256; ++x) { for (int x = 0x00; x < 256; ++x) {
uint8_t key = x; uint8_t key = x;
uint32_t value = x * x; uint32_t value = x * x;
BOOST_CHECK(dbw.Write(key, value)); BOOST_CHECK(dbw.Write(key, value));
} }
std::unique_ptr<CDBIterator> it( std::unique_ptr<CDBIterator> it(
const_cast<CDBWrapper &>(dbw).NewIterator()); const_cast<CDBWrapper &>(dbw).NewIterator());
for (int c = 0; c < 2; ++c) { for (int seek_start : {0x00, 0x80}) {
int seek_start;
if (c == 0)
seek_start = 0x00;
else
seek_start = 0x80;
it->Seek((uint8_t)seek_start); it->Seek((uint8_t)seek_start);
for (int x = seek_start; x < 256; ++x) { for (int x = seek_start; x < 256; ++x) {
uint8_t key; uint8_t key;
uint32_t value; uint32_t value;
BOOST_CHECK(it->Valid()); BOOST_CHECK(it->Valid());
// Avoid spurious errors about invalid iterator's key and value in // Avoid spurious errors about invalid iterator's key and value in
// case of failure // case of failure
if (!it->Valid()) break; if (!it->Valid()) break;
▲ Show 20 Lines • Show All 57 Lines • ▼ Show 20 Lines for (int x = 0x00; x < 10; ++x) {
key += key; key += key;
uint32_t value = x * x; uint32_t value = x * x;
BOOST_CHECK(dbw.Write(key, value)); BOOST_CHECK(dbw.Write(key, value));
} }
} }
std::unique_ptr<CDBIterator> it( std::unique_ptr<CDBIterator> it(
const_cast<CDBWrapper &>(dbw).NewIterator()); const_cast<CDBWrapper &>(dbw).NewIterator());
for (int c = 0; c < 2; ++c) { for (int seek_start : {0, 5}) {
int seek_start;
if (c == 0)
seek_start = 0;
else
seek_start = 5;
sprintf(buf, "%d", seek_start); sprintf(buf, "%d", seek_start);
StringContentsSerializer seek_key(buf); StringContentsSerializer seek_key(buf);
it->Seek(seek_key); it->Seek(seek_key);
for (int x = seek_start; x < 10; ++x) { for (int x = seek_start; x < 10; ++x) {
for (int y = 0; y < 10; y++) { for (int y = 0; y < 10; y++) {
sprintf(buf, "%d", x); sprintf(buf, "%d", x);
std::string exp_key(buf); std::string exp_key(buf);
for (int z = 0; z < y; z++) for (int z = 0; z < y; z++)
Show All 19 Lines