summaryrefslogtreecommitdiff
path: root/jstests/core/find_size.js
blob: 14949f0f4778e12b0affdce8f06ef8f0ddcc0f49 (plain)
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
// Tests the behavior of $size for match expressions.

(function() {
"use strict";

const coll = db.jstests_find_size;
coll.drop();

// Check nested arrays.
assert.commandWorked(coll.insert([
    {a: [{b: [1, 2, 3]}, {b: 1}]},
    {a: [1, [2], [3]]},
    {a: [1, [2]]},
    {a: [[1, [1]]]},
    {a: [[[1]]]},
    {a: 1},
    {a: {}},
]));

assert.eq(1, coll.find({"a": {$size: 3}}).itcount());
assert.eq(2, coll.find({"a": {$size: 2}}).itcount());
assert.eq(2, coll.find({"a": {$size: 1}}).itcount());
assert.eq(0, coll.find({"a": {$size: 4}}).itcount());
assert.eq(0, coll.find({"b": {$size: 1}}).itcount());

// Check dotted paths.
assert.commandWorked(coll.insert([{a: {b: [1, 2]}}, {a: {b: {c: [1]}}}]));

assert.eq(1, coll.find({"a.b": {$size: 2}}).itcount());
assert.eq(1, coll.find({"a.b.c": {$size: 1}}).itcount());
assert.eq(0, coll.find({"a.b.c.d": {$size: 1}}).itcount());
assert.eq(0, coll.find({"a.b": {$size: 1}}).itcount());

// Check nested arrays with dotted paths.
assert.commandWorked(coll.insert({a: {b: {c: [1, [2]]}}, b: {a: {c: []}, b: {a: [[1, 2, 3]]}}}));

assert.eq(1, coll.find({"a.b.c": {$size: 2}}).itcount());
assert.eq(1, coll.find({"b.a.c": {$size: 0}}).itcount());
assert.eq(1, coll.find({"b.b.a": {$size: 1}}).itcount());
assert.eq(0, coll.find({"b.b.a": {$size: 0}}).itcount());
assert.eq(0, coll.find({"a.b.b.a": {$size: 1}}).itcount());

assert(coll.drop());

// Check more nested arrays with dotted paths and that as long as one
// of the elements of array matches, the document matches.
assert.commandWorked(coll.insert([
    {a: [{b: [1, 2, 3]}, {b: 1}]},
    {a: [{b: [1, [2]]}, {b: [1]}]},
    {a: [{b: [{c: [1]}, {c: []}]}, {b: [1]}]},
    {a: {b: [1, [2], [[3]]]}},
    {a: {b: [1, 2, 3]}},
    {b: {a: [1, [2], [[3]]]}},
    {b: {a: []}},
]));

assert.eq(3, coll.find({"a.b": {$size: 3}}).itcount());
assert.eq(2, coll.find({"a.b": {$size: 2}}).itcount());
assert.eq(1, coll.find({"a.b.c": {$size: 1}}).itcount());
assert.eq(1, coll.find({"a.b.c": {$size: 0}}).itcount());
assert.eq(1, coll.find({"b.a": {$size: 3}}).itcount());
assert.eq(1, coll.find({"b.a": {$size: 0}}).itcount());
assert.eq(0, coll.find({"a.b.c": {$size: 3}}).itcount());
assert.eq(0, coll.find({"a.b.c.d": {$size: 1}}).itcount());

// Check arrays of size 0.
assert.commandWorked(coll.insert([{a: []}, {a: []}, {a: {b: []}}]));

assert.eq(2, coll.find({"a": {$size: 0}}).itcount());
assert.eq(1, coll.find({"a.b": {$size: 0}}).itcount());

assert(coll.drop());

// Check ints and longs.
assert.commandWorked(coll.insert([{a: []}, {a: []}, {a: [1]}, {a: [1, 2, 3, 4]}]));

assert.eq(2, coll.find({"a": {$size: 0}}).itcount());
assert.eq(2, coll.find({"a": {$size: NumberLong(0)}}).itcount());
assert.eq(1, coll.find({"a": {$size: NumberInt(4)}}).itcount());

// Check bad inputs.
const badInputs = [-1, NumberLong(-10000), "str", 3.2, 0.1, NumberLong(-9223372036854775808)];
badInputs.forEach(function(input) {
    assert.commandFailed(db.runCommand({find: coll.getName(), filter: {"a": {$size: input}}}),
                         "$size argument " + input + " should have failed");
});
}());