diff options
Diffstat (limited to 'src/mongo/dbtests/rollbacktests.cpp')
-rw-r--r-- | src/mongo/dbtests/rollbacktests.cpp | 15 |
1 files changed, 8 insertions, 7 deletions
diff --git a/src/mongo/dbtests/rollbacktests.cpp b/src/mongo/dbtests/rollbacktests.cpp index f742fb68b00..74078abf1b1 100644 --- a/src/mongo/dbtests/rollbacktests.cpp +++ b/src/mongo/dbtests/rollbacktests.cpp @@ -99,16 +99,17 @@ namespace { const NamespaceString& nss, const BSONObj& data ) { Collection* coll = dbHolder().get( txn, nss.db() )->getCollection(nss.ns() ); - scoped_ptr<RecordIterator> iter( coll->getIterator( txn ) ); - ASSERT( !iter->isEOF() ); - RecordId loc = iter->getNext(); - ASSERT( iter->isEOF() ); - ASSERT_EQ( data, coll->docFor( txn, loc ).value() ); + auto cursor = coll->getCursor(txn); + + auto record = cursor->next(); + ASSERT(record); + ASSERT_EQ(data, record->data.releaseToBson()); + + ASSERT(!cursor->next()); } void assertEmpty( OperationContext* txn, const NamespaceString& nss ) { Collection* coll = dbHolder().get( txn, nss.db() )->getCollection(nss.ns() ); - scoped_ptr<RecordIterator> iter( coll->getIterator( txn ) ); - ASSERT( iter->isEOF() ); + ASSERT(!coll->getCursor(txn)->next()); } bool indexExists( OperationContext* txn, const NamespaceString& nss, const string& idxName ) { Collection* coll = dbHolder().get( txn, nss.db() )->getCollection(nss.ns() ); |