Skip to content

Commit 39e7757

Browse files
committed
fix: proper sinon stubs
1 parent 2ed629e commit 39e7757

File tree

1 file changed

+12
-16
lines changed

1 file changed

+12
-16
lines changed

test/integration/change-streams/change_stream.test.ts

Lines changed: 12 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -2315,14 +2315,12 @@ describe('ChangeStream resumability', function () {
23152315
await collection.insertOne({ name: 'bailey' });
23162316
await changeStream.next();
23172317

2318-
const mock = sinon
2319-
.stub(changeStream.cursor, 'getMore')
2320-
.callsFake((_batchSize, callback) => {
2321-
mock.restore();
2322-
const error = new MongoServerError({ message: message });
2323-
error.code = code;
2324-
callback(error);
2325-
});
2318+
const mock = sinon.stub(changeStream.cursor, 'getMore').callsFake(async _batchSize => {
2319+
mock.restore();
2320+
const error = new MongoServerError({ message: message });
2321+
error.code = code;
2322+
throw error;
2323+
});
23262324

23272325
try {
23282326
// tryNext is not blocking and on sharded clusters we don't have control of when
@@ -2459,14 +2457,12 @@ describe('ChangeStream resumability', function () {
24592457
await collection.insertOne({ city: 'New York City' });
24602458
await changeStreamIterator.next();
24612459

2462-
const mock = sinon
2463-
.stub(changeStream.cursor, 'getMore')
2464-
.callsFake((_batchSize, callback) => {
2465-
mock.restore();
2466-
const error = new MongoServerError({ message });
2467-
error.code = code;
2468-
callback(error);
2469-
});
2460+
const mock = sinon.stub(changeStream.cursor, 'getMore').callsFake(async _batchSize => {
2461+
mock.restore();
2462+
const error = new MongoServerError({ message });
2463+
error.code = code;
2464+
throw error;
2465+
});
24702466

24712467
const docs = [{ city: 'Seattle' }, { city: 'Boston' }];
24722468
await collection.insertMany(docs);

0 commit comments

Comments
 (0)