OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 <string> | 5 #include <string> |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/compiler_specific.h" | 8 #include "base/compiler_specific.h" |
9 #include "base/location.h" | 9 #include "base/location.h" |
10 #include "base/memory/ref_counted.h" | 10 #include "base/memory/ref_counted.h" |
11 #include "base/message_loop/message_loop.h" | 11 #include "base/message_loop/message_loop.h" |
12 #include "base/message_loop/message_loop_proxy.h" | 12 #include "base/message_loop/message_loop_proxy.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/synchronization/waitable_event.h" | 15 #include "base/synchronization/waitable_event.h" |
16 #include "base/threading/thread.h" | 16 #include "base/threading/thread.h" |
17 #include "chrome/test/chromedriver/net/net_util.h" | 17 #include "chrome/test/chromedriver/net/net_util.h" |
18 #include "chrome/test/chromedriver/net/url_request_context_getter.h" | 18 #include "chrome/test/chromedriver/net/url_request_context_getter.h" |
19 #include "net/base/ip_endpoint.h" | 19 #include "net/base/ip_endpoint.h" |
20 #include "net/base/net_errors.h" | 20 #include "net/base/net_errors.h" |
21 #include "net/server/http_server.h" | 21 #include "net/server/http_server.h" |
22 #include "net/server/http_server_request_info.h" | 22 #include "net/server/http_server_request_info.h" |
23 #include "net/socket/tcp_listen_socket.h" | 23 #include "net/socket/tcp_server_socket.h" |
24 #include "net/url_request/url_request_context_getter.h" | 24 #include "net/url_request/url_request_context_getter.h" |
25 #include "testing/gtest/include/gtest/gtest.h" | 25 #include "testing/gtest/include/gtest/gtest.h" |
26 | 26 |
27 namespace { | 27 namespace { |
28 | 28 |
29 class FetchUrlTest : public testing::Test, | 29 class FetchUrlTest : public testing::Test, |
30 public net::HttpServer::Delegate { | 30 public net::HttpServer::Delegate { |
31 public: | 31 public: |
32 FetchUrlTest() | 32 FetchUrlTest() |
33 : io_thread_("io"), | 33 : io_thread_("io"), |
(...skipping 13 matching lines...) Expand all Loading... |
47 virtual ~FetchUrlTest() { | 47 virtual ~FetchUrlTest() { |
48 base::WaitableEvent event(false, false); | 48 base::WaitableEvent event(false, false); |
49 io_thread_.message_loop_proxy()->PostTask( | 49 io_thread_.message_loop_proxy()->PostTask( |
50 FROM_HERE, | 50 FROM_HERE, |
51 base::Bind(&FetchUrlTest::DestroyServerOnIO, | 51 base::Bind(&FetchUrlTest::DestroyServerOnIO, |
52 base::Unretained(this), &event)); | 52 base::Unretained(this), &event)); |
53 event.Wait(); | 53 event.Wait(); |
54 } | 54 } |
55 | 55 |
56 void InitOnIO(base::WaitableEvent* event) { | 56 void InitOnIO(base::WaitableEvent* event) { |
57 net::TCPListenSocketFactory factory("127.0.0.1", 0); | 57 scoped_ptr<net::ServerSocket> server_socket( |
58 server_ = new net::HttpServer(factory, this); | 58 new net::TCPServerSocket(NULL, net::NetLog::Source())); |
| 59 server_socket->ListenWithAddressAndPort("127.0.0.1", 0, 1); |
| 60 server_.reset(new net::HttpServer(server_socket.Pass(), this)); |
59 net::IPEndPoint address; | 61 net::IPEndPoint address; |
60 CHECK_EQ(net::OK, server_->GetLocalAddress(&address)); | 62 CHECK_EQ(net::OK, server_->GetLocalAddress(&address)); |
61 server_url_ = base::StringPrintf("http://127.0.0.1:%d", address.port()); | 63 server_url_ = base::StringPrintf("http://127.0.0.1:%d", address.port()); |
62 event->Signal(); | 64 event->Signal(); |
63 } | 65 } |
64 | 66 |
65 void DestroyServerOnIO(base::WaitableEvent* event) { | 67 void DestroyServerOnIO(base::WaitableEvent* event) { |
66 server_ = NULL; | 68 server_.reset(NULL); |
67 event->Signal(); | 69 event->Signal(); |
68 } | 70 } |
69 | 71 |
70 // Overridden from net::HttpServer::Delegate: | 72 // Overridden from net::HttpServer::Delegate: |
71 virtual void OnHttpRequest(int connection_id, | 73 virtual void OnHttpRequest(int connection_id, |
72 const net::HttpServerRequestInfo& info) OVERRIDE { | 74 const net::HttpServerRequestInfo& info) OVERRIDE { |
73 switch (response_) { | 75 switch (response_) { |
74 case kSendHello: | 76 case kSendHello: |
75 server_->Send200(connection_id, "hello", "text/plain"); | 77 server_->Send200(connection_id, "hello", "text/plain"); |
76 break; | 78 break; |
77 case kSend404: | 79 case kSend404: |
78 server_->Send404(connection_id); | 80 server_->Send404(connection_id); |
79 break; | 81 break; |
80 case kClose: | 82 case kClose: |
81 // net::HttpServer doesn't allow us to close connection during callback. | 83 server_->Close(connection_id); |
82 base::MessageLoop::current()->PostTask( | |
83 FROM_HERE, | |
84 base::Bind(&net::HttpServer::Close, server_, connection_id)); | |
85 break; | 84 break; |
86 default: | 85 default: |
87 break; | 86 break; |
88 } | 87 } |
89 } | 88 } |
90 | 89 |
91 virtual void OnWebSocketRequest( | 90 virtual void OnWebSocketRequest( |
92 int connection_id, | 91 int connection_id, |
93 const net::HttpServerRequestInfo& info) OVERRIDE {} | 92 const net::HttpServerRequestInfo& info) OVERRIDE {} |
94 virtual void OnWebSocketMessage(int connection_id, | 93 virtual void OnWebSocketMessage(int connection_id, |
95 const std::string& data) OVERRIDE {} | 94 const std::string& data) OVERRIDE {} |
96 virtual void OnClose(int connection_id) OVERRIDE {} | 95 virtual void OnClose(int connection_id) OVERRIDE {} |
97 | 96 |
98 protected: | 97 protected: |
99 enum ServerResponse { | 98 enum ServerResponse { |
100 kSendHello = 0, | 99 kSendHello = 0, |
101 kSend404, | 100 kSend404, |
102 kClose, | 101 kClose, |
103 }; | 102 }; |
104 | 103 |
105 base::Thread io_thread_; | 104 base::Thread io_thread_; |
106 ServerResponse response_; | 105 ServerResponse response_; |
107 scoped_refptr<net::HttpServer> server_; | 106 scoped_ptr<net::HttpServer> server_; |
108 scoped_refptr<URLRequestContextGetter> context_getter_; | 107 scoped_refptr<URLRequestContextGetter> context_getter_; |
109 std::string server_url_; | 108 std::string server_url_; |
110 }; | 109 }; |
111 | 110 |
112 } // namespace | 111 } // namespace |
113 | 112 |
114 TEST_F(FetchUrlTest, Http200) { | 113 TEST_F(FetchUrlTest, Http200) { |
115 std::string response("stuff"); | 114 std::string response("stuff"); |
116 ASSERT_TRUE(FetchUrl(server_url_, context_getter_.get(), &response)); | 115 ASSERT_TRUE(FetchUrl(server_url_, context_getter_.get(), &response)); |
117 ASSERT_STREQ("hello", response.c_str()); | 116 ASSERT_STREQ("hello", response.c_str()); |
(...skipping 12 matching lines...) Expand all Loading... |
130 ASSERT_FALSE(FetchUrl(server_url_, context_getter_.get(), &response)); | 129 ASSERT_FALSE(FetchUrl(server_url_, context_getter_.get(), &response)); |
131 ASSERT_STREQ("stuff", response.c_str()); | 130 ASSERT_STREQ("stuff", response.c_str()); |
132 } | 131 } |
133 | 132 |
134 TEST_F(FetchUrlTest, NoServer) { | 133 TEST_F(FetchUrlTest, NoServer) { |
135 std::string response("stuff"); | 134 std::string response("stuff"); |
136 ASSERT_FALSE( | 135 ASSERT_FALSE( |
137 FetchUrl("http://localhost:33333", context_getter_.get(), &response)); | 136 FetchUrl("http://localhost:33333", context_getter_.get(), &response)); |
138 ASSERT_STREQ("stuff", response.c_str()); | 137 ASSERT_STREQ("stuff", response.c_str()); |
139 } | 138 } |
OLD | NEW |