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_server_socket.h" | 23 #include "net/socket/tcp_listen_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 scoped_ptr<net::ServerSocket> server_socket( | 57 net::TCPListenSocketFactory factory("127.0.0.1", 0); |
58 new net::TCPServerSocket(NULL, net::NetLog::Source())); | 58 server_ = new net::HttpServer(factory, this); |
59 server_socket->ListenWithAddressAndPort("127.0.0.1", 0, 1); | |
60 server_.reset(new net::HttpServer(server_socket.Pass(), this)); | |
61 net::IPEndPoint address; | 59 net::IPEndPoint address; |
62 CHECK_EQ(net::OK, server_->GetLocalAddress(&address)); | 60 CHECK_EQ(net::OK, server_->GetLocalAddress(&address)); |
63 server_url_ = base::StringPrintf("http://127.0.0.1:%d", address.port()); | 61 server_url_ = base::StringPrintf("http://127.0.0.1:%d", address.port()); |
64 event->Signal(); | 62 event->Signal(); |
65 } | 63 } |
66 | 64 |
67 void DestroyServerOnIO(base::WaitableEvent* event) { | 65 void DestroyServerOnIO(base::WaitableEvent* event) { |
68 server_.reset(NULL); | 66 server_ = NULL; |
69 event->Signal(); | 67 event->Signal(); |
70 } | 68 } |
71 | 69 |
72 // Overridden from net::HttpServer::Delegate: | 70 // Overridden from net::HttpServer::Delegate: |
73 virtual void OnHttpRequest(int connection_id, | 71 virtual void OnHttpRequest(int connection_id, |
74 const net::HttpServerRequestInfo& info) OVERRIDE { | 72 const net::HttpServerRequestInfo& info) OVERRIDE { |
75 switch (response_) { | 73 switch (response_) { |
76 case kSendHello: | 74 case kSendHello: |
77 server_->Send200(connection_id, "hello", "text/plain"); | 75 server_->Send200(connection_id, "hello", "text/plain"); |
78 break; | 76 break; |
79 case kSend404: | 77 case kSend404: |
80 server_->Send404(connection_id); | 78 server_->Send404(connection_id); |
81 break; | 79 break; |
82 case kClose: | 80 case kClose: |
83 server_->Close(connection_id); | 81 // net::HttpServer doesn't allow us to close connection during callback. |
| 82 base::MessageLoop::current()->PostTask( |
| 83 FROM_HERE, |
| 84 base::Bind(&net::HttpServer::Close, server_, connection_id)); |
84 break; | 85 break; |
85 default: | 86 default: |
86 break; | 87 break; |
87 } | 88 } |
88 } | 89 } |
89 | 90 |
90 virtual void OnWebSocketRequest( | 91 virtual void OnWebSocketRequest( |
91 int connection_id, | 92 int connection_id, |
92 const net::HttpServerRequestInfo& info) OVERRIDE {} | 93 const net::HttpServerRequestInfo& info) OVERRIDE {} |
93 virtual void OnWebSocketMessage(int connection_id, | 94 virtual void OnWebSocketMessage(int connection_id, |
94 const std::string& data) OVERRIDE {} | 95 const std::string& data) OVERRIDE {} |
95 virtual void OnClose(int connection_id) OVERRIDE {} | 96 virtual void OnClose(int connection_id) OVERRIDE {} |
96 | 97 |
97 protected: | 98 protected: |
98 enum ServerResponse { | 99 enum ServerResponse { |
99 kSendHello = 0, | 100 kSendHello = 0, |
100 kSend404, | 101 kSend404, |
101 kClose, | 102 kClose, |
102 }; | 103 }; |
103 | 104 |
104 base::Thread io_thread_; | 105 base::Thread io_thread_; |
105 ServerResponse response_; | 106 ServerResponse response_; |
106 scoped_ptr<net::HttpServer> server_; | 107 scoped_refptr<net::HttpServer> server_; |
107 scoped_refptr<URLRequestContextGetter> context_getter_; | 108 scoped_refptr<URLRequestContextGetter> context_getter_; |
108 std::string server_url_; | 109 std::string server_url_; |
109 }; | 110 }; |
110 | 111 |
111 } // namespace | 112 } // namespace |
112 | 113 |
113 TEST_F(FetchUrlTest, Http200) { | 114 TEST_F(FetchUrlTest, Http200) { |
114 std::string response("stuff"); | 115 std::string response("stuff"); |
115 ASSERT_TRUE(FetchUrl(server_url_, context_getter_.get(), &response)); | 116 ASSERT_TRUE(FetchUrl(server_url_, context_getter_.get(), &response)); |
116 ASSERT_STREQ("hello", response.c_str()); | 117 ASSERT_STREQ("hello", response.c_str()); |
(...skipping 12 matching lines...) Expand all Loading... |
129 ASSERT_FALSE(FetchUrl(server_url_, context_getter_.get(), &response)); | 130 ASSERT_FALSE(FetchUrl(server_url_, context_getter_.get(), &response)); |
130 ASSERT_STREQ("stuff", response.c_str()); | 131 ASSERT_STREQ("stuff", response.c_str()); |
131 } | 132 } |
132 | 133 |
133 TEST_F(FetchUrlTest, NoServer) { | 134 TEST_F(FetchUrlTest, NoServer) { |
134 std::string response("stuff"); | 135 std::string response("stuff"); |
135 ASSERT_FALSE( | 136 ASSERT_FALSE( |
136 FetchUrl("http://localhost:33333", context_getter_.get(), &response)); | 137 FetchUrl("http://localhost:33333", context_getter_.get(), &response)); |
137 ASSERT_STREQ("stuff", response.c_str()); | 138 ASSERT_STREQ("stuff", response.c_str()); |
138 } | 139 } |
OLD | NEW |