~ubuntu-branches/ubuntu/wily/juju-mongodb/wily-proposed

// Test that the shardCollection command fails when a preexisting document lacks a shard key field.

// SERVER-8772

var st = new ShardingTest( { shards: 1 } );

st.stopBalancer();

var db = st.s.getDB( 'testDb' );

var coll = db.testColl;

coll.insert( { x:1, z:1 } );

coll.insert( { y:1, z:1 } );

db.adminCommand( { enableSharding:'testDb' } );

/**

 * Assert that the shardCollection command fails, with a preexisting index on the provided

 * 'shardKey'.

 */

function assertInvalidShardKey( shardKey ) {

    // Manually create a shard key index.

    coll.dropIndexes();

    coll.ensureIndex( shardKey );

    // Ensure that the shard key index identifies 'x' as present in one document and absent in the

    // other.

    assert.eq( 1, coll.find( { x:1 } ).hint( shardKey ).itcount() );

    assert.eq( 1, coll.find( { x:{ $exists:false } } ).hint( shardKey ).itcount() );

    // Assert that the shardCollection command fails with the provided 'shardKey'.

    assert.commandFailed( db.adminCommand( { shardCollection:'testDb.testColl', key:shardKey } ),

                          'shardCollection should have failed on key ' + tojson( shardKey ) );

}

// Test single, compound, and hashed shard keys.

assertInvalidShardKey( { x:1 } );

assertInvalidShardKey( { x:1, y:1 } );

assertInvalidShardKey( { y:1, x:1 } );

assertInvalidShardKey( { x:'hashed' } );

st.stop();