fix: throw error on generator

Author: sxzz

src/index.ts

@@ -67,7 +67,12 @@ function iterateSync<Return>(generator: QuansyncGenerator<Return, unknown>): Ret
 async function iterateAsync<Return>(generator: QuansyncGenerator<Return, unknown>): Promise<Return> {
   let current = generator.next()
   while (!current.done) {
-    current = generator.next(await unwrapYield(current.value, true))
+    try {
+      current = generator.next(await unwrapYield(current.value, true))
+    }
+    catch (err) {
+      current = generator.throw(err)
+    }
   }
   return current.value
 }

test/index.test.ts

@@ -118,7 +118,7 @@ it('yield promise', async () => {
   await expect(run.async('foo')).resolves.toBe('foo')
 })
 
-it('handle errors', async () => {
+it('throw errors', async () => {
   const throwError = quansync({
     name: 'throwError',
     sync: () => {
@@ -134,6 +134,31 @@ it('handle errors', async () => {
   expect(() => throwError.sync()).toThrowErrorMatchingInlineSnapshot(`[Error: sync error]`)
 })
 
+it('handle errors', async () => {
+  const throwError = quansync({
+    name: 'throwError',
+    sync: () => {
+      throw new Error('sync error')
+    },
+    async: async () => {
+      return Promise.reject(new Error('async error'))
+    },
+  })
+
+  const fn = quansync(function* () {
+    try {
+      yield * throwError()
+    }
+    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]`)
+})
+
 it('yield generator', async () => {
   const toString = quansync({
     name: 'toString',