From 7f5025ddc43be633395ebe7741777424bb45a148 Mon Sep 17 00:00:00 2001
From: Luigi Pinca <luigipinca@gmail.com>
Date: Tue, 19 Mar 2019 18:19:47 +0100
Subject: [PATCH] [test] Fix flaky test

---
 test/websocket.test.js | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/test/websocket.test.js b/test/websocket.test.js
index b2a2777d2..a1fd98b6f 100644
--- a/test/websocket.test.js
+++ b/test/websocket.test.js
@@ -380,13 +380,14 @@ describe('WebSocket', () => {
     });
 
     it('does not re-emit `net.Socket` errors', (done) => {
+      const codes = ['EPIPE', 'ECONNABORTED', 'ECANCELED', 'ECONNRESET'];
       const wss = new WebSocket.Server({ port: 0 }, () => {
         const ws = new WebSocket(`ws://localhost:${wss.address().port}`);
 
         ws.on('open', () => {
           ws._socket.on('error', (err) => {
             assert.ok(err instanceof Error);
-            assert.ok(err.message.startsWith('write E'));
+            assert.ok(codes.includes(err.code), `Unexpected code: ${err.code}`);
             ws.on('close', (code, message) => {
               assert.strictEqual(message, '');
               assert.strictEqual(code, 1006);