From e75339958f29dccdf0469913a1f0707dfab7c405 Mon Sep 17 00:00:00 2001
From: Mariusz Nowak <medyk@medikoo.com>
Date: Tue, 6 Oct 2020 12:56:23 +0200
Subject: [PATCH] test(AWS Cognito): Improve tests setup

---
 test/integration/cognitoUserPool.test.js | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/test/integration/cognitoUserPool.test.js b/test/integration/cognitoUserPool.test.js
index 8390a86da9b..1e3f4354e59 100644
--- a/test/integration/cognitoUserPool.test.js
+++ b/test/integration/cognitoUserPool.test.js
@@ -75,7 +75,13 @@ describe('AWS - Cognito User Pool Integration Test', function() {
       await createUser(userPoolId, 'johndoe', '!!!wAsD123456wAsD!!!');
       const events = await confirmCloudWatchLogs(
         `/aws/lambda/${stackName}-${functionName}`,
-        async () => {}
+        async () => {},
+        {
+          checkIsComplete: soFarEvents => {
+            const logs = soFarEvents.reduce((data, event) => data + event.message, '');
+            return logs.includes('userName');
+          },
+        }
       );
       const logs = events.reduce((data, event) => data + event.message, '');
       expect(logs).to.include(`"userPoolId":"${userPoolId}"`);