test(NODE-7204): update tests

Author: tadjik1

test/integration/node-specific/cursor_stream.test.ts

@@ -30,69 +30,62 @@ describe('Cursor Streams', function () {
     await collection.insertMany(docs, { writeConcern: { w: 1 } });
   }
 
-  describe('using Async Iterator (for await...of)', function () {
-    it('should stream all documents correctly, triggering getMores', async function () {
-      const collection = db.collection<{ _id: number }>('streaming_test');
-      await setupCollection(collection, 100);
+  it('should stream all documents and emit "end"', async function () {
+    const collection = db.collection<{ _id: number }>('streaming_test');
+    await setupCollection(collection, 100);
 
-      // Use a small batchSize to force the driver to issue getMore commands
-      const cursor = collection.find({}, { batchSize: 10 });
-      let docCount = 0;
+    const stream = collection.find({}, { batchSize: 10 }).stream();
+    let docCount = 0;
 
-      for await (const doc of cursor) {
+    // Wrap the stream logic in a Promise to use await
+    await new Promise((resolve, reject) => {
+      stream.on('data', doc => {
         expect(doc).to.have.property('_id', docCount);
         docCount++;
-
-        await sleep(100);
-      }
-
-      expect(docCount).to.equal(100);
+      });
+      stream.on('end', resolve);
+      stream.on('error', reject);
     });
+
+    expect(docCount).to.equal(100);
   });
 
-  describe('using Event Emitter API', function () {
-    it('should stream all documents and emit "end"', async function () {
-      const collection = db.collection<{ _id: number }>('streaming_test');
-      await setupCollection(collection, 100);
-
-      const stream = collection.find({}, { batchSize: 10 }).stream();
-      let docCount = 0;
-
-      // Wrap the stream logic in a Promise to use await
-      await new Promise((resolve, reject) => {
-        stream.on('data', doc => {
-          expect(doc).to.have.property('_id', docCount);
-          docCount++;
-        });
-        stream.on('end', resolve);
-        stream.on('error', reject);
-      });
+  it('should stream all documents in for..of', async function () {
+    const collection = db.collection<{ _id: number }>('streaming_test');
+    await setupCollection(collection, 100);
 
-      expect(docCount).to.equal(100);
-    });
+    const stream = collection.find({}, { batchSize: 10 }).stream();
+    let docCount = 0;
 
-    it('should respect manual pause() and resume() calls', async function () {
-      const collection = db.collection<{ _id: number }>('streaming_test');
-      await setupCollection(collection, 10);
+    for await (const doc of stream) {
+      expect(doc).to.have.property('_id', docCount);
+      docCount++;
+    }
 
-      const stream = collection.find({}, { batchSize: 2 }).stream();
-      let docCount = 0;
+    expect(docCount).to.equal(100);
+  });
 
-      await new Promise((resolve, reject) => {
-        stream.on('error', reject);
-        stream.on('end', resolve);
-        stream.on('data', () => {
-          docCount++;
-          // Manually pause the stream
-          stream.pause();
+  it('should respect manual pause() and resume() calls', async function () {
+    const collection = db.collection<{ _id: number }>('streaming_test');
+    await setupCollection(collection, 10);
 
-          // Perform an async operation, then resume
-          sleep(100).then(() => stream.resume());
-        });
-      });
+    const stream = collection.find({}, { batchSize: 2 }).stream();
+    let docCount = 0;
+
+    await new Promise((resolve, reject) => {
+      stream.on('error', reject);
+      stream.on('end', resolve);
+      stream.on('data', () => {
+        docCount++;
+        // Manually pause the stream
+        stream.pause();
 
-      expect(docCount).to.equal(10);
+        // Perform an async operation, then resume
+        sleep(100).then(() => stream.resume());
+      });
     });
+
+    expect(docCount).to.equal(10);
   });
 
   it('should throws error', async function () {