Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(304)

Side by Side Diff: chrome/test/chromedriver/net/net_util_unittest.cc

Issue 497223003: Revert of Replace StreamListenSocket with StreamSocket in HttpServer. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Created 6 years, 3 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
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
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
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 }
OLDNEW
« no previous file with comments | « chrome/browser/devtools/remote_debugging_server.cc ('k') | chrome/test/chromedriver/net/test_http_server.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698