Fix for ESNext

compulim · Oct 13, 2024 · 35cb4d4 · 35cb4d4
1 parent 8492ed7
commit 35cb4d4
Show file tree

Hide file tree

Showing 4 changed files with 26 additions and 0 deletions.
diff --git a/packages/iter-fest/src/asyncGeneratorWithLastValue.spec.ts b/packages/iter-fest/src/asyncGeneratorWithLastValue.spec.ts
@@ -191,3 +191,17 @@ test('return in try-finally', async () => {
   await expect(generator.next()).resolves.toEqual({ done: true, value: undefined });
   expect(generator.lastValue()).toEqual(undefined);
 });
+
+test('passthrough asyncDispose', async () => {
+  const dispose = jest.fn();
+  const symbolAsyncDispose: typeof Symbol.asyncDispose = Symbol.asyncDispose || Symbol.for('Symbol.asyncDispose');
+
+  const generator = (async function* () {})();
+
+  // Currently, in Node.js 22.9.0, `Generator` does not have `asyncDispose` yet. We are ponyfilling it in.
+  generator[symbolAsyncDispose] || (generator[symbolAsyncDispose] = dispose);
+
+  await asyncGeneratorWithLastValue<number, void, void>(generator)[symbolAsyncDispose]?.();
+
+  expect(dispose).toHaveBeenCalledTimes(1);
+});
diff --git a/packages/iter-fest/src/asyncGeneratorWithLastValue.ts b/packages/iter-fest/src/asyncGeneratorWithLastValue.ts
@@ -16,6 +16,7 @@ export function asyncGeneratorWithLastValue<T = unknown, TReturn = any, TNext =
   let lastValue: typeof STILL_ITERATING | TReturn = STILL_ITERATING;
 
   const asyncGeneratorWithLastValue = {
+    ...generator,
     [Symbol.asyncIterator]() {
       return asyncGeneratorWithLastValue;
     },

diff --git a/packages/iter-fest/src/generatorWithLastValue.spec.ts b/packages/iter-fest/src/generatorWithLastValue.spec.ts
@@ -162,3 +162,13 @@ test('return in try-finally', () => {
   expect(generator.next()).toEqual({ done: true, value: undefined });
   expect(generator.lastValue()).toEqual(undefined);
 });
+
+test('passthrough map', () => {
+  const generator = (function* () {
+    yield 1;
+    yield 2;
+    yield 3;
+  })();
+
+  expect(generator.reduce((sum, value) => sum + value, 0)).toBe(6);
+});
diff --git a/packages/iter-fest/src/generatorWithLastValue.ts b/packages/iter-fest/src/generatorWithLastValue.ts
@@ -12,6 +12,7 @@ export function generatorWithLastValue<T = unknown, TReturn = any, TNext = unkno
   let lastValue: typeof STILL_ITERATING | TReturn = STILL_ITERATING;
 
   const generatorWithLastValue = {
+    ...generator,
     [Symbol.iterator]() {
       return generatorWithLastValue;
     },