expection option for benchRnu find

1 files changed, 6 insertions, 0 deletions

diff --git a/src/mongo/scripting/bench.cpp b/src/mongo/scripting/bench.cpp
index 01291b1e1f0..c2ae2b3d4a4 100644
--- a/src/mongo/scripting/bench.cpp
+++ b/src/mongo/scripting/bench.cpp
@@ -260,11 +260,17 @@ namespace mongo {
                         int options = e["options"].eoo() ? 0 : e["options"].Int();
                         int batchSize = e["batchSize"].eoo() ? 0 : e["batchSize"].Int();
                         BSONObj filter = e["filter"].eoo() ? BSONObj() : e["filter"].Obj();
+                        int expected = e["expected"].eoo() ? -1 : e["expected"].Int();
 
                         auto_ptr<DBClientCursor> cursor = conn->query( ns, fixQuery( e["query"].Obj() ), limit, skip, &filter, options, batchSize );
 
                         int count = cursor->itcount();
 
+                        if ( expected >= 0 &&  count != expected ) {
+                            cout << "bench query on: " << ns << " expected: " << expected << " got: " << cout << endl;
+                            assert(false);
+                        }
+
                         if( check ){
                             BSONObj thisValue = BSON( "count" << count );
                             int err = scope->invoke( scopeFunc , 0 , &thisValue, 1000 * 60 , false );