| OLD | NEW |
| 1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/test/chromedriver/net/test_http_server.h" | 5 #include "chrome/test/chromedriver/net/test_http_server.h" |
| 6 | 6 |
| 7 #include <utility> | 7 #include <utility> |
| 8 | 8 |
| 9 #include "base/bind.h" | 9 #include "base/bind.h" |
| 10 #include "base/callback_helpers.h" | 10 #include "base/callback_helpers.h" |
| 11 #include "base/location.h" | 11 #include "base/location.h" |
| 12 #include "base/message_loop/message_loop.h" | 12 #include "base/message_loop/message_loop.h" |
| 13 #include "base/single_thread_task_runner.h" | 13 #include "base/single_thread_task_runner.h" |
| 14 #include "base/strings/stringprintf.h" | 14 #include "base/strings/stringprintf.h" |
| 15 #include "base/time/time.h" | 15 #include "base/time/time.h" |
| 16 #include "net/base/ip_endpoint.h" | 16 #include "net/base/ip_endpoint.h" |
| 17 #include "net/base/net_errors.h" | 17 #include "net/base/net_errors.h" |
| 18 #include "net/server/http_server_request_info.h" | 18 #include "net/server/http_server_request_info.h" |
| 19 #include "net/socket/tcp_server_socket.h" | 19 #include "net/socket/tcp_server_socket.h" |
| 20 #include "testing/gtest/include/gtest/gtest.h" | 20 #include "testing/gtest/include/gtest/gtest.h" |
| 21 | 21 |
| 22 const int kBufferSize = 100 * 1024 * 1024; // 100 MB | 22 const int kBufferSize = 100 * 1024 * 1024; // 100 MB |
| 23 | 23 |
| 24 TestHttpServer::TestHttpServer() | 24 TestHttpServer::TestHttpServer() |
| 25 : thread_("ServerThread"), | 25 : thread_("ServerThread"), |
| 26 all_closed_event_(false, true), | 26 all_closed_event_(base::WaitableEvent::ResetPolicy::AUTOMATIC, |
| 27 base::WaitableEvent::InitialState::SIGNALED), |
| 27 request_action_(kAccept), | 28 request_action_(kAccept), |
| 28 message_action_(kEchoMessage) { | 29 message_action_(kEchoMessage) {} |
| 29 } | |
| 30 | 30 |
| 31 TestHttpServer::~TestHttpServer() { | 31 TestHttpServer::~TestHttpServer() { |
| 32 } | 32 } |
| 33 | 33 |
| 34 bool TestHttpServer::Start() { | 34 bool TestHttpServer::Start() { |
| 35 base::Thread::Options options(base::MessageLoop::TYPE_IO, 0); | 35 base::Thread::Options options(base::MessageLoop::TYPE_IO, 0); |
| 36 bool thread_started = thread_.StartWithOptions(options); | 36 bool thread_started = thread_.StartWithOptions(options); |
| 37 EXPECT_TRUE(thread_started); | 37 EXPECT_TRUE(thread_started); |
| 38 if (!thread_started) | 38 if (!thread_started) |
| 39 return false; | 39 return false; |
| 40 bool success; | 40 bool success; |
| 41 base::WaitableEvent event(false, false); | 41 base::WaitableEvent event(base::WaitableEvent::ResetPolicy::AUTOMATIC, |
| 42 base::WaitableEvent::InitialState::NOT_SIGNALED); |
| 42 thread_.task_runner()->PostTask( | 43 thread_.task_runner()->PostTask( |
| 43 FROM_HERE, base::Bind(&TestHttpServer::StartOnServerThread, | 44 FROM_HERE, base::Bind(&TestHttpServer::StartOnServerThread, |
| 44 base::Unretained(this), &success, &event)); | 45 base::Unretained(this), &success, &event)); |
| 45 event.Wait(); | 46 event.Wait(); |
| 46 return success; | 47 return success; |
| 47 } | 48 } |
| 48 | 49 |
| 49 void TestHttpServer::Stop() { | 50 void TestHttpServer::Stop() { |
| 50 if (!thread_.IsRunning()) | 51 if (!thread_.IsRunning()) |
| 51 return; | 52 return; |
| 52 base::WaitableEvent event(false, false); | 53 base::WaitableEvent event(base::WaitableEvent::ResetPolicy::AUTOMATIC, |
| 54 base::WaitableEvent::InitialState::NOT_SIGNALED); |
| 53 thread_.task_runner()->PostTask( | 55 thread_.task_runner()->PostTask( |
| 54 FROM_HERE, base::Bind(&TestHttpServer::StopOnServerThread, | 56 FROM_HERE, base::Bind(&TestHttpServer::StopOnServerThread, |
| 55 base::Unretained(this), &event)); | 57 base::Unretained(this), &event)); |
| 56 event.Wait(); | 58 event.Wait(); |
| 57 thread_.Stop(); | 59 thread_.Stop(); |
| 58 } | 60 } |
| 59 | 61 |
| 60 bool TestHttpServer::WaitForConnectionsToClose() { | 62 bool TestHttpServer::WaitForConnectionsToClose() { |
| 61 return all_closed_event_.TimedWait(base::TimeDelta::FromSeconds(10)); | 63 return all_closed_event_.TimedWait(base::TimeDelta::FromSeconds(10)); |
| 62 } | 64 } |
| (...skipping 92 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 155 server_.reset(NULL); | 157 server_.reset(NULL); |
| 156 } | 158 } |
| 157 *success = server_.get(); | 159 *success = server_.get(); |
| 158 event->Signal(); | 160 event->Signal(); |
| 159 } | 161 } |
| 160 | 162 |
| 161 void TestHttpServer::StopOnServerThread(base::WaitableEvent* event) { | 163 void TestHttpServer::StopOnServerThread(base::WaitableEvent* event) { |
| 162 server_.reset(NULL); | 164 server_.reset(NULL); |
| 163 event->Signal(); | 165 event->Signal(); |
| 164 } | 166 } |
| OLD | NEW |