fix: Handle custom authorizer 401 in non in-process runners (#1319)

Co-authored-by: Romulo Campos <56273783+romulocrj@users.noreply.github.com>
dherault · Jan 6, 2022 · 8d61bde · 8d61bde
1 parent 8db63dd
commit 8d61bde
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/events/http/createAuthScheme.js b/src/events/http/createAuthScheme.js
@@ -125,7 +125,7 @@ export default function createAuthScheme(
 
       try {
         const result = await lambdaFunction.runHandler()
-
+        if (result === 'Unauthorized') return Boom.unauthorized('Unauthorized')
         // return processResponse(null, result)
         const policy = result
 

diff --git a/src/lambda/handler-runner/in-process-runner/InProcessRunner.js b/src/lambda/handler-runner/in-process-runner/InProcessRunner.js
@@ -115,8 +115,13 @@ export default class InProcessRunner {
 
     const callbackCalled = new Promise((resolve, reject) => {
       callback = (err, data) => {
+        if (err === 'Unauthorized') {
+          resolve('Unauthorized')
+          return
+        }
         if (err) {
           reject(err)
+          return
         }
         resolve(data)
       }