| OLD | NEW |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 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 "base/command_line.h" | 5 #include "base/command_line.h" |
| 6 #include "chrome/browser/browsing_data/browsing_data_helper.h" |
| 7 #include "chrome/browser/browsing_data/browsing_data_remover.h" |
| 8 #include "chrome/browser/browsing_data/browsing_data_remover_factory.h" |
| 6 #include "chrome/browser/predictors/resource_prefetch_predictor.h" | 9 #include "chrome/browser/predictors/resource_prefetch_predictor.h" |
| 7 #include "chrome/browser/predictors/resource_prefetch_predictor_factory.h" | 10 #include "chrome/browser/predictors/resource_prefetch_predictor_factory.h" |
| 8 #include "chrome/browser/predictors/resource_prefetch_predictor_test_util.h" | 11 #include "chrome/browser/predictors/resource_prefetch_predictor_test_util.h" |
| 9 #include "chrome/browser/profiles/profile.h" | 12 #include "chrome/browser/profiles/profile.h" |
| 10 #include "chrome/browser/ui/browser.h" | 13 #include "chrome/browser/ui/browser.h" |
| 11 #include "chrome/browser/ui/browser_commands.h" | |
| 12 #include "chrome/browser/ui/tabs/tab_strip_model.h" | 14 #include "chrome/browser/ui/tabs/tab_strip_model.h" |
| 13 #include "chrome/common/chrome_switches.h" | 15 #include "chrome/common/chrome_switches.h" |
| 14 #include "chrome/test/base/in_process_browser_test.h" | 16 #include "chrome/test/base/in_process_browser_test.h" |
| 15 #include "chrome/test/base/ui_test_utils.h" | 17 #include "chrome/test/base/ui_test_utils.h" |
| 16 #include "content/public/browser/render_frame_host.h" | 18 #include "content/public/browser/render_frame_host.h" |
| 17 #include "content/public/browser/render_process_host.h" | 19 #include "content/public/browser/render_process_host.h" |
| 18 #include "net/test/embedded_test_server/http_request.h" | 20 #include "net/test/embedded_test_server/http_request.h" |
| 19 #include "net/test/embedded_test_server/http_response.h" | 21 #include "net/test/embedded_test_server/http_response.h" |
| 20 #include "testing/gmock/include/gmock/gmock.h" | 22 #include "testing/gmock/include/gmock/gmock.h" |
| 21 #include "testing/gtest/include/gtest/gtest.h" | 23 #include "testing/gtest/include/gtest/gtest.h" |
| (...skipping 63 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 85 void OnPredictorInitialized() override { run_loop_.Quit(); } | 87 void OnPredictorInitialized() override { run_loop_.Quit(); } |
| 86 | 88 |
| 87 void Wait() { run_loop_.Run(); } | 89 void Wait() { run_loop_.Run(); } |
| 88 | 90 |
| 89 private: | 91 private: |
| 90 base::RunLoop run_loop_; | 92 base::RunLoop run_loop_; |
| 91 | 93 |
| 92 DISALLOW_COPY_AND_ASSIGN(InitializationObserver); | 94 DISALLOW_COPY_AND_ASSIGN(InitializationObserver); |
| 93 }; | 95 }; |
| 94 | 96 |
| 97 class BrowsingDataRemoverObserver : public BrowsingDataRemover::Observer { |
| 98 public: |
| 99 explicit BrowsingDataRemoverObserver(BrowsingDataRemover* remover) |
| 100 : remover_(remover) { |
| 101 remover_->AddObserver(this); |
| 102 } |
| 103 ~BrowsingDataRemoverObserver() override { remover_->RemoveObserver(this); } |
| 104 |
| 105 void OnBrowsingDataRemoverDone() override { run_loop_.Quit(); } |
| 106 |
| 107 void Wait() { run_loop_.Run(); } |
| 108 |
| 109 private: |
| 110 BrowsingDataRemover* remover_; |
| 111 base::RunLoop run_loop_; |
| 112 |
| 113 DISALLOW_COPY_AND_ASSIGN(BrowsingDataRemoverObserver); |
| 114 }; |
| 115 |
| 95 using PageRequestSummary = ResourcePrefetchPredictor::PageRequestSummary; | 116 using PageRequestSummary = ResourcePrefetchPredictor::PageRequestSummary; |
| 96 using URLRequestSummary = ResourcePrefetchPredictor::URLRequestSummary; | 117 using URLRequestSummary = ResourcePrefetchPredictor::URLRequestSummary; |
| 97 | 118 |
| 98 void RemoveDuplicateSubresources(std::vector<URLRequestSummary>* subresources) { | 119 void RemoveDuplicateSubresources(std::vector<URLRequestSummary>* subresources) { |
| 99 std::stable_sort(subresources->begin(), subresources->end(), | 120 std::stable_sort(subresources->begin(), subresources->end(), |
| 100 [](const URLRequestSummary& x, const URLRequestSummary& y) { | 121 [](const URLRequestSummary& x, const URLRequestSummary& y) { |
| 101 return x.resource_url < y.resource_url; | 122 return x.resource_url < y.resource_url; |
| 102 }); | 123 }); |
| 103 subresources->erase( | 124 subresources->erase( |
| 104 std::unique(subresources->begin(), subresources->end(), | 125 std::unique(subresources->begin(), subresources->end(), |
| (...skipping 110 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 215 DISALLOW_COPY_AND_ASSIGN(PrefetchingObserver); | 236 DISALLOW_COPY_AND_ASSIGN(PrefetchingObserver); |
| 216 }; | 237 }; |
| 217 | 238 |
| 218 class ResourcePrefetchPredictorBrowserTest : public InProcessBrowserTest { | 239 class ResourcePrefetchPredictorBrowserTest : public InProcessBrowserTest { |
| 219 protected: | 240 protected: |
| 220 using URLRequestSummary = ResourcePrefetchPredictor::URLRequestSummary; | 241 using URLRequestSummary = ResourcePrefetchPredictor::URLRequestSummary; |
| 221 | 242 |
| 222 void SetUpCommandLine(base::CommandLine* command_line) override { | 243 void SetUpCommandLine(base::CommandLine* command_line) override { |
| 223 command_line->AppendSwitchASCII( | 244 command_line->AppendSwitchASCII( |
| 224 switches::kSpeculativeResourcePrefetching, | 245 switches::kSpeculativeResourcePrefetching, |
| 225 switches::kSpeculativeResourcePrefetchingEnabled); | 246 switches::kSpeculativeResourcePrefetchingEnabledExternal); |
| 226 } | 247 } |
| 227 | 248 |
| 228 void SetUpOnMainThread() override { | 249 void SetUpOnMainThread() override { |
| 229 embedded_test_server()->RegisterRequestHandler( | 250 embedded_test_server()->RegisterRequestHandler( |
| 230 base::Bind(&ResourcePrefetchPredictorBrowserTest::HandleRedirectRequest, | 251 base::Bind(&ResourcePrefetchPredictorBrowserTest::HandleRedirectRequest, |
| 231 base::Unretained(this))); | 252 base::Unretained(this))); |
| 232 embedded_test_server()->RegisterRequestHandler( | 253 embedded_test_server()->RegisterRequestHandler( |
| 233 base::Bind(&ResourcePrefetchPredictorBrowserTest::HandleResourceRequest, | 254 base::Bind(&ResourcePrefetchPredictorBrowserTest::HandleResourceRequest, |
| 234 base::Unretained(this))); | 255 base::Unretained(this))); |
| 235 ASSERT_TRUE(embedded_test_server()->Start()); | 256 ASSERT_TRUE(embedded_test_server()->Start()); |
| (...skipping 89 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 325 ASSERT_FALSE(redirect_chain.empty()); | 346 ASSERT_FALSE(redirect_chain.empty()); |
| 326 GURL current = initial_url; | 347 GURL current = initial_url; |
| 327 for (const auto& edge : redirect_chain) { | 348 for (const auto& edge : redirect_chain) { |
| 328 auto result = redirects_.insert(std::make_pair(current, edge)); | 349 auto result = redirects_.insert(std::make_pair(current, edge)); |
| 329 EXPECT_TRUE(result.second) << current << " already has a redirect."; | 350 EXPECT_TRUE(result.second) << current << " already has a redirect."; |
| 330 current = edge.url; | 351 current = edge.url; |
| 331 } | 352 } |
| 332 } | 353 } |
| 333 | 354 |
| 334 void ClearCache() { | 355 void ClearCache() { |
| 335 chrome::ClearCache(browser()); | 356 BrowsingDataRemover* remover = |
| 357 BrowsingDataRemoverFactory::GetForBrowserContext(browser()->profile()); |
| 358 BrowsingDataRemoverObserver observer(remover); |
| 359 remover->RemoveAndReply(BrowsingDataRemover::Unbounded(), |
| 360 BrowsingDataRemover::REMOVE_CACHE, |
| 361 BrowsingDataHelper::UNPROTECTED_WEB, &observer); |
| 362 observer.Wait(); |
| 363 |
| 336 for (auto& kv : resources_) | 364 for (auto& kv : resources_) |
| 337 kv.second.request.was_cached = false; | 365 kv.second.request.was_cached = false; |
| 338 } | 366 } |
| 339 | 367 |
| 340 // Shortcut for convenience. | 368 // Shortcut for convenience. |
| 341 GURL GetURL(const std::string& path) const { | 369 GURL GetURL(const std::string& path) const { |
| 342 return embedded_test_server()->GetURL(path); | 370 return embedded_test_server()->GetURL(path); |
| 343 } | 371 } |
| 344 | 372 |
| 345 void EnableHttpsServer() { | 373 void EnableHttpsServer() { |
| (...skipping 239 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 585 AddResource(GetURL(kStylePath2), content::RESOURCE_TYPE_STYLESHEET, | 613 AddResource(GetURL(kStylePath2), content::RESOURCE_TYPE_STYLESHEET, |
| 586 net::HIGHEST); | 614 net::HIGHEST); |
| 587 AddResource(GetURL(kScriptPath2), content::RESOURCE_TYPE_SCRIPT, net::MEDIUM); | 615 AddResource(GetURL(kScriptPath2), content::RESOURCE_TYPE_SCRIPT, net::MEDIUM); |
| 588 // Included from <iframe src="html_subresources.html"> and not recored. | 616 // Included from <iframe src="html_subresources.html"> and not recored. |
| 589 AddUnrecordedResources({GetURL(kImagePath), GetURL(kStylePath), | 617 AddUnrecordedResources({GetURL(kImagePath), GetURL(kStylePath), |
| 590 GetURL(kScriptPath), GetURL(kFontPath)}); | 618 GetURL(kScriptPath), GetURL(kFontPath)}); |
| 591 TestLearningAndPrefetching(GetURL(kHtmlIframePath)); | 619 TestLearningAndPrefetching(GetURL(kHtmlIframePath)); |
| 592 } | 620 } |
| 593 | 621 |
| 594 } // namespace predictors | 622 } // namespace predictors |
| OLD | NEW |