add configureable delay for the first message on presence endpoint (n…

…eeded for tests)

jest.config.js

@@ -1,4 +1,5 @@
 module.exports = {
   preset: "ts-jest",
   testEnvironment: "node",
+  setupFiles: ["./jestEnvVars.js"]
 };

jestEnvVars.js

@@ -0,0 +1 @@
+process.env.WS_FIRST_MESSAGE_DELAY=50;

src/__tests__/server.test.ts

@@ -415,10 +415,10 @@ describe("TunnelServer", () => {
         const presence = await incomingSocket("client", {
           "Authorization": `Bearer ${jwtToken}`
         }, undefined, false, "presence");
-                
+
         presence.ws.onmessage = (message) => {
-          expect(message.data).toBe(JSON.stringify({ 
-            "presence" : {        
+          expect(message.data).toBe(JSON.stringify({
+            "presence" : {
               "userIds" : []
             }
           })
@@ -445,10 +445,10 @@ describe("TunnelServer", () => {
         const presence = await incomingSocket("client", {
           "Authorization": `Bearer ${jwtToken}`
         }, undefined, false, "presence");
-                
+
         presence.ws.onmessage = (message) => {
           expect(message.data).toBe(JSON.stringify({ 
-            "presence" : {        
+            "presence" : {
               "userIds" : ["lens-user"]
             }
           })
@@ -468,12 +468,12 @@ describe("TunnelServer", () => {
         const presence = await incomingSocket("client", {
           "Authorization": `Bearer ${jwtToken}`
         }, undefined, false, "presence");
-                
+
         await sleep(200); //waits until first message was sent
 
         presence.ws.onmessage = (message) => {
           expect(message.data).toBe(JSON.stringify({ 
-            "presence" : {        
+            "presence" : {
               "userIds" : ["lens-user"]
             }
           })
@@ -501,7 +501,7 @@ describe("TunnelServer", () => {
         const presence = await incomingSocket("client", {
           "Authorization": `Bearer ${jwtToken}`
         }, undefined, false, "presence");
-                
+
         await sleep(200); //waits until first message was sent
 
         const agent = await incomingSocket("agent", {
@@ -515,7 +515,7 @@ describe("TunnelServer", () => {
         presence.ws.onmessage = (message) => {
           console.log(message.data);
           expect(message.data).toBe(JSON.stringify({ 
-            "presence" : {        
+            "presence" : {
               "userIds" : []
             }
           })
@@ -526,11 +526,9 @@ describe("TunnelServer", () => {
         client.ws.close();
 
         await sleep(200); //waits until ClientDisconnected message was received
-    
+
         presence.ws.close();
       });
-
-
     });
   });
 });

src/request-handlers/client-socket.ts

@@ -50,7 +50,7 @@ export function handleClientSocket(req: IncomingMessage, socket: WebSocket, serv
 }
 
 
-export function handleClientPresenceSocket(req: IncomingMessage, socket: WebSocket, server: TunnelServer) {
+export function handleClientPresenceSocket(req: IncomingMessage, socket: WebSocket, server: TunnelServer, firstMessageDelay = 0) {
 
   if (!req.headers.authorization) {
     console.log("SERVER: client did not specify authorization header, closing connection.");
@@ -84,8 +84,8 @@ export function handleClientPresenceSocket(req: IncomingMessage, socket: WebSock
 
   setTimeout(function() {
     sendPresenceData(socket, server, clusterId);
-  }, 50);
-  
+  }, firstMessageDelay);
+
   server.on("ClientConnected", () => {
     sendPresenceData(socket, server, clusterId);
   });

src/server.ts

@@ -43,7 +43,6 @@ export class TunnelServer {
       });
     });
 
-
     this.server.listen(port);
 
     return listenPromise;
@@ -123,7 +122,7 @@ export class TunnelServer {
       });
     } else if (url.pathname === "/client/presence") {
       this.ws?.handleUpgrade(req, socket, head, (socket: WebSocket) => {
-        handleClientPresenceSocket(req, socket, this);
+        handleClientPresenceSocket(req, socket, this, parseInt(process.env.WS_FIRST_MESSAGE_DELAY || ""));
       });
     }
   }