[chromedriver] Remove extra call to HttpServer::Close() and increase TestHttpServer buffer limit

chrome/test/chromedriver/net/test_http_server.cc

 // Copyright (c) 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/test/chromedriver/net/test_http_server.h"
 
 #include "base/bind.h"
 #include "base/location.h"
 #include "base/message_loop/message_loop.h"
 #include "base/message_loop/message_loop_proxy.h"
 #include "base/strings/stringprintf.h"
 #include "base/time/time.h"
 #include "net/base/ip_endpoint.h"
 #include "net/base/net_errors.h"
 #include "net/server/http_server_request_info.h"
 #include "net/socket/tcp_server_socket.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
+const int kBufferSize = 100 * 1024 * 1024;  // 100 MB
+
 TestHttpServer::TestHttpServer()
     : thread_("ServerThread"),
       all_closed_event_(false, true),
       request_action_(kAccept),
       message_action_(kEchoMessage) {
 }
 
 TestHttpServer::~TestHttpServer() {
 }
 
@@ skipping 37 matching lines @@
 void TestHttpServer::SetMessageAction(WebSocketMessageAction action) {
   base::AutoLock lock(action_lock_);
   message_action_ = action;
 }
 
 GURL TestHttpServer::web_socket_url() const {
   base::AutoLock lock(url_lock_);
   return web_socket_url_;
 }
 
+void TestHttpServer::OnHttpRequest(
+    int connection_id,
+    const net::HttpServerRequestInfo& info) {
+  server_->SetSendBufferSize(connection_id, kBufferSize);
+  server_->SetReceiveBufferSize(connection_id, kBufferSize);
+}
+
 void TestHttpServer::OnWebSocketRequest(
     int connection_id,
     const net::HttpServerRequestInfo& info) {
+  server_->SetSendBufferSize(connection_id, kBufferSize);
+  server_->SetReceiveBufferSize(connection_id, kBufferSize);
+
   WebSocketRequestAction action;
   {
     base::AutoLock lock(action_lock_);
     action = request_action_;
   }
   connections_.insert(connection_id);
   all_closed_event_.Reset();
 
   switch (action) {
     case kAccept:
       server_->AcceptWebSocket(connection_id, info);
       break;
     case kNotFound:
       server_->Send404(connection_id);
       break;
     case kClose:
       server_->Close(connection_id);
       break;
   }
 }
 
 void TestHttpServer::OnWebSocketMessage(int connection_id,
                                         const std::string& data) {
+  server_->SetSendBufferSize(connection_id, kBufferSize);
+  server_->SetReceiveBufferSize(connection_id, kBufferSize);
   WebSocketMessageAction action;
   {
     base::AutoLock lock(action_lock_);
     action = message_action_;
   }
   switch (action) {
     case kEchoMessage:
       server_->SendOverWebSocket(connection_id, data);
       break;
     case kCloseOnMessage:
@@ skipping 26 matching lines @@
     server_.reset(NULL);
   }
   *success = server_.get();
   event->Signal();
 }
 
 void TestHttpServer::StopOnServerThread(base::WaitableEvent* event) {
   server_.reset(NULL);
   event->Signal();
 }