nodejs
diff --git a/Collapse file
‎test/parallel/test-http-upgrade-agent.js‎
Copy file name to clipboardExpand all lines: test/parallel/test-http-upgrade-agent.js
+8-3Lines changed: 8 additions & 3 deletions b/Collapse file
‎test/parallel/test-http-upgrade-agent.js‎
Copy file name to clipboardExpand all lines: test/parallel/test-http-upgrade-agent.js
+8-3Lines changed: 8 additions & 3 deletions
@@ -46,9 +46,14 @@ srv.listen(common.PORT, '127.0.0.1', function() {
   req.end();
 
   req.on('upgrade', function(res, socket, upgradeHead) {
-    // XXX: This test isn't fantastic, as it assumes that the entire response
-    //      from the server will arrive in a single data callback
-    assert.equal(upgradeHead, 'nurtzo');
+    var recvData = upgradeHead;
+    socket.on('data', function(d) {
+      recvData += d;
+    });
+
+    socket.on('close', common.mustCall(function() {
+      assert.equal(recvData, 'nurtzo');
+    }));
 
     console.log(res.headers);
     var expectedHeaders = { 'hello': 'world',