tests: fixed write after end error in proxy tests

MatrixAI · May 26, 2022 · 5f97ff6 · 5f97ff6
1 parent 116a4e7
commit 5f97ff6
Showing 1 changed file with 16 additions and 4 deletions.
diff --git a/tests/network/Proxy.test.ts b/tests/network/Proxy.test.ts
@@ -638,6 +638,9 @@ describe(Proxy.name, () => {
       utpConn.on('error', (e) => {
         utpConnError(e);
       });
+      utpConn.on('end', async () => {
+        utpConn.destroy();
+      })
       const tlsSocket = new tls.TLSSocket(utpConn, {
         key: Buffer.from(serverKeyPairPem.privateKey, 'ascii'),
         cert: Buffer.from(serverCertPem, 'ascii'),
@@ -659,7 +662,7 @@ describe(Proxy.name, () => {
           tlsSocket.destroy();
         } else {
           logger.debug('Reverse: responds tlsSocket ending');
-          tlsSocket.end();
+          utpConn.end();
           tlsSocket.destroy();
         }
       });
@@ -754,6 +757,9 @@ describe(Proxy.name, () => {
       utpConn.on('error', (e) => {
         utpConnError(e);
       });
+      utpConn.on('end', async () => {
+        utpConn.destroy();
+      })
       const tlsSocket = new tls.TLSSocket(utpConn, {
         key: Buffer.from(serverKeyPairPem.privateKey, 'ascii'),
         cert: Buffer.from(serverCertPem, 'ascii'),
@@ -775,7 +781,7 @@ describe(Proxy.name, () => {
           tlsSocket.destroy();
         } else {
           logger.debug('Reverse: responds tlsSocket ending');
-          tlsSocket.end();
+          utpConn.end();
           tlsSocket.destroy();
         }
       });
@@ -877,6 +883,9 @@ describe(Proxy.name, () => {
       utpConn.on('error', (e) => {
         utpConnError(e);
       });
+      utpConn.on('end', async () => {
+        utpConn.destroy();
+      })
       const tlsSocket = new tls.TLSSocket(utpConn, {
         key: Buffer.from(serverKeyPairPem.privateKey, 'ascii'),
         cert: Buffer.from(serverCertPem, 'ascii'),
@@ -898,7 +907,7 @@ describe(Proxy.name, () => {
           tlsSocket.destroy();
         } else {
           logger.debug('Reverse: responds tlsSocket ending');
-          tlsSocket.end();
+          utpConn.end();
           tlsSocket.destroy();
           logger.debug('Reverse: responded tlsSocket ending');
         }
@@ -1154,6 +1163,9 @@ describe(Proxy.name, () => {
       utpConn.on('error', (e) => {
         utpConnError(e);
       });
+      utpConn.on('end', async () => {
+        utpConn.destroy();
+      })
       const tlsSocket = new tls.TLSSocket(utpConn, {
         key: Buffer.from(serverKeyPairPem.privateKey, 'ascii'),
         cert: Buffer.from(serverCertPem, 'ascii'),
@@ -1175,7 +1187,7 @@ describe(Proxy.name, () => {
           tlsSocket.destroy();
         } else {
           logger.debug('Reverse: responds tlsSocket ending');
-          tlsSocket.end();
+          utpConn.end();
           tlsSocket.destroy();
           logger.debug('Reverse: responded tlsSocket ending');
         }