fix: throw error on generator for sync too

Author: sxzz

src/index.ts

@@ -59,7 +59,12 @@ function unwrapYield(value: any, isAsync?: boolean): any {
 function iterateSync<Return>(generator: QuansyncGenerator<Return, unknown>): Return {
   let current = generator.next()
   while (!current.done) {
-    current = generator.next(unwrapYield(current.value))
+    try {
+      current = generator.next(unwrapYield(current.value))
+    }
+    catch (err) {
+      current = generator.throw(err)
+    }
   }
   return unwrapYield(current.value)
 }

test/index.test.ts

@@ -145,18 +145,25 @@ it('handle errors', async () => {
     },
   })
 
-  const fn = quansync(function* () {
+  const returnError = quansync(function* (fn: () => any) {
     try {
-      yield * throwError()
+      yield * fn()
     }
     catch (err) {
       return err
     }
   })
 
-  await expect(fn()).resolves.toThrowErrorMatchingInlineSnapshot(`[Error: async error]`)
-  await expect(fn.async()).resolves.toThrowErrorMatchingInlineSnapshot(`[Error: async error]`)
-  expect(fn.sync()).toMatchInlineSnapshot(`[Error: sync error]`)
+  const fn = quansync(function* (fn: () => any) {
+    return yield * fn()
+  })
+
+  await expect(returnError(throwError)).resolves.toThrowErrorMatchingInlineSnapshot(`[Error: async error]`)
+  await expect(returnError.async(throwError)).resolves.toThrowErrorMatchingInlineSnapshot(`[Error: async error]`)
+  expect(returnError.sync(throwError)).toMatchInlineSnapshot(`[Error: sync error]`)
+
+  expect(returnError.sync(() => fn(throwError))).toMatchInlineSnapshot(`[Error: sync error]`)
+  await expect(returnError.async(() => fn(throwError))).resolves.toMatchInlineSnapshot(`[Error: async error]`)
 })
 
 it('yield generator', async () => {