Buffer will be managed in the connection class

Author: kblok

lib/PuppeteerSharp/Connection.cs

@@ -104,11 +104,27 @@ private async Task<object> GetResponseAsync()
                     return null;
                 }
 
-                var result = await WebSocket.ReceiveAsync(new ArraySegment<byte>(buffer), CancellationToken.None);
+                var endOfMessage = false;
+                string response = string.Empty;
 
-                if (result.MessageType == WebSocketMessageType.Text)
+                while (!endOfMessage)
+                {
+                    var result = await WebSocket.ReceiveAsync(new ArraySegment<byte>(buffer), CancellationToken.None);
+                    endOfMessage = result.EndOfMessage;
+
+                    if (result.MessageType == WebSocketMessageType.Text)
+                    {
+                        response += Encoding.UTF8.GetString(buffer, 0, result.Count);
+                    }
+                    else if (result.MessageType == WebSocketMessageType.Close)
+                    {
+                        OnClose();
+                        return null;
+                    }
+                }
+
+                if (!string.IsNullOrEmpty(response))
                 {
-                    var response = Encoding.UTF8.GetString(buffer, 0, result.Count);
                     dynamic obj = JsonConvert.DeserializeObject(response);
                     var objAsJObject = obj as JObject;
 
@@ -155,12 +171,6 @@ private async Task<object> GetResponseAsync()
                         }
                     }
                 }
-                else if (result.MessageType == WebSocketMessageType.Close)
-                {
-                    OnClose();
-                    return null;
-                }
-
             }
         }
         #endregion