jstests/sharding/missing_key.js


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39

// 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();