1 files changed, 106 insertions, 0 deletions
diff --git a/jstests/sharding/enable_sharding.js b/jstests/sharding/enable_sharding.js
new file mode 100644
index 00000000000..0fe510083c6
--- /dev/null
+++ b/jstests/sharding/enable_sharding.js
@@ -0,0 +1,106 @@
+//
+// Basic tests for enableSharding command.
+//
+
+(function() {
+'use strict';
+
+var st = new ShardingTest({mongos: 2, shards: 2});
+
+jsTest.log('enableSharding can run only against the admin database');
+{
+    assert.commandFailedWithCode(st.s0.getDB('test').runCommand({enableSharding: 'db'}),
+                                 ErrorCodes.Unauthorized);
+}
+
+jsTest.log('Cannot shard system databases except for the config db');
+{
+    assert.commandWorked(st.s0.adminCommand({enableSharding: 'config'}));
+    assert.commandFailed(st.s0.adminCommand({enableSharding: 'local'}));
+    assert.commandFailed(st.s0.adminCommand({enableSharding: 'admin'}));
+}
+
+jsTest.log('Cannot shard db with the name that just differ on case');
+{
+    assert.commandWorked(st.s0.adminCommand({enableSharding: 'db'}));
+    assert.eq(st.s0.getDB('config').databases.findOne({_id: 'db'}).partitioned, true);
+    assert.commandFailedWithCode(st.s0.adminCommand({enableSharding: 'DB'}),
+                                 ErrorCodes.DatabaseDifferCase);
+}
+
+jsTest.log('Cannot shard invalid db name');
+{
+    assert.commandFailed(st.s0.adminCommand({enableSharding: 'a.b'}));
+    assert.commandFailed(st.s0.adminCommand({enableSharding: ''}));
+}
+
+jsTest.log('Attempting to shard already sharded database returns success');
+{
+    assert.commandWorked(st.s0.adminCommand({enableSharding: 'db'}));
+    assert.eq(st.s0.getDB('config').databases.findOne({_id: 'db'}).partitioned, true);
+}
+
+jsTest.log('Verify config.databases metadata');
+{
+    assert.commandWorked(st.s0.getDB('unsharded').foo.insert({aKey: "aValue"}));
+    assert.eq(st.s0.getDB('config').databases.findOne({_id: 'unsharded'}).partitioned, false);
+    assert.commandWorked(st.s0.adminCommand({enableSharding: 'unsharded'}));
+    assert.eq(st.s0.getDB('config').databases.findOne({_id: 'unsharded'}).partitioned, true);
+}
+
+jsTest.log('Sharding a collection before enableSharding is called fails');
+{
+    assert.commandFailed(st.s0.adminCommand({shardCollection: 'TestDB.TestColl', key: {_id: 1}}));
+    assert.commandFailed(st.s1.adminCommand({shardCollection: 'TestDB.TestColl', key: {_id: 1}}));
+
+    assert.commandWorked(st.s0.getDB('TestDB').TestColl.insert({_id: 0}));
+    assert.commandWorked(st.s1.getDB('TestDB').TestColl.insert({_id: 1}));
+}
+
+jsTest.log('Calling enableSharding on one mongos and shardCollection through another must work');
+{
+    assert.commandWorked(st.s0.adminCommand({enableSharding: 'TestDB'}));
+    assert.commandWorked(st.s1.adminCommand({shardCollection: 'TestDB.TestColl', key: {_id: 1}}));
+    assert.commandWorked(st.s0.adminCommand({shardCollection: 'TestDB.TestColl', key: {_id: 1}}));
+}
+
+jsTest.log('Cannot enable sharding on a database using a wrong shard name');
+{
+    assert.commandFailed(st.s0.adminCommand(
+        {enableSharding: 'db2', primaryShard: st.shard1.shardName + '_unenxisting_name_postfix'}));
+}
+
+jsTest.log('Enabling sharding on a database with a valid shard name must work');
+{
+    assert.commandWorked(
+        st.s0.adminCommand({enableSharding: 'db_on_shard0', primaryShard: st.shard0.shardName}));
+    assert.commandWorked(
+        st.s0.adminCommand({enableSharding: 'db_on_shard1', primaryShard: st.shard1.shardName}));
+    assert.eq(st.s0.getDB('config').databases.findOne({_id: 'db_on_shard0'}).primary,
+              st.shard0.shardName);
+    assert.eq(st.s0.getDB('config').databases.findOne({_id: 'db_on_shard1'}).primary,
+              st.shard1.shardName);
+}
+
+jsTest.log(
+    'Enable sharding on a database already created with the correct primary shard name must work');
+{
+    assert.commandWorked(
+        st.s0.adminCommand({enableSharding: 'db_on_shard0', primaryShard: st.shard0.shardName}));
+    assert.commandWorked(
+        st.s0.adminCommand({enableSharding: 'db_on_shard1', primaryShard: st.shard1.shardName}));
+}
+
+jsTest.log(
+    'Cannot enable sharding of a database already created with a different primary shard name');
+{
+    assert.commandFailedWithCode(
+        st.s0.adminCommand({enableSharding: 'db_on_shard0', primaryShard: st.shard1.shardName}),
+        ErrorCodes.NamespaceExists);
+    assert.commandFailedWithCode(
+        st.s0.adminCommand({enableSharding: 'db_on_shard1', primaryShard: st.shard0.shardName}),
+        ErrorCodes.NamespaceExists);
+}
+
+st.stop();
+})();