From 3f0fe5986cab83a050fc353111728f6ec9ecd56d Mon Sep 17 00:00:00 2001
From: Mariusz Nowak <medyk@medikoo.com>
Date: Fri, 20 Nov 2020 17:32:36 +0100
Subject: [PATCH] test: Improvements

---
 test/integration/schedule.test.js |  4 ++++
 test/integration/sns.test.js      | 11 +++++++++--
 2 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/test/integration/schedule.test.js b/test/integration/schedule.test.js
index 278d37b8b..235f0cdf7 100644
--- a/test/integration/schedule.test.js
+++ b/test/integration/schedule.test.js
@@ -41,6 +41,10 @@ describe('AWS - Schedule Integration Test', function() {
 
       return confirmCloudWatchLogs(`/aws/lambda/${stackName}-${functionName}`, async () => {}, {
         timeout: 3 * 60 * 1000,
+        checkIsComplete: soFarEvents => {
+          const logs = soFarEvents.reduce((data, event) => data + event.message, '');
+          return logs.includes('transformedInput');
+        },
       }).then(events => {
         const logs = events.reduce((data, event) => data + event.message, '');
         expect(logs).to.include(functionName);
diff --git a/test/integration/sns.test.js b/test/integration/sns.test.js
index 8dce6b2ec..57c1c4179 100644
--- a/test/integration/sns.test.js
+++ b/test/integration/sns.test.js
@@ -47,8 +47,15 @@ describe('AWS - SNS Integration Test', function() {
       const functionName = 'snsMinimal';
       const message = 'Hello from SNS!';
 
-      return confirmCloudWatchLogs(`/aws/lambda/${stackName}-${functionName}`, () =>
-        publishSnsMessage(minimalTopicName, message)
+      return confirmCloudWatchLogs(
+        `/aws/lambda/${stackName}-${functionName}`,
+        () => publishSnsMessage(minimalTopicName, message),
+        {
+          checkIsComplete: soFarEvents => {
+            const logs = soFarEvents.reduce((data, event) => data + event.message, '');
+            return logs.includes(message);
+          },
+        }
       ).then(events => {
         const logs = events.reduce((data, event) => data + event.message, '');
         expect(logs).to.include(functionName);