summaryrefslogtreecommitdiff
path: root/jstests/noPassthrough/logical_session_cursor_checks.js
blob: cbdb2a38011e6e99c73dbf000dad1b76e8b094cb (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
88
89
90
91
92
93
94
// @tags: [
//   requires_sharding,
// ]

(function() {
'use strict';

function runFixture(Fixture) {
    var fixture = new Fixture();
    var conn = fixture.getConn();
    var admin = conn.getDB("admin");
    var data = conn.getDB("data_storage");

    admin.createUser({user: 'admin', pwd: 'admin', roles: jsTest.adminUserRoles});
    admin.auth("admin", "admin");
    data.createUser({user: 'admin', pwd: 'admin', roles: jsTest.basicUserRoles});
    data.createUser({user: 'user0', pwd: 'password', roles: jsTest.basicUserRoles});
    admin.logout();

    data.auth("user0", "password");
    assert.commandWorked(data.test.insert({name: "first", data: 1}));
    assert.commandWorked(data.test.insert({name: "second", data: 2}));

    // Test that getMore works correctly on the same session.
    {
        var session1 = conn.startSession();
        var session2 = conn.startSession();
        var res = assert.commandWorked(
            session1.getDatabase("data_storage").runCommand({find: "test", batchSize: 0}));
        var cursorId = res.cursor.id;
        assert.commandWorked(session1.getDatabase("data_storage")
                                 .runCommand({getMore: cursorId, collection: "test"}));

        session2.endSession();
        session1.endSession();
    }

    // Test that getMore correctly gives an error, when using a cursor on a different session.
    {
        var session1 = conn.startSession();
        var session2 = conn.startSession();
        var res = assert.commandWorked(
            session1.getDatabase("data_storage").runCommand({find: "test", batchSize: 0}));
        var cursorId = res.cursor.id;
        assert.commandFailed(session2.getDatabase("data_storage")
                                 .runCommand({getMore: cursorId, collection: "test"}));

        session2.endSession();
        session1.endSession();
    }

    // Test that query.js driven getMore works correctly on the same session.
    {
        var session1 = conn.startSession();
        var session2 = conn.startSession();
        var cursor = session1.getDatabase("data_storage").test.find({}).batchSize(1);
        cursor.next();
        cursor.next();
        cursor.close();

        session2.endSession();
        session1.endSession();
    }

    fixture.stop();
}

function Standalone() {
    this.standalone = MongoRunner.runMongod({auth: "", nojournal: ""});
}

Standalone.prototype.stop = function() {
    MongoRunner.stopMongod(this.standalone);
};

Standalone.prototype.getConn = function() {
    return this.standalone;
};

function Sharding() {
    this.st =
        new ShardingTest({shards: 1, config: 1, mongos: 1, other: {keyFile: 'jstests/libs/key1'}});
}

Sharding.prototype.stop = function() {
    this.st.stop();
};

Sharding.prototype.getConn = function() {
    return this.st.s0;
};

[Standalone, Sharding].forEach(runFixture);
})();