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 "chrome/browser/ui/webui/net_internals/net_internals_ui_browsertest.h" | 5 #include "chrome/browser/ui/webui/net_internals/net_internals_ui_browsertest.h" |
6 | 6 |
7 #include <utility> | 7 #include <utility> |
8 | 8 |
9 #include "base/bind.h" | 9 #include "base/bind.h" |
10 #include "base/bind_helpers.h" | 10 #include "base/bind_helpers.h" |
(...skipping 86 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
97 //////////////////////////////////////////////////////////////////////////////// | 97 //////////////////////////////////////////////////////////////////////////////// |
98 | 98 |
99 // Class to handle messages from the renderer needed by certain tests. | 99 // Class to handle messages from the renderer needed by certain tests. |
100 class NetInternalsTest::MessageHandler : public content::WebUIMessageHandler { | 100 class NetInternalsTest::MessageHandler : public content::WebUIMessageHandler { |
101 public: | 101 public: |
102 explicit MessageHandler(NetInternalsTest* net_internals_test); | 102 explicit MessageHandler(NetInternalsTest* net_internals_test); |
103 | 103 |
104 private: | 104 private: |
105 void RegisterMessages() override; | 105 void RegisterMessages() override; |
106 | 106 |
107 void RegisterMessage(const std::string& message, | |
108 const content::WebUI::MessageCallback& handler); | |
109 | |
110 void HandleMessage(const content::WebUI::MessageCallback& handler, | |
111 const base::ListValue* data); | |
112 | |
113 // Runs NetInternalsTest.callback with the given value. | 107 // Runs NetInternalsTest.callback with the given value. |
114 void RunJavascriptCallback(base::Value* value); | 108 void RunJavascriptCallback(base::Value* value); |
115 | 109 |
116 // Takes a string and provides the corresponding URL from the test server, | 110 // Takes a string and provides the corresponding URL from the test server, |
117 // which must already have been started. | 111 // which must already have been started. |
118 void GetTestServerURL(const base::ListValue* list_value); | 112 void GetTestServerURL(const base::ListValue* list_value); |
119 | 113 |
120 // Called on UI thread. Adds an entry to the cache for the specified | 114 // Called on UI thread. Adds an entry to the cache for the specified |
121 // hostname by posting a task to the IO thread. Takes the host name, | 115 // hostname by posting a task to the IO thread. Takes the host name, |
122 // ip address, net error code, and expiration time in days from now | 116 // ip address, net error code, and expiration time in days from now |
(...skipping 38 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
161 DISALLOW_COPY_AND_ASSIGN(MessageHandler); | 155 DISALLOW_COPY_AND_ASSIGN(MessageHandler); |
162 }; | 156 }; |
163 | 157 |
164 NetInternalsTest::MessageHandler::MessageHandler( | 158 NetInternalsTest::MessageHandler::MessageHandler( |
165 NetInternalsTest* net_internals_test) | 159 NetInternalsTest* net_internals_test) |
166 : net_internals_test_(net_internals_test), | 160 : net_internals_test_(net_internals_test), |
167 incognito_browser_(NULL) { | 161 incognito_browser_(NULL) { |
168 } | 162 } |
169 | 163 |
170 void NetInternalsTest::MessageHandler::RegisterMessages() { | 164 void NetInternalsTest::MessageHandler::RegisterMessages() { |
171 RegisterMessage( | 165 web_ui()->RegisterMessageCallback("getTestServerURL", |
172 "getTestServerURL", | |
173 base::Bind(&NetInternalsTest::MessageHandler::GetTestServerURL, | 166 base::Bind(&NetInternalsTest::MessageHandler::GetTestServerURL, |
174 base::Unretained(this))); | 167 base::Unretained(this))); |
175 RegisterMessage("addCacheEntry", | 168 web_ui()->RegisterMessageCallback("addCacheEntry", |
176 base::Bind(&NetInternalsTest::MessageHandler::AddCacheEntry, | 169 base::Bind(&NetInternalsTest::MessageHandler::AddCacheEntry, |
177 base::Unretained(this))); | 170 base::Unretained(this))); |
178 RegisterMessage("changeNetwork", | 171 web_ui()->RegisterMessageCallback( |
179 base::Bind(&NetInternalsTest::MessageHandler::ChangeNetwork, | 172 "changeNetwork", |
180 base::Unretained(this))); | 173 base::Bind(&NetInternalsTest::MessageHandler::ChangeNetwork, |
181 RegisterMessage("loadPage", | 174 base::Unretained(this))); |
182 base::Bind(&NetInternalsTest::MessageHandler::LoadPage, | 175 web_ui()->RegisterMessageCallback("loadPage", |
183 base::Unretained(this))); | 176 base::Bind(&NetInternalsTest::MessageHandler::LoadPage, |
184 RegisterMessage("prerenderPage", | 177 base::Unretained(this))); |
185 base::Bind(&NetInternalsTest::MessageHandler::PrerenderPage, | 178 web_ui()->RegisterMessageCallback("prerenderPage", |
186 base::Unretained(this))); | 179 base::Bind(&NetInternalsTest::MessageHandler::PrerenderPage, |
187 RegisterMessage( | 180 base::Unretained(this))); |
188 "navigateToPrerender", | 181 web_ui()->RegisterMessageCallback("navigateToPrerender", |
189 base::Bind(&NetInternalsTest::MessageHandler::NavigateToPrerender, | 182 base::Bind(&NetInternalsTest::MessageHandler::NavigateToPrerender, |
190 base::Unretained(this))); | 183 base::Unretained(this))); |
191 RegisterMessage( | 184 web_ui()->RegisterMessageCallback("createIncognitoBrowser", |
192 "createIncognitoBrowser", | |
193 base::Bind(&NetInternalsTest::MessageHandler::CreateIncognitoBrowser, | 185 base::Bind(&NetInternalsTest::MessageHandler::CreateIncognitoBrowser, |
194 base::Unretained(this))); | 186 base::Unretained(this))); |
195 RegisterMessage( | 187 web_ui()->RegisterMessageCallback("closeIncognitoBrowser", |
196 "closeIncognitoBrowser", | |
197 base::Bind(&NetInternalsTest::MessageHandler::CloseIncognitoBrowser, | 188 base::Bind(&NetInternalsTest::MessageHandler::CloseIncognitoBrowser, |
198 base::Unretained(this))); | 189 base::Unretained(this))); |
199 RegisterMessage( | 190 web_ui()->RegisterMessageCallback("getNetLogFileContents", |
200 "getNetLogFileContents", | 191 base::Bind( |
201 base::Bind(&NetInternalsTest::MessageHandler::GetNetLogFileContents, | 192 &NetInternalsTest::MessageHandler::GetNetLogFileContents, |
202 base::Unretained(this))); | 193 base::Unretained(this))); |
203 RegisterMessage( | 194 web_ui()->RegisterMessageCallback("enableDataReductionProxy", |
204 "enableDataReductionProxy", | 195 base::Bind( |
205 base::Bind(&NetInternalsTest::MessageHandler::EnableDataReductionProxy, | 196 &NetInternalsTest::MessageHandler::EnableDataReductionProxy, |
206 base::Unretained(this))); | 197 base::Unretained(this))); |
207 } | |
208 | |
209 void NetInternalsTest::MessageHandler::RegisterMessage( | |
210 const std::string& message, | |
211 const content::WebUI::MessageCallback& handler) { | |
212 web_ui()->RegisterMessageCallback( | |
213 message, base::Bind(&NetInternalsTest::MessageHandler::HandleMessage, | |
214 base::Unretained(this), handler)); | |
215 } | |
216 | |
217 void NetInternalsTest::MessageHandler::HandleMessage( | |
218 const content::WebUI::MessageCallback& handler, | |
219 const base::ListValue* data) { | |
220 // The handler might run a nested loop to wait for something. | |
221 base::MessageLoop::ScopedNestableTaskAllower nestable_task_allower( | |
222 base::MessageLoop::current()); | |
223 handler.Run(data); | |
224 } | 198 } |
225 | 199 |
226 void NetInternalsTest::MessageHandler::RunJavascriptCallback( | 200 void NetInternalsTest::MessageHandler::RunJavascriptCallback( |
227 base::Value* value) { | 201 base::Value* value) { |
228 web_ui()->CallJavascriptFunctionUnsafe("NetInternalsTest.callback", *value); | 202 web_ui()->CallJavascriptFunctionUnsafe("NetInternalsTest.callback", *value); |
229 } | 203 } |
230 | 204 |
231 void NetInternalsTest::MessageHandler::GetTestServerURL( | 205 void NetInternalsTest::MessageHandler::GetTestServerURL( |
232 const base::ListValue* list_value) { | 206 const base::ListValue* list_value) { |
233 ASSERT_TRUE(net_internals_test_->StartTestServer()); | 207 ASSERT_TRUE(net_internals_test_->StartTestServer()); |
(...skipping 168 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
402 if (test_server_started_) | 376 if (test_server_started_) |
403 return true; | 377 return true; |
404 test_server_started_ = embedded_test_server()->Start(); | 378 test_server_started_ = embedded_test_server()->Start(); |
405 | 379 |
406 // Sample domain for SDCH-view test. Dictionaries for localhost/127.0.0.1 | 380 // Sample domain for SDCH-view test. Dictionaries for localhost/127.0.0.1 |
407 // are forbidden. | 381 // are forbidden. |
408 host_resolver()->AddRule("testdomain.com", "127.0.0.1"); | 382 host_resolver()->AddRule("testdomain.com", "127.0.0.1"); |
409 host_resolver()->AddRule("sub.testdomain.com", "127.0.0.1"); | 383 host_resolver()->AddRule("sub.testdomain.com", "127.0.0.1"); |
410 return test_server_started_; | 384 return test_server_started_; |
411 } | 385 } |
OLD | NEW |