Add test for dbwrapper iterators with same-prefix keys.
This commit is contained in:
parent
6030625631
commit
269a4402a8
|
@ -238,4 +238,90 @@ BOOST_AUTO_TEST_CASE(iterator_ordering)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
struct StringContentsSerializer {
|
||||||
|
// Used to make two serialized objects the same while letting them have a different lengths
|
||||||
|
// This is a terrible idea
|
||||||
|
string str;
|
||||||
|
StringContentsSerializer() {}
|
||||||
|
StringContentsSerializer(const string& inp) : str(inp) {}
|
||||||
|
|
||||||
|
StringContentsSerializer& operator+=(const string& s) {
|
||||||
|
str += s;
|
||||||
|
return *this;
|
||||||
|
}
|
||||||
|
StringContentsSerializer& operator+=(const StringContentsSerializer& s) { return *this += s.str; }
|
||||||
|
|
||||||
|
ADD_SERIALIZE_METHODS;
|
||||||
|
|
||||||
|
template <typename Stream, typename Operation>
|
||||||
|
inline void SerializationOp(Stream& s, Operation ser_action, int nType, int nVersion) {
|
||||||
|
if (ser_action.ForRead()) {
|
||||||
|
str.clear();
|
||||||
|
char c = 0;
|
||||||
|
while (true) {
|
||||||
|
try {
|
||||||
|
READWRITE(c);
|
||||||
|
str.push_back(c);
|
||||||
|
} catch (const std::ios_base::failure& e) {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
for (size_t i = 0; i < str.size(); i++)
|
||||||
|
READWRITE(str[i]);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
BOOST_AUTO_TEST_CASE(iterator_string_ordering)
|
||||||
|
{
|
||||||
|
char buf[10];
|
||||||
|
|
||||||
|
path ph = temp_directory_path() / unique_path();
|
||||||
|
CDBWrapper dbw(ph, (1 << 20), true, false, false);
|
||||||
|
for (int x=0x00; x<10; ++x) {
|
||||||
|
for (int y = 0; y < 10; y++) {
|
||||||
|
sprintf(buf, "%d", x);
|
||||||
|
StringContentsSerializer key(buf);
|
||||||
|
for (int z = 0; z < y; z++)
|
||||||
|
key += key;
|
||||||
|
uint32_t value = x*x;
|
||||||
|
BOOST_CHECK(dbw.Write(key, value));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
boost::scoped_ptr<CDBIterator> it(const_cast<CDBWrapper*>(&dbw)->NewIterator());
|
||||||
|
for (int c=0; c<2; ++c) {
|
||||||
|
int seek_start;
|
||||||
|
if (c == 0)
|
||||||
|
seek_start = 0;
|
||||||
|
else
|
||||||
|
seek_start = 5;
|
||||||
|
sprintf(buf, "%d", seek_start);
|
||||||
|
StringContentsSerializer seek_key(buf);
|
||||||
|
it->Seek(seek_key);
|
||||||
|
for (int x=seek_start; x<10; ++x) {
|
||||||
|
for (int y = 0; y < 10; y++) {
|
||||||
|
sprintf(buf, "%d", x);
|
||||||
|
string exp_key(buf);
|
||||||
|
for (int z = 0; z < y; z++)
|
||||||
|
exp_key += exp_key;
|
||||||
|
StringContentsSerializer key;
|
||||||
|
uint32_t value;
|
||||||
|
BOOST_CHECK(it->Valid());
|
||||||
|
if (!it->Valid()) // Avoid spurious errors about invalid iterator's key and value in case of failure
|
||||||
|
break;
|
||||||
|
BOOST_CHECK(it->GetKey(key));
|
||||||
|
BOOST_CHECK(it->GetValue(value));
|
||||||
|
BOOST_CHECK_EQUAL(key.str, exp_key);
|
||||||
|
BOOST_CHECK_EQUAL(value, x*x);
|
||||||
|
it->Next();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
BOOST_CHECK(!it->Valid());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
BOOST_AUTO_TEST_SUITE_END()
|
BOOST_AUTO_TEST_SUITE_END()
|
||||||
|
|
Loading…
Reference in New Issue