From 884cdfff025f33b62155ab5ef1ef5da007feade8 Mon Sep 17 00:00:00 2001
From: Mariusz Nowak <medyk@medikoo.com>
Date: Wed, 7 Jul 2021 13:34:53 +0200
Subject: [PATCH] test(AWS MSK): Improve tests reliability

---
 test/integration/aws/infra-dependent/msk.test.js | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/test/integration/aws/infra-dependent/msk.test.js b/test/integration/aws/infra-dependent/msk.test.js
index 8d73b5d07..3f31b8ef2 100644
--- a/test/integration/aws/infra-dependent/msk.test.js
+++ b/test/integration/aws/infra-dependent/msk.test.js
@@ -86,7 +86,10 @@ describe('AWS - MSK Integration Test', function () {
           FunctionName: `${stackName}-producer`,
           InvocationType: 'RequestResponse',
         }),
-      { timeout: 120 * 1000 }
+      {
+        checkIsComplete: (soFarEvents) =>
+          soFarEvents.reduce((data, event) => data + event.message, '').includes(message),
+      }
     );
 
     const logs = events.reduce((data, event) => data + event.message, '');