summaryrefslogtreecommitdiff
path: root/jstests/core/find_getmore_cmd.js
blob: 3f3d50993e76f94366087926bed1fee3be6e9a07 (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 that explicitly invoke the find and getMore commands.
(function() {
    'use strict';

    var cmdRes;
    var cursorId;
    var defaultBatchSize = 101;
    var collName = 'find_getmore_cmd';
    var coll = db[collName];

    coll.drop();
    for (var i = 0; i < 150; i++) {
        assert.writeOK(coll.insert({a: i}));
    }

    // Verify result of a find command that specifies none of the optional arguments.
    cmdRes = db.runCommand({find: collName});
    assert.commandWorked(cmdRes);
    assert.gt(cmdRes.cursor.id, NumberLong(0));
    assert.eq(cmdRes.cursor.ns, coll.getFullName());
    assert.eq(cmdRes.cursor.firstBatch.length, defaultBatchSize);

    // Use a getMore command to get the next batch.
    cursorId = cmdRes.cursor.id;
    cmdRes = db.runCommand({getMore: cmdRes.cursor.id, collection: collName});
    assert.commandWorked(cmdRes);
    assert.eq(cmdRes.cursor.id, NumberLong(0));
    assert.eq(cmdRes.cursor.ns, coll.getFullName());
    assert.eq(cmdRes.cursor.nextBatch.length, 150 - defaultBatchSize);

    // Cursor should have been closed, so attempting to get another batch should fail.
    cmdRes = db.runCommand({getMore: cursorId, collection: collName});
    assert.commandFailed(cmdRes);

    // Find command with limit.
    cmdRes = db.runCommand({find: collName, limit: 10});
    assert.commandWorked(cmdRes);
    assert.eq(cmdRes.cursor.id, NumberLong(0));
    assert.eq(cmdRes.cursor.ns, coll.getFullName());
    assert.eq(cmdRes.cursor.firstBatch.length, 10);

    // Find command with positive batchSize followed by getMore command with positive batchSize.
    cmdRes = db.runCommand({find: collName, batchSize: 10});
    assert.commandWorked(cmdRes);
    assert.gt(cmdRes.cursor.id, NumberLong(0));
    assert.eq(cmdRes.cursor.ns, coll.getFullName());
    assert.eq(cmdRes.cursor.firstBatch.length, 10);
    cmdRes =
        db.runCommand({getMore: cmdRes.cursor.id, collection: collName, batchSize: NumberInt(5)});
    assert.gt(cmdRes.cursor.id, NumberLong(0));
    assert.eq(cmdRes.cursor.ns, coll.getFullName());
    assert.eq(cmdRes.cursor.nextBatch.length, 5);

    // Find command with zero batchSize followed by getMore command (default batchSize).
    cmdRes = db.runCommand({find: collName, batchSize: 0});
    assert.commandWorked(cmdRes);
    assert.gt(cmdRes.cursor.id, NumberLong(0));
    assert.eq(cmdRes.cursor.ns, coll.getFullName());
    assert.eq(cmdRes.cursor.firstBatch.length, 0);
    cmdRes =
        db.runCommand({getMore: cmdRes.cursor.id, collection: collName, batchSize: NumberInt(5)});
    assert.gt(cmdRes.cursor.id, NumberLong(0));
    assert.eq(cmdRes.cursor.ns, coll.getFullName());
    assert.eq(cmdRes.cursor.nextBatch.length, 5);

    // Batch size and limit together.
    cmdRes = db.runCommand({find: collName, batchSize: 10, limit: 20});
    assert.commandWorked(cmdRes);
    assert.gt(cmdRes.cursor.id, NumberLong(0));
    assert.eq(cmdRes.cursor.ns, coll.getFullName());
    assert.eq(cmdRes.cursor.firstBatch.length, 10);
    cmdRes =
        db.runCommand({getMore: cmdRes.cursor.id, collection: collName, batchSize: NumberInt(11)});
    assert.eq(cmdRes.cursor.id, NumberLong(0));
    assert.eq(cmdRes.cursor.ns, coll.getFullName());
    assert.eq(cmdRes.cursor.nextBatch.length, 10);

    // Find command with batchSize and singleBatch.
    cmdRes = db.runCommand({find: collName, batchSize: 10, singleBatch: true});
    assert.commandWorked(cmdRes);
    assert.eq(cmdRes.cursor.id, NumberLong(0));
    assert.eq(cmdRes.cursor.ns, coll.getFullName());
    assert.eq(cmdRes.cursor.firstBatch.length, 10);

    // Error on invalid collection name.
    assert.commandFailedWithCode(db.runCommand({find: ""}), ErrorCodes.InvalidNamespace);
})();