test(AWS SQS): Improve test reliability

serverless · Jul 8, 2021 · a2f7c00 · a2f7c00
1 parent 780a67a
commit a2f7c00
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/test/integration/aws/sqs.test.js b/test/integration/aws/sqs.test.js
@@ -43,8 +43,13 @@ describe('AWS - SQS Integration Test', function () {
       const functionName = 'sqsBasic';
       const message = 'Hello from SQS!';
 
-      return confirmCloudWatchLogs(`/aws/lambda/${stackName}-${functionName}`, () =>
-        sendSqsMessage(queueName, message)
+      return confirmCloudWatchLogs(
+        `/aws/lambda/${stackName}-${functionName}`,
+        () => sendSqsMessage(queueName, message),
+        {
+          checkIsComplete: (events) =>
+            events.reduce((data, event) => data + event.message, '').includes(message),
+        }
       ).then((events) => {
         const logs = events.reduce((data, event) => data + event.message, '');
         expect(logs).to.include(functionName);