SERVER-1026 handle stacked constraints from in clauses

1 files changed, 48 insertions, 0 deletions

diff --git a/jstests/in4.js b/jstests/in4.js
new file mode 100644
index 00000000000..b763d7f8e51
--- /dev/null
+++ b/jstests/in4.js
@@ -0,0 +1,48 @@
+t = db.jstests_in4;
+
+function checkRanges( a, b ) {
+    expectedCount = a;
+    r = b;
+//    printjson( r );
+    assert.eq.automsg( "expectedCount", "r.length" );
+    for( i in r ) {
+        assert.eq.automsg( "r[ i ][ 0 ]", "r[ i ][ 1 ]" );
+    }
+}
+
+t.drop();
+t.ensureIndex( {a:1,b:1} );
+checkRanges( 1, t.find( {a:2,b:3} ).explain().indexBounds );
+checkRanges( 2, t.find( {a:{$in:[2,3]},b:4} ).explain().indexBounds );
+checkRanges( 2, t.find( {a:2,b:{$in:[3,4]}} ).explain().indexBounds );
+checkRanges( 4, t.find( {a:{$in:[2,3]},b:{$in:[4,5]}} ).explain().indexBounds );
+
+assert.eq.automsg( "2", "t.find( {a:{$in:[2,3]},b:{$gt:4,$lt:10}} ).explain().indexBounds.length" );
+
+t.save( {a:1,b:1} );
+t.save( {a:2,b:4.5} );
+t.save( {a:2,b:4} );
+assert.eq.automsg( "1", "t.find( {a:{$in:[2,3]},b:{$in:[4,5]}} ).explain().nscanned" );
+assert.eq.automsg( "2", "t.findOne( {a:{$in:[2,3]},b:{$in:[4,5]}} ).a" );
+assert.eq.automsg( "4", "t.findOne( {a:{$in:[2,3]},b:{$in:[4,5]}} ).b" );
+
+t.drop();
+t.ensureIndex( {a:1,b:1,c:1} );
+checkRanges( 2, t.find( {a:2,b:{$in:[3,4]},c:5} ).explain().indexBounds );
+
+t.save( {a:2,b:3,c:5} );
+t.save( {a:2,b:3,c:4} );
+assert.eq.automsg( "1", "t.find( {a:2,b:{$in:[3,4]},c:5} ).explain().nscanned" );
+t.remove();
+t.save( {a:2,b:4,c:5} );
+t.save( {a:2,b:4,c:4} );
+assert.eq.automsg( "1", "t.find( {a:2,b:{$in:[3,4]},c:5} ).explain().nscanned" );
+
+t.drop();
+t.ensureIndex( {a:1,b:-1} );
+ib = t.find( {a:2,b:{$in:[3,4]}} ).explain().indexBounds;
+checkRanges( 2, ib );
+assert.automsg( "ib[ 0 ][ 0 ].b > ib[ 1 ][ 0 ].b" );
+ib = t.find( {a:2,b:{$in:[3,4]}} ).sort( {a:-1,b:1} ).explain().indexBounds;
+checkRanges( 2, ib );
+assert.automsg( "ib[ 0 ][ 0 ].b < ib[ 1 ][ 0 ].b" );
\ No newline at end of file