From efa7dae2931fc601f771b8a8f4669faee1fe9ad1 Mon Sep 17 00:00:00 2001
From: Mariusz Nowak <medyk@medikoo.com>
Date: Wed, 2 Jun 2021 16:39:30 +0200
Subject: [PATCH] test(AWS Stream): Improve race condition handling

---
 test/integration/stream.test.js | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test/integration/stream.test.js b/test/integration/stream.test.js
index 4af92ddd4..389fad8a9 100644
--- a/test/integration/stream.test.js
+++ b/test/integration/stream.test.js
@@ -50,7 +50,7 @@ describe('AWS - Stream Integration Test', function () {
         () => putKinesisRecord(streamName, message),
         {
           checkIsComplete: (events) =>
-            events.reduce((data, event) => data + event.message, '').includes(functionName),
+            events.reduce((data, event) => data + event.message, '').includes(message),
         }
       ).then((events) => {
         const logs = events.reduce((data, event) => data + event.message, '');