| Index: chrome/browser/prerender/prerender_test_utils.cc
|
| diff --git a/chrome/browser/prerender/prerender_test_utils.cc b/chrome/browser/prerender/prerender_test_utils.cc
|
| index 364bfe579f57baa7b1c17ef274cb65c77f01f73d..b95f60cdaa60296e2828fad158e14e3ec04fd010 100644
|
| --- a/chrome/browser/prerender/prerender_test_utils.cc
|
| +++ b/chrome/browser/prerender/prerender_test_utils.cc
|
| @@ -7,11 +7,9 @@
|
| #include <memory>
|
| #include <set>
|
| #include <string>
|
| -#include <utility>
|
| #include <vector>
|
|
|
| #include "base/command_line.h"
|
| -#include "base/strings/string_split.h"
|
| #include "base/strings/utf_string_conversions.h"
|
| #include "chrome/browser/loader/chrome_resource_dispatcher_host_delegate.h"
|
| #include "chrome/browser/prerender/prerender_manager.h"
|
| @@ -31,7 +29,6 @@
|
| #include "content/public/common/content_switches.h"
|
| #include "content/public/common/url_constants.h"
|
| #include "content/public/test/ppapi_test_utils.h"
|
| -#include "net/test/embedded_test_server/request_handler_util.h"
|
| #include "net/url_request/url_request_filter.h"
|
| #include "ppapi/shared_impl/ppapi_switches.h"
|
| #include "testing/gtest/include/gtest/gtest.h"
|
| @@ -349,7 +346,6 @@
|
|
|
| TestPrerender::TestPrerender()
|
| : contents_(nullptr),
|
| - final_status_(FINAL_STATUS_MAX),
|
| number_of_loads_(0),
|
| expected_number_of_loads_(0),
|
| started_(false),
|
| @@ -358,12 +354,6 @@
|
| TestPrerender::~TestPrerender() {
|
| if (contents_)
|
| contents_->RemoveObserver(this);
|
| -}
|
| -
|
| -FinalStatus TestPrerender::GetFinalStatus() const {
|
| - if (contents_)
|
| - return contents_->final_status();
|
| - return final_status_;
|
| }
|
|
|
| void TestPrerender::WaitForCreate() {
|
| @@ -418,7 +408,6 @@
|
| void TestPrerender::OnPrerenderStop(PrerenderContents* contents) {
|
| DCHECK(contents_);
|
| contents_ = nullptr;
|
| - final_status_ = contents->final_status();
|
| stopped_ = true;
|
| stop_loop_.Quit();
|
| // If there is a WaitForLoads call and it has yet to see the expected number
|
| @@ -539,6 +528,36 @@
|
| prerender_data ? prerender_data->contents() : nullptr);
|
| }
|
|
|
| +std::unique_ptr<TestPrerender> PrerenderInProcessBrowserTest::PrerenderTestURL(
|
| + const std::string& html_file,
|
| + FinalStatus expected_final_status,
|
| + int expected_number_of_loads) {
|
| + GURL url = src_server()->GetURL(MakeAbsolute(html_file));
|
| + return PrerenderTestURL(url, expected_final_status, expected_number_of_loads);
|
| +}
|
| +
|
| +std::unique_ptr<TestPrerender> PrerenderInProcessBrowserTest::PrerenderTestURL(
|
| + const GURL& url,
|
| + FinalStatus expected_final_status,
|
| + int expected_number_of_loads) {
|
| + std::vector<FinalStatus> expected_final_status_queue(1,
|
| + expected_final_status);
|
| + auto prerenders = PrerenderTestURLImpl(url, expected_final_status_queue,
|
| + expected_number_of_loads);
|
| + CHECK_EQ(1u, prerenders.size());
|
| + return std::move(prerenders[0]);
|
| +}
|
| +
|
| +std::vector<std::unique_ptr<TestPrerender>>
|
| +PrerenderInProcessBrowserTest::PrerenderTestURL(
|
| + const std::string& html_file,
|
| + const std::vector<FinalStatus>& expected_final_status_queue,
|
| + int expected_number_of_loads) {
|
| + GURL url = src_server()->GetURL(MakeAbsolute(html_file));
|
| + return PrerenderTestURLImpl(url, expected_final_status_queue,
|
| + expected_number_of_loads);
|
| +}
|
| +
|
| net::EmbeddedTestServer* PrerenderInProcessBrowserTest::src_server() {
|
| if (https_src_server_)
|
| return https_src_server_.get();
|
| @@ -610,7 +629,8 @@
|
| std::vector<std::unique_ptr<TestPrerender>>
|
| PrerenderInProcessBrowserTest::NavigateWithPrerenders(
|
| const GURL& loader_url,
|
| - const std::vector<FinalStatus>& expected_final_status_queue) {
|
| + const std::vector<FinalStatus>& expected_final_status_queue,
|
| + int expected_number_of_loads) {
|
| CHECK(!expected_final_status_queue.empty());
|
| std::vector<std::unique_ptr<TestPrerender>> prerenders;
|
| for (size_t i = 0; i < expected_final_status_queue.size(); i++) {
|
| @@ -621,23 +641,10 @@
|
| // Navigate to the loader URL and then wait for the first prerender to be
|
| // created.
|
| ui_test_utils::NavigateToURL(current_browser(), loader_url);
|
| - prerenders[0]->WaitForCreate();
|
| + prerenders.at(0)->WaitForCreate();
|
| + prerenders.at(0)->WaitForLoads(expected_number_of_loads);
|
|
|
| return prerenders;
|
| -}
|
| -
|
| -GURL PrerenderInProcessBrowserTest::ServeLoaderURL(
|
| - const std::string& loader_path,
|
| - const std::string& replacement_variable,
|
| - const GURL& url_to_prerender,
|
| - const std::string& loader_query) {
|
| - base::StringPairs replacement_text;
|
| - replacement_text.push_back(
|
| - make_pair(replacement_variable, url_to_prerender.spec()));
|
| - std::string replacement_path;
|
| - net::test_server::GetFilePathWithReplacements(loader_path, replacement_text,
|
| - &replacement_path);
|
| - return src_server()->GetURL(replacement_path + loader_query);
|
| }
|
|
|
| void CreateCountingInterceptorOnIO(
|
|
|