test: fix

Author: nbbeeken

test/integration/run-command/run_cursor_command.test.ts

@@ -1,6 +1,8 @@
+import { expect } from 'chai';
+
 import { CommandStartedEvent, Db, MongoClient } from '../../mongodb';
 
-describe('RunCommand API', () => {
+describe('class RunCommandCursor', () => {
   let client: MongoClient;
   let db: Db;
   let commandsStarted: CommandStartedEvent[];
@@ -21,31 +23,31 @@ describe('RunCommand API', () => {
     await client.close();
   });
 
-  it('should not rerun init command', async () => {
+  it('should only run init command once', async () => {
     const cursor = db.runCursorCommand({ find: 'collection', filter: {}, batchSize: 1 });
-    cursor.batchSize = 1;
+    cursor.setBatchSize(1);
     const it0 = cursor[Symbol.asyncIterator]();
     const it1 = cursor[Symbol.asyncIterator]();
 
-    const doc0_0 = await it0.next();
-    const doc0_1 = await it1.next();
-    const doc1_0 = await it0.next();
-    const done = await it1.next();
-    const done2 = await it0.next();
-    const done3 = await it1.next();
+    const next0it0 = await it0.next(); // find, 1 doc
+    const next0it1 = await it1.next(); // getMore, 1 doc
 
-    // console.log(doc0_0, doc0_1, doc1_0, done, done2, done3);
-    // console.log(commandsStarted);
-  });
+    expect(next0it0).to.deep.equal({ value: { _id: 0 }, done: false });
+    expect(next0it1).to.deep.equal({ value: { _id: 1 }, done: false });
+    expect(commandsStarted.map(c => c.commandName)).to.have.lengthOf(2);
+
+    const next1it0 = await it0.next(); // getMore, 1 doc
+    const next1it1 = await it1.next(); // getMore, 0 doc & exhausted id
+
+    expect(next1it0).to.deep.equal({ value: { _id: 2 }, done: false });
+    expect(next1it1).to.deep.equal({ value: undefined, done: true });
+    expect(commandsStarted.map(c => c.commandName)).to.have.lengthOf(4);
+
+    const next2it0 = await it0.next();
+    const next2it1 = await it1.next();
 
-  it('should be not strange', async () => {
-    // const cursor = db.runCursorCommand({ find: 'collection', filter: {}, batchSize: 1 });
-    // for await (const doc0 of cursor) {
-    //   for await (const doc1 of cursor) {
-    //     for await (const doc2 of cursor) {
-    //       console.log(doc0, doc1, doc2);
-    //     }
-    //   }
-    // }
+    expect(next2it0).to.deep.equal({ value: undefined, done: true });
+    expect(next2it1).to.deep.equal({ value: undefined, done: true });
+    expect(commandsStarted.map(c => c.commandName)).to.have.lengthOf(4);
   });
 });