mongodb/jstests/noPassthroughWithMongod/index_check9.js

// Randomized index testing

Random.setRandomSeed();

t = db.test_index_check9;

function doIt() {

t.drop();

function sort() {
    var sort = {};
    for( var i = 0; i < n; ++i ) {
        sort[ fields[ i ] ] = Random.rand() > 0.5 ? 1 : -1;
    }    
    return sort;
}

var fields = [ 'a', 'b', 'c', 'd', 'e' ];
n = Random.randInt( 5 ) + 1;
var idx = sort();

var chars = "abcdefghijklmnopqrstuvwxyz";
var alphas = []
for( var i = 0; i < n; ++i ) {
    alphas.push( Random.rand() > 0.5 );
}
    
t.ensureIndex( idx );

function obj() {
    var ret = {};
    for( var i = 0; i < n; ++i ) {
        ret[ fields[ i ] ] = r( alphas[ i ] );
    }
    return ret;
}

function r( alpha ) {
    if ( !alpha ) {
        return Random.randInt( 10 );
    } else {
        var len = Random.randInt( 10 );
        buf = "";
        for( var i = 0; i < len; ++i ) {
            buf += chars.charAt( Random.randInt( chars.length ) );
        }
        return buf;
    }
}

function check() {
    var v = t.validate();
    if ( !t.valid ) {
        printjson( t );
        assert( t.valid );
    }
    var spec = {};
    for( var i = 0; i < n; ++i ) {
        var predicateType = Random.randInt( 4 );
        switch( predicateType ) {
        case 0 /* range */ : {
            var bounds = [ r( alphas[ i ] ), r( alphas[ i ] ) ];
            if ( bounds[ 0 ] > bounds[ 1 ] ) {
                bounds.reverse();
            }
            var s = {};
            if ( Random.rand() > 0.5 ) {
                s[ "$gte" ] = bounds[ 0 ];
            } else {
                s[ "$gt" ] = bounds[ 0 ];
            }
            if ( Random.rand() > 0.5 ) {
                s[ "$lte" ] = bounds[ 1 ];
            } else {
                s[ "$lt" ] = bounds[ 1 ];
            }
            spec[ fields[ i ] ] = s;
            break;
        }
        case 1 /* $in */ : {
            var vals = []
            var inLength = Random.randInt( 15 );
            for( var j = 0; j < inLength; ++j ) {
                vals.push( r( alphas[ i ] ) );
            }
            spec[ fields[ i ] ] = { $in: vals };
            break;
        }
        case 2 /* equality */ : {
            spec[ fields[ i ] ] = r( alphas[ i ] );
            break;
        }
        default /* no predicate */ :
            break;
        }
    }
    s = sort();
    c1 = t.find( spec, { _id:null } ).sort( s ).hint( idx ).toArray();
    c2 = t.find( spec ).sort( s ).explain().nscanned;
    c3 = t.find( spec, { _id:null } ).sort( s ).hint( {$natural:1} ).toArray();
    count = t.count( spec );
    //    assert.eq( c1, c3, "spec: " + tojson( spec ) + ", sort: " + tojson( s ) );
    //    assert.eq( c1.length, c2 );
    assert.eq( c1, c3 );
    assert.eq( c3.length, count );
}

var bulk = t.initializeUnorderedBulkOp();
for( var i = 0; i < 10000; ++i ) {
    bulk.insert( obj() );
    if( Random.rand() > 0.999 ) {
        print( i );
        assert.writeOK(bulk.execute());
        check();
        bulk = t.initializeUnorderedBulkOp();
    }
}

bulk = t.initializeUnorderedBulkOp();
for( var i = 0; i < 100000; ++i ) {
    if ( Random.rand() > 0.9 ) {
        bulk.insert( obj() );
    } else {
        bulk.find( obj() ).remove(); // improve
    }
    if( Random.rand() > 0.999 ) {
        print( i );
        assert.writeOK(bulk.execute());
        check();
        bulk = t.initializeUnorderedBulkOp();
    }
}
assert.writeOK(bulk.execute());

check();

}

for( var z = 0; z < 5; ++z ) {
    doIt();
}