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
64
65
66
67
68
69
70
71
72
73
74
75
|
/**
* Tests for invalid usages of $unionWith, or invalid stages within the $unionWith sub-pipeline.
*/
(function() {
"use strict";
load("jstests/libs/fixture_helpers.js"); // For isReplSet() and isSharded().
const baseColl = db["base"];
baseColl.drop();
const unionColl = db["union"];
unionColl.drop();
// Ensure the base collection exists.
assert.commandWorked(baseColl.insert({a: 1}));
// Disallowed within an update pipeline.
assert.commandFailedWithCode(baseColl.update({a: 1}, [{$unionWith: unionColl.getName()}]),
ErrorCodes.InvalidOptions);
function assertFailsWithCode(pipeline, errCode) {
assert.commandFailedWithCode(db.runCommand({
aggregate: baseColl.getName(),
pipeline: pipeline,
cursor: {},
}),
errCode);
}
// Change streams are only supported against a replica set.
if (FixtureHelpers.isReplSet(db) || FixtureHelpers.isSharded(baseColl)) {
// Disallowed alongside a $changeStream.
assertFailsWithCode([{$changeStream: {}}, {$unionWith: unionColl.getName()}],
ErrorCodes.IllegalOperation);
// Likewise, $changeStream is disallowed within a $unionWith sub-pipeline.
assertFailsWithCode(
[{$unionWith: {coll: unionColl.getName(), pipeline: [{$changeStream: {}}]}}], 31441);
assert.commandFailedWithCode(db.runCommand({
aggregate: 1,
pipeline: [{$changeStream: {}}, {$unionWith: unionColl.getName()}],
cursor: {},
}),
ErrorCodes.IllegalOperation);
}
// $unionWith sub-pipeline cannot contain stages which write data ($merge, $out).
let subPipe = [{$out: "some_out_coll"}];
assertFailsWithCode([{$unionWith: {coll: unionColl.getName(), pipeline: subPipe}}], 31441);
subPipe = [{
$merge: {
into: {db: db.getName(), coll: "some_merge_coll"},
whenMatched: "replace",
whenNotMatched: "fail"
}
}];
assertFailsWithCode([{$unionWith: {coll: unionColl.getName(), pipeline: subPipe}}], 31441);
// Test that collection-less stages are not allowed within the $unionWith sub-pipeline.
subPipe = [{$listCachedAndActiveUsers: {}}];
assertFailsWithCode([{$unionWith: {coll: unionColl.getName(), pipeline: subPipe}}],
ErrorCodes.InvalidNamespace);
subPipe = [{$listLocalSessions: {}}];
assertFailsWithCode([{$unionWith: {coll: unionColl.getName(), pipeline: subPipe}}],
ErrorCodes.InvalidNamespace);
if (FixtureHelpers.isSharded(baseColl)) {
subPipe = [{$currentOp: {localOps: true}}];
assertFailsWithCode([{$unionWith: {coll: unionColl.getName(), pipeline: subPipe}}],
ErrorCodes.InvalidNamespace);
}
})();
|
