mutate error stack with context

darky · Jan 17, 2025 · 5b913db · 5b913db
1 parent 02ea038
commit 5b913db
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 20 deletions.
diff --git a/index.ts b/index.ts
@@ -4863,14 +4863,11 @@ export function klubok(...fns: KeyedFunction<string, Function>[]) {
                 : (async () => {
                     try {
                       return await Promise.resolve(fn(ctx)).then(resp => ({ ...ctx, [fn.key]: resp }))
-                    } catch (error: any) {
-                      throw inspect(
-                        {
-                          error: error.stack ?? error.message ?? error,
-                          ctx,
-                        },
-                        { depth: 3 }
-                      )
+                    } catch (error) {
+                      if (error instanceof Error) {
+                        error.stack += '\ncontext: ' + inspect(ctx)
+                      }
+                      throw error
                     }
                   })()
             )
@@ -4890,14 +4887,11 @@ export function klubok(...fns: KeyedFunction<string, Function>[]) {
                           ? Reflect.get(mock, fn.key)(ctx)
                           : fn(ctx)
                       ).then(resp => ({ ...ctx, [fn.key]: resp }))
-                    } catch (error: any) {
-                      throw inspect(
-                        {
-                          error: error.stack ?? error.message ?? error,
-                          ctx,
-                        },
-                        { depth: 3 }
-                      )
+                    } catch (error) {
+                      if (error instanceof Error) {
+                        error.stack += '\ncontext: ' + inspect(ctx)
+                      }
+                      throw error
                     }
                   })()
             ),

diff --git a/test.ts b/test.ts
@@ -337,8 +337,8 @@ test('throwable error with context for production fn', async () => {
     })
   )
   const err = await fn({ number: 1 }).catch(e => e)
-  assert.match(err, /error: 'Error: test/)
-  assert.match(err, /ctx: { number: 1, incNumber: 2, strNumber: '2' }/)
+  assert.deepStrictEqual(err, new Error('test'))
+  assert.match(err.stack ?? '', /context: { number: 1, incNumber: 2, strNumber: '2' }/)
 })
 
 test('throwable error with context for test fn', async () => {
@@ -350,6 +350,6 @@ test('throwable error with context for test fn', async () => {
     })
   )
   const err = await fn({ number: 1 }, {}, []).catch(e => e)
-  assert.match(err, /error: 'Error: test/)
-  assert.match(err, /ctx: { number: 1, incNumber: 2, strNumber: '2' }/)
+  assert.deepStrictEqual(err, new Error('test'))
+  assert.match(err.stack ?? '', /context: { number: 1, incNumber: 2, strNumber: '2' }/)
 })