1 files changed, 67 insertions, 0 deletions
diff --git a/jstests/sortg.js b/jstests/sortg.js
new file mode 100644
index 00000000000..09e93949b4b
--- /dev/null
+++ b/jstests/sortg.js
@@ -0,0 +1,67 @@
+// Test that a memory exception is triggered for in memory sorts, but not for indexed sorts.
+
+t = db.jstests_sortg;
+t.drop();
+
+big = new Array( 1000000 ).toString()
+
+for( i = 0; i < 100; ++i ) {
+    t.save( {b:0} );
+}
+
+for( i = 0; i < 40; ++i ) {
+    t.save( {a:big} );
+}
+
+function memoryException( sortSpec, querySpec ) {
+    querySpec = querySpec || {};
+    assert.throws( function() {
+                  t.find( querySpec ).sort( sortSpec ).batchSize( 1000 ).itcount()
+                  } );
+    assert( db.getLastError().match( /too much data for sort\(\) with no index/ ) );
+    assert.throws( function() {
+                  t.find( querySpec ).sort( sortSpec ).batchSize( 1000 ).explain( true )
+                  } );
+    assert( db.getLastError().match( /too much data for sort\(\) with no index/ ) );
+}
+
+function noMemoryException( sortSpec, querySpec ) {
+    querySpec = querySpec || {};
+    t.find( querySpec ).sort( sortSpec ).batchSize( 1000 ).itcount();
+    if ( 0 ) { // SERVER-5016
+    assert( !db.getLastError() );
+    }
+    t.find( querySpec ).sort( sortSpec ).batchSize( 1000 ).explain( true );
+    if ( 0 ) { // SERVER-5016
+    assert( !db.getLastError() );
+    }
+}
+
+// Unindexed sorts.
+memoryException( {a:1} );
+memoryException( {b:1} );
+
+// Indexed sorts.
+noMemoryException( {_id:1} );
+noMemoryException( {$natural:1} );
+
+t.ensureIndex( {a:1} );
+t.ensureIndex( {b:1} );
+t.ensureIndex( {c:1} );
+
+// These sorts are now indexed.
+noMemoryException( {a:1} );
+noMemoryException( {b:1} );
+
+// A memory exception is triggered for an unindexed sort involving multiple plans.
+memoryException( {d:1}, {b:null,c:null} );
+
+// With an indexed plan on _id:1 and an unindexed plan on b:1, the indexed plan
+// should succeed even if the unindexed one would exhaust its memory limit.
+noMemoryException( {_id:1}, {b:null} );
+
+// With an unindexed plan on b:1 recorded for a query, the query should be
+// retried when the unindexed plan exhausts its memory limit.
+assert.eq( 'BtreeCursor b_1', t.find( {b:0} ).sort( {_id:1} ).explain().cursor ); // Record b:1 plan
+noMemoryException( {_id:1}, {b:null} );
+t.drop();
+\ No newline at end of file