diff options
Diffstat (limited to 'src/mongo/db/storage/record_store_test_validate.h')
-rw-r--r-- | src/mongo/db/storage/record_store_test_validate.h | 122 |
1 files changed, 61 insertions, 61 deletions
diff --git a/src/mongo/db/storage/record_store_test_validate.h b/src/mongo/db/storage/record_store_test_validate.h index e8ff30acfa3..e7b435e4738 100644 --- a/src/mongo/db/storage/record_store_test_validate.h +++ b/src/mongo/db/storage/record_store_test_validate.h @@ -39,81 +39,81 @@ namespace mongo { namespace { - class ValidateAdaptorSpy : public ValidateAdaptor { - public: - ValidateAdaptorSpy() { } +class ValidateAdaptorSpy : public ValidateAdaptor { +public: + ValidateAdaptorSpy() {} - ValidateAdaptorSpy( const std::set<std::string> &remain ) - : _remain( remain ) { - } + ValidateAdaptorSpy(const std::set<std::string>& remain) : _remain(remain) {} - ~ValidateAdaptorSpy() { } + ~ValidateAdaptorSpy() {} - Status validate( const RecordData &recordData, size_t *dataSize ) { - std::string s( recordData.data() ); - ASSERT( 1 == _remain.erase( s ) ); + Status validate(const RecordData& recordData, size_t* dataSize) { + std::string s(recordData.data()); + ASSERT(1 == _remain.erase(s)); - *dataSize = recordData.size(); - return Status::OK(); - } + *dataSize = recordData.size(); + return Status::OK(); + } - bool allValidated() { return _remain.empty(); } + bool allValidated() { + return _remain.empty(); + } - private: - std::set<std::string> _remain; // initially contains all inserted records - }; +private: + std::set<std::string> _remain; // initially contains all inserted records +}; - class ValidateTest : public mongo::unittest::Test { - public: - ValidateTest() - : _harnessHelper( newHarnessHelper() ), - _rs( _harnessHelper->newNonCappedRecordStore() ) { - } +class ValidateTest : public mongo::unittest::Test { +public: + ValidateTest() + : _harnessHelper(newHarnessHelper()), _rs(_harnessHelper->newNonCappedRecordStore()) {} - OperationContext* newOperationContext() { - return _harnessHelper->newOperationContext(); - } + OperationContext* newOperationContext() { + return _harnessHelper->newOperationContext(); + } - RecordStore& getRecordStore() { return *_rs; } + RecordStore& getRecordStore() { + return *_rs; + } - const std::set<std::string>& getInsertedRecords() { return _remain; } + const std::set<std::string>& getInsertedRecords() { + return _remain; + } - void setUp() { - { - std::unique_ptr<OperationContext> opCtx( newOperationContext() ); - ASSERT_EQUALS( 0, _rs->numRecords( opCtx.get() ) ); - } - - int nToInsert = 10; - for ( int i = 0; i < nToInsert; i++ ) { - std::unique_ptr<OperationContext> opCtx( newOperationContext() ); - { - std::stringstream ss; - ss << "record " << i; - std::string data = ss.str(); - ASSERT( _remain.insert( data ).second ); - - WriteUnitOfWork uow( opCtx.get() ); - StatusWith<RecordId> res = _rs->insertRecord( opCtx.get(), - data.c_str(), - data.size() + 1, - false ); - ASSERT_OK( res.getStatus() ); - uow.commit(); - } - } + void setUp() { + { + std::unique_ptr<OperationContext> opCtx(newOperationContext()); + ASSERT_EQUALS(0, _rs->numRecords(opCtx.get())); + } + int nToInsert = 10; + for (int i = 0; i < nToInsert; i++) { + std::unique_ptr<OperationContext> opCtx(newOperationContext()); { - std::unique_ptr<OperationContext> opCtx( newOperationContext() ); - ASSERT_EQUALS( nToInsert, _rs->numRecords( opCtx.get() ) ); + std::stringstream ss; + ss << "record " << i; + std::string data = ss.str(); + ASSERT(_remain.insert(data).second); + + WriteUnitOfWork uow(opCtx.get()); + StatusWith<RecordId> res = + _rs->insertRecord(opCtx.get(), data.c_str(), data.size() + 1, false); + ASSERT_OK(res.getStatus()); + uow.commit(); } } - private: - std::unique_ptr<HarnessHelper> _harnessHelper; - std::unique_ptr<RecordStore> _rs; - std::set<std::string> _remain; - }; + { + std::unique_ptr<OperationContext> opCtx(newOperationContext()); + ASSERT_EQUALS(nToInsert, _rs->numRecords(opCtx.get())); + } + } + +private: + std::unique_ptr<HarnessHelper> _harnessHelper; + std::unique_ptr<RecordStore> _rs; + std::set<std::string> _remain; +}; -} // namespace -} // namespace mongo +} // namespace +} // namespace mongo |