Add idle timeout smoke test

uNetworkingAB · web-flow · commit da323442662b · 2025-06-18T18:36:03.000+02:00
diff --git a/tests/smoke.js b/tests/smoke.js
@@ -4,6 +4,7 @@ const WebSocket = require('ws');
 const expectedCloseReasons = {
   invalidFrame: 'Received invalid WebSocket frame',
   tcpFinBeforeClose: 'Received TCP FIN before WebSocket close frame',
+  timeout: 'WebSocket timed out from inactivity',
 };
 
 let closeTestsPassed = true;
@@ -12,6 +13,8 @@ let closeTestsPassed = true;
 const port = 9001;
 
 const server = uWS.App().ws('/*', {
+  idleTimeout: 8, // Set idle timeout to 8 seconds
+  sendPingsAutomatically: false, // Needed for timeout
   open: (ws) => {
     console.log('A WebSocket connected!');
   },
@@ -27,6 +30,8 @@ const server = uWS.App().ws('/*', {
         console.log('Test passed: Invalid WebSocket frame');
       } else if (reason === expectedCloseReasons.tcpFinBeforeClose) {
         console.log('Test passed: TCP FIN before WebSocket close frame');
+      } else if (reason === expectedCloseReasons.timeout) {
+        console.log('Test passed: WebSocket timed out from inactivity');
       } else {
         console.error('Test failed: Unexpected close reason:', reason);
         closeTestsPassed = false;
@@ -40,15 +45,17 @@ const server = uWS.App().ws('/*', {
   if (token) {
     console.log('Listening to port', port);
 
-    const triggerClosure = (malformedData, useEnd) => {
+    const triggerClosure = (malformedData, useEnd, useTimeout) => {
       const client = new WebSocket(`ws://localhost:${port}`);
       
       client.on('open', () => {
         console.log('Client connected to server');
-
         if (useEnd) {
           // Trigger "Received TCP FIN before WebSocket close frame"
           client._socket.end();
+        } else if (useTimeout) {
+          // Trigger timeout by keeping connection idle
+          console.log('Starting timeout test: Keeping connection open for timeout');
         } else {
           // Trigger "Received invalid WebSocket frame"
           client._socket.write(malformedData);
@@ -66,15 +73,21 @@ const server = uWS.App().ws('/*', {
 
     // Test 1: Trigger invalid WebSocket frame
     const malformedData = Buffer.from([0xFF, 0x80, 0x00, 0x00, 0x00, 0x01]);
-    triggerClosure(malformedData, false);
+    triggerClosure(malformedData, false, false);
 
-    // Wait for a short period before triggering the next test to ensure the server handles sequentially
+    // Wait for Test 1 to complete before triggering the next test
     setTimeout(() => {
       // Test 2: Trigger TCP FIN before WebSocket close frame
-      triggerClosure(null, true);
-    }, 1000); // Adjust the delay as necessary
+      triggerClosure(null, true, false);
+      
+      // Wait for Test 2 to complete before starting Test 3
+      setTimeout(() => {
+        // Test 3: Trigger timeout (idleTimeout = 8 should close after ~32s)
+        triggerClosure(null, false, true);
+      }, 1000);
+    }, 1000);
 
-    // Allow some time for tests to run before exiting
+    // Allow time for all tests to complete before exiting
     setTimeout(() => {
       if (closeTestsPassed) {
         console.log('All tests passed.');
@@ -83,7 +96,7 @@ const server = uWS.App().ws('/*', {
         console.error('Some tests failed.');
         process.exit(1);
       }
-    }, 3000); // Adjust the delay as necessary
+    }, 16000); // Increased to 40s to account for timeout test (~32s + buffer)
   } else {
     console.log('Failed to listen to port', port);
     process.exit(1);
