From c6974a63c8e600450cdc4e9f52b84500aaa81d35 Mon Sep 17 00:00:00 2001
From: Mariusz Nowak <mariusz@medikoo.com>
Date: Wed, 22 Apr 2020 14:33:07 +0200
Subject: [PATCH] test(AWS Websocket): Resend messsage if reception was not
 confirmed

---
 tests/integration-all/websocket/tests.js | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/tests/integration-all/websocket/tests.js b/tests/integration-all/websocket/tests.js
index 1b9c18042..27cc883c8 100644
--- a/tests/integration-all/websocket/tests.js
+++ b/tests/integration-all/websocket/tests.js
@@ -68,15 +68,20 @@ describe('AWS - API Gateway Websocket Integration Test', function() {
           }
         })(reject);
 
-        ws.on('error', reject);
-        ws.on('open', () => {
+        let timeoutId;
+        const sendMessage = () => {
           log.debug("Sending message to 'hello' route");
           ws.send(JSON.stringify({ action: 'hello', name: 'serverless' }));
-        });
+          timeoutId = setTimeout(sendMessage, 1000);
+        };
+
+        ws.on('error', reject);
+        ws.on('open', sendMessage);
 
         ws.on('close', resolve);
 
         ws.on('message', event => {
+          clearTimeout(timeoutId);
           try {
             log.debug(`Received WebSocket message: ${event}`);
             expect(event).to.equal('Hello, serverless');