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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
|
/**
* Wraps 'schema' in a $jsonSchema predicate and asserts the validity of 'doc' against schema via a
* variety of different methods:
*
* - find command
* - aggregation command via $match
* - insert command with a $jsonSchema validator
* - update command with a $jsonSchema validator (if possible)
*
* Asserts that 'doc' matches 'schema' if and only if 'valid' is true. Drops 'coll' in the process,
* so do not pass a collection whose contents you wish to preserve.
*/
function assertSchemaMatch(coll, schema, doc, valid) {
const errmsg = "Document " + tojson(doc) +
(valid ? " should have matched the schema " : " unexpectedly matched the schema ") +
tojson(schema);
// Test that after inserting 'doc', we can find it again using $jsonSchema in the find command
// iff 'valid' is true.
coll.drop();
assert.commandWorked(coll.insert(doc));
let count = coll.find({$jsonSchema: schema}).itcount();
assert.eq(count, valid ? 1 : 0, errmsg);
// Test that after inserting 'doc', we can find it again using $jsonSchema in an aggregation
// $match stage iff 'valid' is true.
count = coll.aggregate([{$match: {$jsonSchema: schema}}]).itcount();
assert.eq(count, valid ? 1 : 0, errmsg + " in a $match stage");
// Test that 'doc' can be inserted into a collection using 'schema' as its document validator
// iff 'valid' is true. We explicitly use runCommand to issue the drop to avoid an implicit
// collection creation in sharded_collections_jscore_passthrough.
assert.commandWorked(coll.runCommand("drop"));
assert.commandWorked(coll.runCommand("create", {validator: {$jsonSchema: schema}}));
let res = coll.insert(doc);
if (valid) {
assert.commandWorked(res, errmsg + " during insert document validation");
} else {
assert.writeErrorWithCode(res,
ErrorCodes.DocumentValidationFailure,
errmsg + " during insert document validation");
}
// Test that we can update an existing document to look like 'doc' when the collection has
// 'schema' as its document validator in "strict" mode iff 'valid' is true.
assert.commandWorked(coll.runCommand("drop"));
assert.commandWorked(coll.insert({_id: 0}));
assert.commandWorked(
coll.runCommand("collMod", {validator: {$jsonSchema: schema}, validationLevel: "strict"}));
// Before applying the update, remove the _id field if it exists, or the replacement-style
// update will fail.
let docCopy = Object.extend({}, doc);
delete docCopy._id;
res = coll.update({_id: 0}, docCopy);
if (valid) {
assert.commandWorked(res, errmsg + " during update document validation in strict mode");
} else {
assert.writeErrorWithCode(res,
ErrorCodes.DocumentValidationFailure,
errmsg + " during update document validation in strict mode");
}
}
|
