test(AWS Websocket): Resend messsage if reception was not confirmed

serverless · Apr 24, 2020 · c6974a6 · c6974a6
1 parent 90ceecd
commit c6974a6
Showing 1 changed file with 8 additions and 3 deletions.
diff --git 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');