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

Side by Side Diff: content/browser/resolve_proxy_msg_helper_unittest.cc

Issue 7791005: Stop using the default profile's proxy service for plugin proxy requests, and instead use the ass... (Closed) Base URL: svn://chrome-svn/chrome/trunk/src/
Patch Set: fix test Created 9 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 | Annotate | Revision Log
« no previous file with comments | « content/browser/resolve_proxy_msg_helper.cc ('k') | content/common/child_process_messages.h » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. 1 // Copyright (c) 2011 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 "content/browser/resolve_proxy_msg_helper.h" 5 #include "content/browser/resolve_proxy_msg_helper.h"
6 6
7 #include "content/common/child_process_messages.h" 7 #include "content/common/view_messages.h"
8 #include "ipc/ipc_test_sink.h" 8 #include "ipc/ipc_test_sink.h"
9 #include "net/base/net_errors.h" 9 #include "net/base/net_errors.h"
10 #include "net/proxy/mock_proxy_resolver.h" 10 #include "net/proxy/mock_proxy_resolver.h"
11 #include "net/proxy/proxy_config_service.h" 11 #include "net/proxy/proxy_config_service.h"
12 #include "net/proxy/proxy_service.h" 12 #include "net/proxy/proxy_service.h"
13 #include "testing/gtest/include/gtest/gtest.h" 13 #include "testing/gtest/include/gtest/gtest.h"
14 14
15 // This ProxyConfigService always returns "http://pac" as the PAC url to use. 15 // This ProxyConfigService always returns "http://pac" as the PAC url to use.
16 class MockProxyConfigService : public net::ProxyConfigService { 16 class MockProxyConfigService : public net::ProxyConfigService {
17 public: 17 public:
18 virtual void AddObserver(Observer* observer) {} 18 virtual void AddObserver(Observer* observer) {}
19 virtual void RemoveObserver(Observer* observer) {} 19 virtual void RemoveObserver(Observer* observer) {}
20 virtual ConfigAvailability GetLatestProxyConfig(net::ProxyConfig* results) { 20 virtual ConfigAvailability GetLatestProxyConfig(net::ProxyConfig* results) {
21 *results = net::ProxyConfig::CreateFromCustomPacURL(GURL("http://pac")); 21 *results = net::ProxyConfig::CreateFromCustomPacURL(GURL("http://pac"));
22 return CONFIG_VALID; 22 return CONFIG_VALID;
23 } 23 }
24 }; 24 };
25 25
26 class ResolveProxyMsgHelperTest : public testing::Test, 26 class ResolveProxyMsgHelperTest : public testing::Test,
27 public IPC::Channel::Listener { 27 public IPC::Channel::Listener {
28 public: 28 public:
29 struct PendingResult { 29 struct PendingResult {
30 PendingResult(int error_code, 30 PendingResult(bool result,
31 const std::string& proxy_list) 31 const std::string& proxy_list)
32 : error_code(error_code), proxy_list(proxy_list) { 32 : result(result), proxy_list(proxy_list) {
33 } 33 }
34 34
35 int error_code; 35 bool result;
36 std::string proxy_list; 36 std::string proxy_list;
37 }; 37 };
38 38
39 ResolveProxyMsgHelperTest() 39 ResolveProxyMsgHelperTest()
40 : resolver_(new net::MockAsyncProxyResolver), 40 : resolver_(new net::MockAsyncProxyResolver),
41 service_(new net::ProxyService( 41 service_(new net::ProxyService(
42 new MockProxyConfigService, resolver_, NULL)), 42 new MockProxyConfigService, resolver_, NULL)),
43 helper_(new ResolveProxyMsgHelper(service_.get())), 43 helper_(new ResolveProxyMsgHelper(service_.get())),
44 message_loop_(MessageLoop::TYPE_IO), 44 message_loop_(MessageLoop::TYPE_IO),
45 io_thread_(BrowserThread::IO, &message_loop_) { 45 io_thread_(BrowserThread::IO, &message_loop_) {
46 test_sink_.AddFilter(this); 46 test_sink_.AddFilter(this);
47 helper_->OnFilterAdded(&test_sink_); 47 helper_->OnFilterAdded(&test_sink_);
48 } 48 }
49 49
50 protected: 50 protected:
51 const PendingResult* pending_result() const { return pending_result_.get(); } 51 const PendingResult* pending_result() const { return pending_result_.get(); }
52 52
53 void clear_pending_result() { 53 void clear_pending_result() {
54 pending_result_.reset(); 54 pending_result_.reset();
55 } 55 }
56 56
57 IPC::Message* GenerateReply() { 57 IPC::Message* GenerateReply() {
58 int temp_int; 58 bool temp_bool;
59 std::string temp_string; 59 std::string temp_string;
60 ChildProcessHostMsg_ResolveProxy message(GURL(), &temp_int, &temp_string); 60 ViewHostMsg_ResolveProxy message(GURL(), &temp_bool, &temp_string);
61 return IPC::SyncMessage::GenerateReply(&message); 61 return IPC::SyncMessage::GenerateReply(&message);
62 } 62 }
63 63
64 net::MockAsyncProxyResolver* resolver_; 64 net::MockAsyncProxyResolver* resolver_;
65 scoped_ptr<net::ProxyService> service_; 65 scoped_ptr<net::ProxyService> service_;
66 scoped_refptr<ResolveProxyMsgHelper> helper_; 66 scoped_refptr<ResolveProxyMsgHelper> helper_;
67 scoped_ptr<PendingResult> pending_result_; 67 scoped_ptr<PendingResult> pending_result_;
68 68
69 private: 69 private:
70 virtual bool OnMessageReceived(const IPC::Message& msg) { 70 virtual bool OnMessageReceived(const IPC::Message& msg) {
71 TupleTypes<ChildProcessHostMsg_ResolveProxy::ReplyParam>::ValueTuple 71 TupleTypes<ViewHostMsg_ResolveProxy::ReplyParam>::ValueTuple reply_data;
72 reply_data; 72 EXPECT_TRUE(ViewHostMsg_ResolveProxy::ReadReplyParam(&msg, &reply_data));
73 EXPECT_TRUE(
74 ChildProcessHostMsg_ResolveProxy::ReadReplyParam(&msg, &reply_data));
75 DCHECK(!pending_result_.get()); 73 DCHECK(!pending_result_.get());
76 pending_result_.reset(new PendingResult(reply_data.a, reply_data.b)); 74 pending_result_.reset(new PendingResult(reply_data.a, reply_data.b));
77 test_sink_.ClearMessages(); 75 test_sink_.ClearMessages();
78 return true; 76 return true;
79 } 77 }
80 78
81 MessageLoop message_loop_; 79 MessageLoop message_loop_;
82 BrowserThread io_thread_; 80 BrowserThread io_thread_;
83 IPC::TestSink test_sink_; 81 IPC::TestSink test_sink_;
84 }; 82 };
(...skipping 16 matching lines...) Expand all
101 99
102 // Finish ProxyService's initialization. 100 // Finish ProxyService's initialization.
103 resolver_->pending_set_pac_script_request()->CompleteNow(net::OK); 101 resolver_->pending_set_pac_script_request()->CompleteNow(net::OK);
104 102
105 ASSERT_EQ(1u, resolver_->pending_requests().size()); 103 ASSERT_EQ(1u, resolver_->pending_requests().size());
106 EXPECT_EQ(url1, resolver_->pending_requests()[0]->url()); 104 EXPECT_EQ(url1, resolver_->pending_requests()[0]->url());
107 resolver_->pending_requests()[0]->results()->UseNamedProxy("result1:80"); 105 resolver_->pending_requests()[0]->results()->UseNamedProxy("result1:80");
108 resolver_->pending_requests()[0]->CompleteNow(net::OK); 106 resolver_->pending_requests()[0]->CompleteNow(net::OK);
109 107
110 // Check result. 108 // Check result.
111 EXPECT_EQ(net::OK, pending_result()->error_code); 109 EXPECT_EQ(true, pending_result()->result);
112 EXPECT_EQ("PROXY result1:80", pending_result()->proxy_list); 110 EXPECT_EQ("PROXY result1:80", pending_result()->proxy_list);
113 clear_pending_result(); 111 clear_pending_result();
114 112
115 helper_->OnResolveProxy(url2, msg2); 113 helper_->OnResolveProxy(url2, msg2);
116 114
117 ASSERT_EQ(1u, resolver_->pending_requests().size()); 115 ASSERT_EQ(1u, resolver_->pending_requests().size());
118 EXPECT_EQ(url2, resolver_->pending_requests()[0]->url()); 116 EXPECT_EQ(url2, resolver_->pending_requests()[0]->url());
119 resolver_->pending_requests()[0]->results()->UseNamedProxy("result2:80"); 117 resolver_->pending_requests()[0]->results()->UseNamedProxy("result2:80");
120 resolver_->pending_requests()[0]->CompleteNow(net::OK); 118 resolver_->pending_requests()[0]->CompleteNow(net::OK);
121 119
122 // Check result. 120 // Check result.
123 EXPECT_EQ(net::OK, pending_result()->error_code); 121 EXPECT_EQ(true, pending_result()->result);
124 EXPECT_EQ("PROXY result2:80", pending_result()->proxy_list); 122 EXPECT_EQ("PROXY result2:80", pending_result()->proxy_list);
125 clear_pending_result(); 123 clear_pending_result();
126 124
127 helper_->OnResolveProxy(url3, msg3); 125 helper_->OnResolveProxy(url3, msg3);
128 126
129 ASSERT_EQ(1u, resolver_->pending_requests().size()); 127 ASSERT_EQ(1u, resolver_->pending_requests().size());
130 EXPECT_EQ(url3, resolver_->pending_requests()[0]->url()); 128 EXPECT_EQ(url3, resolver_->pending_requests()[0]->url());
131 resolver_->pending_requests()[0]->results()->UseNamedProxy("result3:80"); 129 resolver_->pending_requests()[0]->results()->UseNamedProxy("result3:80");
132 resolver_->pending_requests()[0]->CompleteNow(net::OK); 130 resolver_->pending_requests()[0]->CompleteNow(net::OK);
133 131
134 // Check result. 132 // Check result.
135 EXPECT_EQ(net::OK, pending_result()->error_code); 133 EXPECT_EQ(true, pending_result()->result);
136 EXPECT_EQ("PROXY result3:80", pending_result()->proxy_list); 134 EXPECT_EQ("PROXY result3:80", pending_result()->proxy_list);
137 clear_pending_result(); 135 clear_pending_result();
138 } 136 }
139 137
140 // Issue a request while one is already in progress -- should be queued. 138 // Issue a request while one is already in progress -- should be queued.
141 TEST_F(ResolveProxyMsgHelperTest, QueueRequests) { 139 TEST_F(ResolveProxyMsgHelperTest, QueueRequests) {
142 GURL url1("http://www.google1.com/"); 140 GURL url1("http://www.google1.com/");
143 GURL url2("http://www.google2.com/"); 141 GURL url2("http://www.google2.com/");
144 GURL url3("http://www.google3.com/"); 142 GURL url3("http://www.google3.com/");
145 143
(...skipping 14 matching lines...) Expand all
160 158
161 // ResolveProxyHelper only keeps 1 request outstanding in ProxyService 159 // ResolveProxyHelper only keeps 1 request outstanding in ProxyService
162 // at a time. 160 // at a time.
163 ASSERT_EQ(1u, resolver_->pending_requests().size()); 161 ASSERT_EQ(1u, resolver_->pending_requests().size());
164 EXPECT_EQ(url1, resolver_->pending_requests()[0]->url()); 162 EXPECT_EQ(url1, resolver_->pending_requests()[0]->url());
165 163
166 resolver_->pending_requests()[0]->results()->UseNamedProxy("result1:80"); 164 resolver_->pending_requests()[0]->results()->UseNamedProxy("result1:80");
167 resolver_->pending_requests()[0]->CompleteNow(net::OK); 165 resolver_->pending_requests()[0]->CompleteNow(net::OK);
168 166
169 // Check result. 167 // Check result.
170 EXPECT_EQ(net::OK, pending_result()->error_code); 168 EXPECT_EQ(true, pending_result()->result);
171 EXPECT_EQ("PROXY result1:80", pending_result()->proxy_list); 169 EXPECT_EQ("PROXY result1:80", pending_result()->proxy_list);
172 clear_pending_result(); 170 clear_pending_result();
173 171
174 ASSERT_EQ(1u, resolver_->pending_requests().size()); 172 ASSERT_EQ(1u, resolver_->pending_requests().size());
175 EXPECT_EQ(url2, resolver_->pending_requests()[0]->url()); 173 EXPECT_EQ(url2, resolver_->pending_requests()[0]->url());
176 174
177 resolver_->pending_requests()[0]->results()->UseNamedProxy("result2:80"); 175 resolver_->pending_requests()[0]->results()->UseNamedProxy("result2:80");
178 resolver_->pending_requests()[0]->CompleteNow(net::OK); 176 resolver_->pending_requests()[0]->CompleteNow(net::OK);
179 177
180 // Check result. 178 // Check result.
181 EXPECT_EQ(net::OK, pending_result()->error_code); 179 EXPECT_EQ(true, pending_result()->result);
182 EXPECT_EQ("PROXY result2:80", pending_result()->proxy_list); 180 EXPECT_EQ("PROXY result2:80", pending_result()->proxy_list);
183 clear_pending_result(); 181 clear_pending_result();
184 182
185 ASSERT_EQ(1u, resolver_->pending_requests().size()); 183 ASSERT_EQ(1u, resolver_->pending_requests().size());
186 EXPECT_EQ(url3, resolver_->pending_requests()[0]->url()); 184 EXPECT_EQ(url3, resolver_->pending_requests()[0]->url());
187 185
188 resolver_->pending_requests()[0]->results()->UseNamedProxy("result3:80"); 186 resolver_->pending_requests()[0]->results()->UseNamedProxy("result3:80");
189 resolver_->pending_requests()[0]->CompleteNow(net::OK); 187 resolver_->pending_requests()[0]->CompleteNow(net::OK);
190 188
191 // Check result. 189 // Check result.
192 EXPECT_EQ(net::OK, pending_result()->error_code); 190 EXPECT_EQ(true, pending_result()->result);
193 EXPECT_EQ("PROXY result3:80", pending_result()->proxy_list); 191 EXPECT_EQ("PROXY result3:80", pending_result()->proxy_list);
194 clear_pending_result(); 192 clear_pending_result();
195 } 193 }
196 194
197 // Delete the helper while a request is in progress, and others are pending. 195 // Delete the helper while a request is in progress, and others are pending.
198 TEST_F(ResolveProxyMsgHelperTest, CancelPendingRequests) { 196 TEST_F(ResolveProxyMsgHelperTest, CancelPendingRequests) {
199 GURL url1("http://www.google1.com/"); 197 GURL url1("http://www.google1.com/");
200 GURL url2("http://www.google2.com/"); 198 GURL url2("http://www.google2.com/");
201 GURL url3("http://www.google3.com/"); 199 GURL url3("http://www.google3.com/");
202 200
(...skipping 24 matching lines...) Expand all
227 225
228 // The pending requests sent to the proxy resolver should have been cancelled. 226 // The pending requests sent to the proxy resolver should have been cancelled.
229 227
230 EXPECT_EQ(0u, resolver_->pending_requests().size()); 228 EXPECT_EQ(0u, resolver_->pending_requests().size());
231 229
232 EXPECT_TRUE(pending_result() == NULL); 230 EXPECT_TRUE(pending_result() == NULL);
233 231
234 // It should also be the case that msg1, msg2, msg3 were deleted by the 232 // It should also be the case that msg1, msg2, msg3 were deleted by the
235 // cancellation. (Else will show up as a leak in Valgrind). 233 // cancellation. (Else will show up as a leak in Valgrind).
236 } 234 }
OLDNEW
« no previous file with comments | « content/browser/resolve_proxy_msg_helper.cc ('k') | content/common/child_process_messages.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698