-
Notifications
You must be signed in to change notification settings - Fork 1.8k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
chore(NODE-3521): find/agg cursor session always there
- Loading branch information
Showing
4 changed files
with
144 additions
and
5 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,135 @@ | ||
'use strict'; | ||
|
||
const expect = require('chai').expect; | ||
const mock = require('../../tools/mongodb-mock/index'); | ||
const { Topology } = require('../../../src/sdam/topology'); | ||
const { Long } = require('bson'); | ||
const { MongoDBNamespace, isHello } = require('../../../src/utils'); | ||
const { AggregationCursor } = require('../../../src/cursor/aggregation_cursor'); | ||
|
||
const test = {}; | ||
describe('Aggregation Cursor', function () { | ||
describe('#next', function () { | ||
afterEach(function () { | ||
mock.cleanup(); | ||
}); | ||
beforeEach(function () { | ||
return mock.createServer().then(mockServer => { | ||
test.server = mockServer; | ||
}); | ||
}); | ||
|
||
context('when there is a data bearing server', function () { | ||
beforeEach(function () { | ||
test.server.setMessageHandler(request => { | ||
const doc = request.document; | ||
if (isHello(doc)) { | ||
request.reply(mock.HELLO); | ||
} else if (doc.aggregate) { | ||
request.reply({ | ||
cursor: { | ||
id: Long.fromNumber(1), | ||
ns: 'test.test', | ||
firstBatch: [{ _id: 1, name: 'test' }] | ||
}, | ||
ok: 1 | ||
}); | ||
} | ||
}); | ||
}); | ||
|
||
it('sets the session on the cursor', function (done) { | ||
const topology = new Topology(test.server.hostAddress()); | ||
const cursor = new AggregationCursor( | ||
topology, | ||
MongoDBNamespace.fromString('test.test'), | ||
[], | ||
{} | ||
); | ||
topology.connect(function () { | ||
cursor.next(function () { | ||
expect(cursor.session).to.exist; | ||
topology.close(done); | ||
}); | ||
}); | ||
}); | ||
}); | ||
|
||
context('when there is no data bearing server', function () { | ||
beforeEach(function () { | ||
test.server.setMessageHandler(request => { | ||
const doc = request.document; | ||
if (isHello(doc)) { | ||
request.reply({ errmsg: 'network error' }); | ||
} else if (doc.aggregate) { | ||
request.reply({ | ||
cursor: { | ||
id: Long.fromNumber(1), | ||
ns: 'test.test', | ||
firstBatch: [{ _id: 1, name: 'test' }] | ||
}, | ||
ok: 1 | ||
}); | ||
} | ||
}); | ||
}); | ||
|
||
it('does not set the session on the cursor', function (done) { | ||
const topology = new Topology(test.server.hostAddress(), { | ||
serverSelectionTimeoutMS: 1000 | ||
}); | ||
const cursor = new AggregationCursor( | ||
topology, | ||
MongoDBNamespace.fromString('test.test'), | ||
[], | ||
{} | ||
); | ||
topology.connect(function () { | ||
cursor.next(function () { | ||
expect(cursor.session).to.not.exist; | ||
topology.close(done); | ||
}); | ||
}); | ||
}); | ||
}); | ||
|
||
context('when a data bearing server becomes available', function () { | ||
beforeEach(function () { | ||
let helloCalls = 0; | ||
test.server.setMessageHandler(request => { | ||
const doc = request.document; | ||
if (isHello(doc)) { | ||
request.reply(helloCalls > 0 ? { errmsg: 'network error' } : mock.HELLO); | ||
} else if (doc.aggregate) { | ||
request.reply({ | ||
cursor: { | ||
id: Long.fromNumber(1), | ||
ns: 'test.test', | ||
firstBatch: [{ _id: 1, name: 'test' }] | ||
}, | ||
ok: 1 | ||
}); | ||
} | ||
}); | ||
}); | ||
|
||
it('sets the session on the cursor', function (done) { | ||
const topology = new Topology(test.server.hostAddress(), { | ||
serverSelectionTimeoutMS: 1000 | ||
}); | ||
const cursor = new AggregationCursor( | ||
topology, | ||
MongoDBNamespace.fromString('test.test'), | ||
[], | ||
{} | ||
); | ||
topology.connect(function () { | ||
cursor.next(function () { | ||
expect(cursor.session).to.exist; | ||
topology.close(done); | ||
}); | ||
}); | ||
}); | ||
}); | ||
}); | ||
}); |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters