diff options
Diffstat (limited to 'src/mongo/db/storage/record_store_test_manyiter.cpp')
-rw-r--r-- | src/mongo/db/storage/record_store_test_manyiter.cpp | 108 |
1 files changed, 53 insertions, 55 deletions
diff --git a/src/mongo/db/storage/record_store_test_manyiter.cpp b/src/mongo/db/storage/record_store_test_manyiter.cpp index ec896be50a0..842e2681abc 100644 --- a/src/mongo/db/storage/record_store_test_manyiter.cpp +++ b/src/mongo/db/storage/record_store_test_manyiter.cpp @@ -43,74 +43,72 @@ using std::vector; namespace mongo { - using std::unique_ptr; +using std::unique_ptr; - // Create multiple iterators over an empty record store. - TEST( RecordStoreTestHarness, GetManyIteratorsEmpty ) { - unique_ptr<HarnessHelper> harnessHelper( newHarnessHelper() ); - unique_ptr<RecordStore> rs( harnessHelper->newNonCappedRecordStore() ); +// Create multiple iterators over an empty record store. +TEST(RecordStoreTestHarness, GetManyIteratorsEmpty) { + unique_ptr<HarnessHelper> harnessHelper(newHarnessHelper()); + unique_ptr<RecordStore> rs(harnessHelper->newNonCappedRecordStore()); - { - unique_ptr<OperationContext> opCtx( harnessHelper->newOperationContext() ); - ASSERT_EQUALS( 0, rs->numRecords( opCtx.get() ) ); - } + { + unique_ptr<OperationContext> opCtx(harnessHelper->newOperationContext()); + ASSERT_EQUALS(0, rs->numRecords(opCtx.get())); + } - { - unique_ptr<OperationContext> opCtx( harnessHelper->newOperationContext() ); - for (auto&& cursor : rs->getManyCursors(opCtx.get())) { - ASSERT(!cursor->next()); - ASSERT(!cursor->next()); - } + { + unique_ptr<OperationContext> opCtx(harnessHelper->newOperationContext()); + for (auto&& cursor : rs->getManyCursors(opCtx.get())) { + ASSERT(!cursor->next()); + ASSERT(!cursor->next()); } } +} - // Create multiple iterators over a nonempty record store. - TEST( RecordStoreTestHarness, GetManyIteratorsNonEmpty ) { - unique_ptr<HarnessHelper> harnessHelper( newHarnessHelper() ); - unique_ptr<RecordStore> rs( harnessHelper->newNonCappedRecordStore() ); - - { - unique_ptr<OperationContext> opCtx( harnessHelper->newOperationContext() ); - ASSERT_EQUALS( 0, rs->numRecords( opCtx.get() ) ); - } +// Create multiple iterators over a nonempty record store. +TEST(RecordStoreTestHarness, GetManyIteratorsNonEmpty) { + unique_ptr<HarnessHelper> harnessHelper(newHarnessHelper()); + unique_ptr<RecordStore> rs(harnessHelper->newNonCappedRecordStore()); - const int nToInsert = 10; - RecordId locs[nToInsert]; - for ( int i = 0; i < nToInsert; i++ ) { - unique_ptr<OperationContext> opCtx( harnessHelper->newOperationContext() ); - { - stringstream ss; - ss << "record " << i; - string data = ss.str(); - - WriteUnitOfWork uow( opCtx.get() ); - StatusWith<RecordId> res = rs->insertRecord( opCtx.get(), - data.c_str(), - data.size() + 1, - false ); - ASSERT_OK( res.getStatus() ); - locs[i] = res.getValue(); - uow.commit(); - } - } + { + unique_ptr<OperationContext> opCtx(harnessHelper->newOperationContext()); + ASSERT_EQUALS(0, rs->numRecords(opCtx.get())); + } + const int nToInsert = 10; + RecordId locs[nToInsert]; + for (int i = 0; i < nToInsert; i++) { + unique_ptr<OperationContext> opCtx(harnessHelper->newOperationContext()); { - unique_ptr<OperationContext> opCtx( harnessHelper->newOperationContext() ); - ASSERT_EQUALS( nToInsert, rs->numRecords( opCtx.get() ) ); + stringstream ss; + ss << "record " << i; + string data = ss.str(); + + WriteUnitOfWork uow(opCtx.get()); + StatusWith<RecordId> res = + rs->insertRecord(opCtx.get(), data.c_str(), data.size() + 1, false); + ASSERT_OK(res.getStatus()); + locs[i] = res.getValue(); + uow.commit(); } + } - set<RecordId> remain( locs, locs + nToInsert ); - { - unique_ptr<OperationContext> opCtx( harnessHelper->newOperationContext() ); - for (auto&& cursor : rs->getManyCursors(opCtx.get())) { - while (auto record = cursor->next()) { - ASSERT_EQ(remain.erase(record->id), size_t(1)); - } + { + unique_ptr<OperationContext> opCtx(harnessHelper->newOperationContext()); + ASSERT_EQUALS(nToInsert, rs->numRecords(opCtx.get())); + } - ASSERT(!cursor->next()); + set<RecordId> remain(locs, locs + nToInsert); + { + unique_ptr<OperationContext> opCtx(harnessHelper->newOperationContext()); + for (auto&& cursor : rs->getManyCursors(opCtx.get())) { + while (auto record = cursor->next()) { + ASSERT_EQ(remain.erase(record->id), size_t(1)); } - ASSERT( remain.empty() ); + + ASSERT(!cursor->next()); } + ASSERT(remain.empty()); } +} -} // namespace mongo +} // namespace mongo |