Skip to content

Commit 08a1a7e

Browse files
clean up capturing client
1 parent fbb82c0 commit 08a1a7e

File tree

1 file changed

+24
-27
lines changed

1 file changed

+24
-27
lines changed

test/integration/client-side-encryption/client_side_encryption.prose.deadlock.ts renamed to test/integration/client-side-encryption/client_side_encryption.prose.9.deadlock.test.ts

Lines changed: 24 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -30,11 +30,9 @@ const $jsonSchema = BSON.EJSON.parse(
3030
)
3131
);
3232

33-
const kEvents = Symbol('events');
34-
const kClientsCreated = Symbol('clientsCreated');
3533
class CapturingMongoClient extends MongoClient {
36-
[kEvents]: Array<CommandStartedEvent> = [];
37-
[kClientsCreated] = 0;
34+
commandStartedEvents: Array<CommandStartedEvent> = [];
35+
clientsCreated = 0;
3836
constructor(url: string, options: MongoClientOptions = {}) {
3937
options = { ...options, monitorCommands: true, [Symbol.for('@@mdb.skipPingOnConnect')]: true };
4038
if (process.env.MONGODB_API_VERSION) {
@@ -43,8 +41,8 @@ class CapturingMongoClient extends MongoClient {
4341

4442
super(url, options);
4543

46-
this.on('commandStarted', ev => this[kEvents].push(ev));
47-
this.on('topologyOpening', () => this[kClientsCreated]++);
44+
this.on('commandStarted', ev => this.commandStartedEvents.push(ev));
45+
this.on('topologyOpening', () => this.clientsCreated++);
4846
}
4947
}
5048

@@ -162,9 +160,9 @@ describe.only('Connection Pool Deadlock Prevention', function () {
162160
`Case 1: ${JSON.stringify(CASE1)}`,
163161
metadata,
164162
deadlockTest(CASE1, clientEncrypted => {
165-
expect(clientEncrypted[kClientsCreated], 'Incorrect number of clients created').to.equal(2);
163+
expect(clientEncrypted.clientsCreated, 'Incorrect number of clients created').to.equal(2);
166164

167-
const events = clientEncrypted[kEvents];
165+
const events = clientEncrypted.commandStartedEvents;
168166
expect(events).to.have.lengthOf(4);
169167

170168
expect(events[0].command).to.have.property('listCollections');
@@ -186,9 +184,9 @@ describe.only('Connection Pool Deadlock Prevention', function () {
186184
`Case 2: ${JSON.stringify(CASE2)}`,
187185
metadata,
188186
deadlockTest(CASE2, (clientEncrypted, clientKeyVault) => {
189-
expect(clientEncrypted[kClientsCreated], 'Incorrect number of clients created').to.equal(2);
187+
expect(clientEncrypted.clientsCreated, 'Incorrect number of clients created').to.equal(2);
190188

191-
const events = clientEncrypted[kEvents];
189+
const events = clientEncrypted.commandStartedEvents;
192190
expect(events).to.have.lengthOf(3);
193191

194192
expect(events[0].command).to.have.property('listCollections');
@@ -200,7 +198,7 @@ describe.only('Connection Pool Deadlock Prevention', function () {
200198
expect(events[2].command).to.have.property('find');
201199
expect(events[2].command.$db).to.equal('db');
202200

203-
const keyVaultEvents = clientKeyVault[kEvents];
201+
const keyVaultEvents = clientKeyVault.commandStartedEvents;
204202
expect(keyVaultEvents).to.have.lengthOf(1);
205203

206204
expect(keyVaultEvents[0].command).to.have.property('find');
@@ -213,9 +211,9 @@ describe.only('Connection Pool Deadlock Prevention', function () {
213211
`Case 3: ${JSON.stringify(CASE3)}`,
214212
metadata,
215213
deadlockTest(CASE3, clientEncrypted => {
216-
expect(clientEncrypted[kClientsCreated], 'Incorrect number of clients created').to.equal(2);
214+
expect(clientEncrypted.clientsCreated, 'Incorrect number of clients created').to.equal(2);
217215

218-
const events = clientEncrypted[kEvents];
216+
const events = clientEncrypted.commandStartedEvents;
219217
expect(events).to.have.lengthOf(2);
220218

221219
expect(events[0].command).to.have.property('find');
@@ -231,15 +229,15 @@ describe.only('Connection Pool Deadlock Prevention', function () {
231229
`Case 4: ${JSON.stringify(CASE4)}`,
232230
metadata,
233231
deadlockTest(CASE4, (clientEncrypted, clientKeyVault) => {
234-
expect(clientEncrypted[kClientsCreated], 'Incorrect number of clients created').to.equal(1);
232+
expect(clientEncrypted.clientsCreated, 'Incorrect number of clients created').to.equal(1);
235233

236-
const events = clientEncrypted[kEvents];
234+
const events = clientEncrypted.commandStartedEvents;
237235
expect(events).to.have.lengthOf(1);
238236

239237
expect(events[0].command).to.have.property('find');
240238
expect(events[0].command.$db).to.equal('db');
241239

242-
const keyVaultEvents = clientKeyVault[kEvents];
240+
const keyVaultEvents = clientKeyVault.commandStartedEvents;
243241
expect(keyVaultEvents).to.have.lengthOf(1);
244242

245243
expect(keyVaultEvents[0].command).to.have.property('find');
@@ -252,9 +250,9 @@ describe.only('Connection Pool Deadlock Prevention', function () {
252250
`Case 5: ${JSON.stringify(CASE5)}`,
253251
metadata,
254252
deadlockTest(CASE5, clientEncrypted => {
255-
expect(clientEncrypted[kClientsCreated], 'Incorrect number of clients created').to.equal(1);
253+
expect(clientEncrypted.clientsCreated, 'Incorrect number of clients created').to.equal(1);
256254

257-
const events = clientEncrypted[kEvents];
255+
const events = clientEncrypted.commandStartedEvents;
258256
expect(events).to.have.lengthOf(5);
259257

260258
expect(events[0].command).to.have.property('listCollections');
@@ -277,12 +275,11 @@ describe.only('Connection Pool Deadlock Prevention', function () {
277275
const CASE6 = { maxPoolSize: 0, bypassAutoEncryption: false, useKeyVaultClient: true };
278276
it(
279277
`Case 6: ${JSON.stringify(CASE6)}`,
280-
281278
metadata,
282279
deadlockTest(CASE6, (clientEncrypted, clientKeyVault) => {
283-
expect(clientEncrypted[kClientsCreated], 'Incorrect number of clients created').to.equal(1);
280+
expect(clientEncrypted.clientsCreated, 'Incorrect number of clients created').to.equal(1);
284281

285-
const events = clientEncrypted[kEvents];
282+
const events = clientEncrypted.commandStartedEvents;
286283
expect(events).to.have.lengthOf(3);
287284

288285
expect(events[0].command).to.have.property('listCollections');
@@ -294,7 +291,7 @@ describe.only('Connection Pool Deadlock Prevention', function () {
294291
expect(events[2].command).to.have.property('find');
295292
expect(events[2].command.$db).to.equal('db');
296293

297-
const keyVaultEvents = clientKeyVault[kEvents];
294+
const keyVaultEvents = clientKeyVault.commandStartedEvents;
298295
expect(keyVaultEvents).to.have.lengthOf(1);
299296

300297
expect(keyVaultEvents[0].command).to.have.property('find');
@@ -307,9 +304,9 @@ describe.only('Connection Pool Deadlock Prevention', function () {
307304
`Case 7: ${JSON.stringify(CASE7)}`,
308305
metadata,
309306
deadlockTest(CASE7, clientEncrypted => {
310-
expect(clientEncrypted[kClientsCreated], 'Incorrect number of clients created').to.equal(1);
307+
expect(clientEncrypted.clientsCreated, 'Incorrect number of clients created').to.equal(1);
311308

312-
const events = clientEncrypted[kEvents];
309+
const events = clientEncrypted.commandStartedEvents;
313310
expect(events).to.have.lengthOf(2);
314311

315312
expect(events[0].command).to.have.property('find');
@@ -325,15 +322,15 @@ describe.only('Connection Pool Deadlock Prevention', function () {
325322
`Case 8: ${JSON.stringify(CASE8)}`,
326323
metadata,
327324
deadlockTest(CASE8, (clientEncrypted, clientKeyVault) => {
328-
expect(clientEncrypted[kClientsCreated], 'Incorrect number of clients created').to.equal(1);
325+
expect(clientEncrypted.clientsCreated, 'Incorrect number of clients created').to.equal(1);
329326

330-
const events = clientEncrypted[kEvents];
327+
const events = clientEncrypted.commandStartedEvents;
331328
expect(events).to.have.lengthOf(1);
332329

333330
expect(events[0].command).to.have.property('find');
334331
expect(events[0].command.$db).to.equal('db');
335332

336-
const keyVaultEvents = clientKeyVault[kEvents];
333+
const keyVaultEvents = clientKeyVault.commandStartedEvents;
337334
expect(keyVaultEvents).to.have.lengthOf(1);
338335

339336
expect(keyVaultEvents[0].command).to.have.property('find');

0 commit comments

Comments
 (0)