//
// Tests mongos-only query behavior
//

var st = new ShardingTest({shards: 1, mongos: 1, verbose: 0});

var mongos = st.s0;
var coll = mongos.getCollection("foo.bar");

//
//
// Ensure we can't use exhaust option through mongos
coll.remove({});
assert.commandWorked(coll.insert({a: 'b'}));
var query = coll.find({});
assert.neq(null, query.next());
query = coll.find({}).addOption(DBQuery.Option.exhaust);
assert.throws(function() {
    query.next();
});

//
//
// Ensure we can't trick mongos by inserting exhaust option on a command through mongos
coll.remove({});
assert.commandWorked(coll.insert({a: 'b'}));
var cmdColl = mongos.getCollection(coll.getDB().toString() + ".$cmd");
var cmdQuery = cmdColl.find({ping: 1}).limit(1);
assert.commandWorked(cmdQuery.next());
cmdQuery = cmdColl.find({ping: 1}).limit(1).addOption(DBQuery.Option.exhaust);
assert.throws(function() {
    assert.commandWorked(cmdQuery.next());
});

jsTest.log("DONE!");

st.stop();