static void OutputTestData(const TestData &v) { for (TestData::const_iterator i = v.begin(); i != v.end(); ++i) { cerr << i->first << ": " << i->second << endl; } }
bool DataExists(const TestData &data, const char *name) { TestData::const_iterator i = data.find(name); return (i != data.end()); }
void TestBtreeIteratorString() { SmiKey leftBoundary("c"); SmiKey leftBoundaryE("e"); SmiKey rightBoundary("f"); SmiKey farLeftBoundary("_"); SmiKey farRightBoundary("zzz"); TestData::iterator beginIter; TestData::iterator endIter; EnterTestFun(); SmiKeyedFile* file = new SmiKeyedFile(SmiKey::String, false); tassert(file != 0); tassert(file->Create()); PrefetchingIterator* piter; string str; TestData testData; TestData::iterator iter; GenerateTestData(testData); for(iter = testData.begin(); iter != testData.end(); ++iter) { SmiKey key(iter->first); SmiRecord record; string s = iter->second; tassert(file->InsertRecord(key, record)); tassert(record.Write(s.c_str(), s.size()) == s.size()); } piter = file->SelectAllPrefetched(); tassert(piter != 0); AssertIteratorsEqualString(piter, testData.begin(), testData.end()); delete piter; endIter = testData.lower_bound("ff"); piter = file->SelectLeftRangePrefetched(rightBoundary); tassert(piter != 0); AssertIteratorsEqualString(piter, testData.begin(), endIter); delete piter; beginIter = testData.find("c"); piter = file->SelectRightRangePrefetched(leftBoundary); tassert(piter != 0); AssertIteratorsEqualString(piter, beginIter, testData.end()); delete piter; beginIter = testData.find("e"); piter = file->SelectRightRangePrefetched(leftBoundaryE); tassert(piter != 0); AssertIteratorsEqualString(piter, beginIter, testData.end()); delete piter; beginIter = testData.find("c"); endIter = testData.lower_bound("ff"); piter = file->SelectRangePrefetched(leftBoundary, rightBoundary); tassert(piter != 0); AssertIteratorsEqualString(piter, beginIter, endIter); delete piter; piter = file->SelectRangePrefetched(farLeftBoundary, farRightBoundary); tassert(piter != 0); AssertIteratorsEqualString(piter, testData.begin(), testData.end()); delete piter; beginIter = testData.find("c"); piter = file->SelectRangePrefetched(leftBoundary, farRightBoundary); tassert(piter != 0); AssertIteratorsEqualString(piter, beginIter, testData.end()); delete piter; endIter = testData.lower_bound("ff"); piter = file->SelectRangePrefetched(farLeftBoundary, rightBoundary); tassert(piter != 0); AssertIteratorsEqualString(piter, testData.begin(), endIter); delete piter; piter = file->SelectRangePrefetched(SmiKey("__"), SmiKey("_____")); tassert(piter != 0); tassert(!piter->Next()); delete piter; piter = file->SelectRangePrefetched(SmiKey("yyy"), SmiKey("zzz")); tassert(piter != 0); tassert(!piter->Next()); delete piter; tassert(file->Close()); tassert(file->Drop()); delete file; file = 0; /* Test wether iterator also works correctly for empty files */ file = new SmiKeyedFile(SmiKey::String, false); tassert(file != 0); tassert(file->Create()); piter = file->SelectAllPrefetched(); tassert(piter != 0); tassert(!piter->Next()); delete piter; tassert(file->Close()); tassert(file->Drop()); delete file; ExitTestFun(__PRETTY_FUNCTION__); }
void TestRecnoIterator() { EnterTestFun(); SmiRecordFile* file = new SmiRecordFile(false); tassert(file != 0); tassert(file->Create()); PrefetchingIterator* piter; SmiRecordId id; string str; set<SmiRecordId> createdIds; TestData testData; TestData::iterator iter; GenerateTestData(testData); for(iter = testData.begin(); iter != testData.end(); ++iter) { SmiRecord record; string s = iter->second; tassert(file->AppendRecord(id, record)); tassert(record.Write(s.c_str(), s.size()) == s.size()); createdIds.insert(id); } tassert(createdIds.size() == testData.size()); piter = file->SelectAllPrefetched(); tassert(piter != 0); AssertIteratorsEqualRecno(piter, testData.begin(), testData.end(), createdIds); delete piter; tassert(file->DeleteRecord(*(createdIds.begin()))); createdIds.erase(createdIds.begin()); testData.erase(testData.begin()); piter = file->SelectAllPrefetched(); tassert(piter != 0); AssertIteratorsEqualRecno(piter, testData.begin(), testData.end(), createdIds); delete piter; tassert(file->Close()); tassert(file->Drop()); delete file; file = 0; /* Test wether iterator also works correctly for empty files */ file = new SmiRecordFile(false); tassert(file != 0); tassert(file->Create()); piter = file->SelectAllPrefetched(); tassert(piter != 0); tassert(!piter->Next()); delete piter; tassert(file->Close()); tassert(file->Drop()); delete file; ExitTestFun(__PRETTY_FUNCTION__); }