SERVER-20029 Add test coverage for dataSize command

1 files changed, 16 insertions, 0 deletions

diff --git a/jstests/sharding/mongos_dataSize_test.js b/jstests/sharding/mongos_dataSize_test.js
new file mode 100644
index 00000000000..389529e7d9d
--- /dev/null
+++ b/jstests/sharding/mongos_dataSize_test.js
@@ -0,0 +1,16 @@
+// This tests the command dataSize on sharded clusters to ensure that they can use the command.
+
+(function() {
+    'use strict';
+
+    let s = new ShardingTest({shards: 2, mongos: 1});
+    let db = s.getDB("test");
+    assert.commandWorked(s.s0.adminCommand({enableSharding: "test"}));
+    assert.commandWorked(s.s0.adminCommand({shardcollection: "test.foo", key: {num: 1}}));
+    assert.commandWorked(
+        s.getPrimaryShard("test").getDB("admin").runCommand({datasize: "test.foo"}));
+    assert.commandFailedWithCode(
+        s.getPrimaryShard("test").getDB("admin").runCommand({datasize: "foo"}),
+        ErrorCodes.InvalidNamespace);
+    s.stop();
+})();