feat: handle returned errors (closes #28)

unjs · Jul 27, 2021 · 991fcff · 991fcff
1 parent b10b496
commit 991fcff
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/src/app.ts b/src/app.ts
@@ -121,6 +121,8 @@ export function createHandle (stack: Stack): PHandle {
       } else if (type === 'object' && val !== undefined) {
         if (val.buffer) {
           return send(res, val)
+        } else if (val instanceof Error) {
+          throw createError(val)
         } else {
           return send(res, JSON.stringify(val, null, 2), MIMES.json)
         }

diff --git a/test/error.test.ts b/test/error.test.ts
@@ -82,4 +82,18 @@ describe('error', () => {
     const res = await request.get('/')
     expect(res.status).toBe(500)
   })
+
+  it('can handle returned Error', async () => {
+    app.use('/', () => new Error('failed'), { promisify: true })
+
+    const res = await request.get('/')
+    expect(res.status).toBe(500)
+  })
+
+  it('can handle returned H3Error', async () => {
+    app.use('/', () => createError({ statusCode: 501 }), { promisify: true })
+
+    const res = await request.get('/')
+    expect(res.status).toBe(501)
+  })
 })