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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
/**
* Demonstrate the expected behavior of $lt and $gt comparisons involving arrays. This is only
* tested without an index, results between index and non-index behavior are compared in
* array_index_and_nonIndex_consistent.js
*/
(function() {
"use strict";
load("jstests/aggregation/extras/utils.js"); // arrayEq
const collName = jsTestName();
const coll = db.getCollection(collName);
coll.drop();
const docsInColl = [
{_id: 0, val: [1, 2]},
{_id: 1, val: [3, 4]},
{_id: 2, val: [3, 1]},
{_id: 3, val: {"test": 5}},
{_id: 4, val: [{"test": 7}]},
{_id: 5, val: [true, false]},
{_id: 6, val: 2},
{_id: 7, val: 3},
{_id: 8, val: 4},
{_id: 9, val: [2]},
{_id: 10, val: [3]},
{_id: 11, val: [4]},
{_id: 12, val: [1, true]},
{_id: 13, val: [true, 1]},
{_id: 14, val: [1, 4]},
{_id: 15, val: [null]},
{_id: 16, val: MinKey},
{_id: 17, val: [MinKey]},
{_id: 18, val: [MinKey, 3]},
{_id: 19, val: [3, MinKey]},
{_id: 20, val: MaxKey},
{_id: 21, val: [MaxKey]},
{_id: 22, val: [MaxKey, 3]},
{_id: 23, val: [3, MaxKey]},
{_id: 24, val: true},
{_id: 25, val: false},
];
assert.commandWorked(coll.insert(docsInColl));
function generateFailedEqString(expected, found) {
return "Expected: " + tojson(expected) + "\n Found: " + tojson(found);
}
function generateExpectedResults(indexes) {
resultSet = [];
indexes.forEach(function(index) {
resultSet.push(docsInColl[index]);
});
return resultSet;
}
// Querying for LT/GT an integer returns all arrays that have a single integer element that
// matches the query.
let resultSet = coll.find({val: {$lt: 2}}).toArray();
let expected = generateExpectedResults([0, 2, 12, 14]);
assert(arrayEq(resultSet, expected), generateFailedEqString(resultSet, expected));
resultSet = coll.find({val: {$gt: 2}}).toArray();
expected = generateExpectedResults([1, 2, 8, 14]);
assert(arrayEq(resultSet, expected), generateFailedEqString(resultSet, expected));
// Test that querying for GT MinKey and LT MaxKey returns all results except for those values.
resultSet = coll.find({val: {$gt: MinKey}}).toArray();
let expectedInts = [...Array(25).keys()];
expectedInts.splice(16, 1);
expected = generateExpectedResults(expectedInts);
assert(arrayEq(resultSet, expected), generateFailedEqString(resultSet, expected));
resultSet = coll.find({val: {$lt: MaxKey}}).toArray();
expectedInts = [...Array(25).keys()];
expectedInts.splice(20, 1);
expected = generateExpectedResults(expectedInts);
assert(arrayEq(resultSet, expected), generateFailedEqString(resultSet, expected));
// Test that querying for GT [MinKey] or LT [MaxKey] returns all other arrays, but nothing else.
resultSet = coll.find({val: {$gt: [MinKey]}}).toArray();
expected = generateExpectedResults([0, 1, 2, 4, 5, 9, 10, 11, 12, 13, 14, 15, 18, 19, 21, 22, 23]);
assert(arrayEq(resultSet, expected), generateFailedEqString(resultSet, expected));
resultSet = coll.find({val: {$lt: [MaxKey]}}).toArray();
expected = generateExpectedResults([0, 1, 2, 4, 5, 9, 10, 11, 12, 13, 14, 15, 17, 18, 19, 22, 23]);
assert(arrayEq(resultSet, expected), generateFailedEqString(resultSet, expected));
// Test that querying for LT/GT true returns no other types.
resultSet = coll.find({val: {$gt: true}}).toArray();
expected = [];
assert(arrayEq(resultSet, expected), generateFailedEqString(resultSet, expected));
resultSet = coll.find({val: {$lt: true}}).toArray();
expected = generateExpectedResults([25]);
assert(arrayEq(resultSet, expected), generateFailedEqString(resultSet, expected));
// Test that querying for LT/GT arrays maintains lexicographic (version number) order and only
// returns arrays.
resultSet = coll.find({val: {$lt: [-1]}}).toArray();
expected = generateExpectedResults([15, 17, 18]);
assert(arrayEq(resultSet, expected), generateFailedEqString(resultSet, expected));
resultSet = coll.find({val: {$lt: [3]}}).toArray();
expected = generateExpectedResults([0, 9, 12, 14, 15, 17, 18]);
assert(arrayEq(resultSet, expected), generateFailedEqString(resultSet, expected));
resultSet = coll.find({val: {$lt: [3, 2]}}).toArray();
expected = generateExpectedResults([0, 2, 9, 12, 14, 15, 17, 18]);
assert(arrayEq(resultSet, expected), (resultSet, expected));
resultSet = coll.find({val: {$gt: [2]}}).toArray();
expected = generateExpectedResults([1, 2, 4, 5, 10, 11, 13, 19, 21, 22]);
assert(arrayEq(resultSet, expected), generateFailedEqString(resultSet, expected));
resultSet = coll.find({val: {$gt: [15]}}).toArray();
expected = generateExpectedResults([5, 13, 21, 22]);
assert(arrayEq(resultSet, expected), generateFailedEqString(resultSet, expected));
resultSet = coll.find({val: {$gt: [3, 3]}}).toArray();
expected = generateExpectedResults([1, 4, 5, 11, 13, 19, 21, 22]);
assert(arrayEq(resultSet, expected), generateFailedEqString(resultSet, expected));
// $gt the empty array should return all arrays.
resultSet = coll.find({val: {$gt: []}}).toArray();
expected = generateExpectedResults([0, 1, 2, 4, 9, 10, 11, 12, 13, 14, 15, 17, 18, 19, 21, 22, 23]);
assert(arrayEq(resultSet, expected), generateFailedEqString(resultSet, expected));
// $lt the empty array should return no arrays.
resultSet = coll.find({val: {$lt: []}}).toArray();
expected = generateExpectedResults([3, 6, 7, 8, 16]);
assert(arrayEq(resultSet, expected), generateFailedEqString(resultSet, expected));
})();
|
