SERVER-30178 extend JSON Schema parser to handle "minItems" and "maxItems"

1 files changed, 49 insertions, 0 deletions

diff --git a/jstests/core/json_schema/array_keywords.js b/jstests/core/json_schema/array_keywords.js
new file mode 100644
index 00000000000..9faf0275226
--- /dev/null
+++ b/jstests/core/json_schema/array_keywords.js
@@ -0,0 +1,49 @@
+/**
+ * Tests JSON Schema keywords related to arrays:
+ *   - minItems
+ *   - maxItems
+ */
+(function() {
+    "use strict";
+
+    const coll = db.getCollection("json_schema_arrays");
+    coll.drop();
+
+    assert.writeOK(coll.insert({_id: 0, a: 1}));
+    assert.writeOK(coll.insert({_id: 1, a: []}));
+    assert.writeOK(coll.insert({_id: 2, a: [2, "str"]}));
+    assert.writeOK(coll.insert({_id: 3, a: [3, 3, "foo"]}));
+    assert.writeOK(coll.insert({_id: 4}));
+
+    function assertFindResultsSortedEq(query, expected) {
+        assert.eq(coll.find(query, {_id: 1}).sort({_id: 1}).toArray(),
+                  expected,
+                  "JSON Schema keyword did not match the expected documents");
+    }
+
+    // Test that the JSON Schema fails to parse if "minItems" is not a valid number.
+    assert.throws(() => coll.find({$jsonSchema: {minItems: "blah"}}).itcount());
+    assert.throws(() => coll.find({$jsonSchema: {minItems: -1}}).itcount());
+    assert.throws(() => coll.find({$jsonSchema: {minItems: 12.5}}).itcount());
+
+    // Test that "minItems" matches non-arrays, or arrays with at least the given number of items.
+    assertFindResultsSortedEq({$jsonSchema: {minItems: 10}},
+                              [{_id: 0}, {_id: 1}, {_id: 2}, {_id: 3}, {_id: 4}]);
+    assertFindResultsSortedEq({$jsonSchema: {properties: {a: {minItems: 10}}}},
+                              [{_id: 0}, {_id: 4}]);
+    assertFindResultsSortedEq({$jsonSchema: {properties: {a: {minItems: 2}}}},
+                              [{_id: 0}, {_id: 2}, {_id: 3}, {_id: 4}]);
+
+    // Test that the JSON Schema fails to parse if "maxItems" is not a valid number.
+    assert.throws(() => coll.find({$jsonSchema: {maxItems: "blah"}}).itcount());
+    assert.throws(() => coll.find({$jsonSchema: {maxItems: -1}}).itcount());
+    assert.throws(() => coll.find({$jsonSchema: {maxItems: 12.5}}).itcount());
+
+    // Test that "maxItems" matches non-arrays, or arrays with at most the given number of items.
+    assertFindResultsSortedEq({$jsonSchema: {maxItems: 0}},
+                              [{_id: 0}, {_id: 1}, {_id: 2}, {_id: 3}, {_id: 4}]);
+    assertFindResultsSortedEq({$jsonSchema: {properties: {a: {maxItems: 0}}}},
+                              [{_id: 0}, {_id: 1}, {_id: 4}]);
+    assertFindResultsSortedEq({$jsonSchema: {properties: {a: {maxItems: 2}}}},
+                              [{_id: 0}, {_id: 1}, {_id: 2}, {_id: 4}]);
+}());