fix(@whook/aws-lambda): fix lambdas testing command

nfroidure · Oct 22, 2024 · 123da4f · 123da4f
1 parent 7c62c83
commit 123da4f
Show file tree

Hide file tree

Showing 7 changed files with 25 additions and 132 deletions.
diff --git a/packages/whook-aws-lambda/src/commands/testConsumerLambda.ts b/packages/whook-aws-lambda/src/commands/testConsumerLambda.ts
@@ -69,25 +69,10 @@ async function initTestConsumerLambdaCommand({
       extension,
     );
     const parsedEvent = JSON.parse(event);
-    const result = await new Promise((resolve, reject) => {
-      const handlerPromise = handler(
-        parsedEvent,
-        {
-          succeed: (...args: unknown[]) => {
-            handlerPromise.then(resolve.bind(null, ...args));
-          },
-          fail: reject,
-        },
-        (err: Error, ...args: unknown[]) => {
-          if (err) {
-            reject(err);
-            return;
-          }
-          handlerPromise.then(resolve.bind(null, ...args));
-        },
-      ).catch(reject);
-    });
+    const result = await handler(parsedEvent, {});
 
     log('info', 'SUCCESS:', result as string);
+
+    process.emit('SIGTERM');
   };
 }
diff --git a/packages/whook-aws-lambda/src/commands/testCronLambda.ts b/packages/whook-aws-lambda/src/commands/testCronLambda.ts
@@ -81,29 +81,13 @@ async function initTestCronLambdaCommand({
       type,
       extension,
     );
-
-    const result = await new Promise((resolve, reject) => {
-      const handlerPromise = handler(
-        {
-          time: date === 'now' ? new Date(time()).toISOString() : date,
-          body: JSON.parse(body),
-        },
-        {
-          succeed: (...args: unknown[]) => {
-            handlerPromise.then(resolve.bind(null, ...args));
-          },
-          fail: reject,
-        },
-        (err: Error, ...args: unknown[]) => {
-          if (err) {
-            reject(err);
-            return;
-          }
-          handlerPromise.then(resolve.bind(null, ...args));
-        },
-      ).catch(reject);
-    });
+    const result = await handler({
+      time: date === 'now' ? new Date(time()).toISOString() : date,
+      body: JSON.parse(body),
+    }, {});
 
     log('info', 'SUCCESS:', result as string);
+
+    process.emit('SIGTERM');
   };
 }
diff --git a/packages/whook-aws-lambda/src/commands/testHTTPLambda.ts b/packages/whook-aws-lambda/src/commands/testHTTPLambda.ts
@@ -166,26 +166,10 @@ async function initTestHTTPLambdaCommand({
     }
     log('info', 'AWS_REQUEST:', awsRequest as unknown as string);
 
-    const result: APIGatewayProxyResult = await new Promise(
-      (resolve, reject) => {
-        const handlerPromise = handler(
-          awsRequest,
-          {
-            succeed: (...args: any[]) => {
-              handlerPromise.then(resolve.bind(null, ...args));
-            },
-            fail: reject,
-          },
-          (err: Error, ...args: any[]) => {
-            if (err) {
-              reject(err);
-              return;
-            }
-            handlerPromise.then(resolve.bind(null, ...args));
-          },
-        ).catch(reject);
-      },
-    );
+    const result: APIGatewayProxyResult = await handler(awsRequest);
+
     log('info', 'SUCCESS:', result as unknown as string);
+
+    process.emit('SIGTERM');
   };
 }
diff --git a/packages/whook-aws-lambda/src/commands/testKafkaConsumerLambda.ts b/packages/whook-aws-lambda/src/commands/testKafkaConsumerLambda.ts
@@ -107,25 +107,10 @@ async function initTestKafkaConsumerLambdaCommand({
       extension,
     );
     const parsedEvent = JSON.parse(event);
-    const result = await new Promise((resolve, reject) => {
-      const handlerPromise = handler(
-        parsedEvent,
-        {
-          succeed: (...args: unknown[]) => {
-            handlerPromise.then(resolve.bind(null, ...args));
-          },
-          fail: reject,
-        },
-        (err: Error, ...args: unknown[]) => {
-          if (err) {
-            reject(err);
-            return;
-          }
-          handlerPromise.then(resolve.bind(null, ...args));
-        },
-      ).catch(reject);
-    });
+    const result = await handler(parsedEvent, {});
 
     log('info', 'SUCCESS:', result as string);
+
+    process.emit('SIGTERM');
   };
 }
diff --git a/packages/whook-aws-lambda/src/commands/testLogSubscriberLambda.ts b/packages/whook-aws-lambda/src/commands/testLogSubscriberLambda.ts
@@ -97,25 +97,10 @@ async function initTestS3LambdaCommand({
         data: await encodePayload(JSON.parse(event)),
       },
     };
-    const result = await new Promise((resolve, reject) => {
-      const handlerPromise = handler(
-        parsedEvent,
-        {
-          succeed: (...args: unknown[]) => {
-            handlerPromise.then(resolve.bind(null, ...args));
-          },
-          fail: reject,
-        },
-        (err: Error, ...args: unknown[]) => {
-          if (err) {
-            reject(err);
-            return;
-          }
-          handlerPromise.then(resolve.bind(null, ...args));
-        },
-      ).catch(reject);
-    });
+    const result = await handler(parsedEvent, {});
 
     log('info', 'SUCCESS:', result as string);
+
+    process.emit('SIGTERM');
   };
 }
diff --git a/packages/whook-aws-lambda/src/commands/testS3Lambda.ts b/packages/whook-aws-lambda/src/commands/testS3Lambda.ts
@@ -109,25 +109,10 @@ async function initTestS3LambdaCommand({
       extension,
     );
     const parsedEvent = JSON.parse(event);
-    const result = await new Promise((resolve, reject) => {
-      const handlerPromise = handler(
-        parsedEvent,
-        {
-          succeed: (...args: unknown[]) => {
-            handlerPromise.then(resolve.bind(null, ...args));
-          },
-          fail: reject,
-        },
-        (err: Error, ...args: unknown[]) => {
-          if (err) {
-            reject(err);
-            return;
-          }
-          handlerPromise.then(resolve.bind(null, ...args));
-        },
-      ).catch(reject);
-    });
+    const result = await handler(parsedEvent, {});
 
     log('info', 'SUCCESS:', result as string);
+
+    process.emit('SIGTERM');
   };
 }
diff --git a/packages/whook-aws-lambda/src/commands/testTransformerLambda.ts b/packages/whook-aws-lambda/src/commands/testTransformerLambda.ts
@@ -80,25 +80,10 @@ async function initTestTransformerLambdaCommand({
       extension,
     );
     const parsedEvent = JSON.parse(event);
-    const result = await new Promise((resolve, reject) => {
-      const handlerPromise = handler(
-        parsedEvent,
-        {
-          succeed: (...args: unknown[]) => {
-            handlerPromise.then(resolve.bind(null, ...args));
-          },
-          fail: reject,
-        },
-        (err: Error, ...args: unknown[]) => {
-          if (err) {
-            reject(err);
-            return;
-          }
-          handlerPromise.then(resolve.bind(null, ...args));
-        },
-      ).catch(reject);
-    });
+    const result = await handler(parsedEvent, {});
 
     log('info', 'SUCCESS:', result as string);
+
+    process.emit('SIGTERM');
   };
 }