fix: direct invoke

Author: solaris007

src/index.js

@@ -93,17 +93,11 @@ const runDirect = wrap(processTask)
   .with(secrets, { name: getSecretName })
   .with(helixStatus);
 
-function isSqsEvent(event, context) {
-  if (Array.isArray(event?.Records)) {
-    return true;
-  }
-  if (Array.isArray(context?.invocation?.event?.Records)) {
-    return true;
-  }
-  return typeof event?.type !== 'string';
+function isSqsEvent(event) {
+  return Array.isArray(event?.Records);
 }
 
 export const main = async (event, context) => {
-  const handler = isSqsEvent(event, context) ? runSQS : runDirect;
+  const handler = isSqsEvent(event) ? runSQS : runDirect;
   return handler(event, context);
 };

test/index.test.js

@@ -195,4 +195,25 @@ describe('Index Tests', () => {
     expect(resp.status).to.equal(200);
     expect(directContext.log.info.calledWith('Found task handler for type: dummy')).to.be.true;
   });
+
+  it('treats direct invocation with context invocation records as direct', async () => {
+    const directContext = {
+      ...context,
+      invocation: {
+        event: {
+          Records: [{
+            body: JSON.stringify({ fake: 'value' }),
+          }],
+        },
+      },
+    };
+    const directEvent = {
+      type: 'dummy',
+      siteId: 'direct-site',
+    };
+
+    const resp = await main(directEvent, directContext);
+    expect(resp.status).to.equal(200);
+    expect(directContext.log.info.calledWith('Found task handler for type: dummy')).to.be.true;
+  });
 });