diff options
Diffstat (limited to 'net/proxy/proxy_resolver_perftest.cc')
-rw-r--r-- | net/proxy/proxy_resolver_perftest.cc | 28 |
1 files changed, 13 insertions, 15 deletions
diff --git a/net/proxy/proxy_resolver_perftest.cc b/net/proxy/proxy_resolver_perftest.cc index 0c57955..dda9587 100644 --- a/net/proxy/proxy_resolver_perftest.cc +++ b/net/proxy/proxy_resolver_perftest.cc @@ -2,12 +2,17 @@ // Use of this source code is governed by a BSD-style license that can be // found in the LICENSE file. +#include "base/base_paths.h" +#include "base/file_util.h" +#include "base/path_service.h" #include "base/perftimer.h" #include "base/string_util.h" #include "net/base/mock_host_resolver.h" +#include "net/base/net_errors.h" +#include "net/proxy/proxy_info.h" #include "net/proxy/proxy_resolver_js_bindings.h" #include "net/proxy/proxy_resolver_v8.h" -#include "net/url_request/url_request_unittest.h" +#include "net/test/test_server.h" #include "testing/gtest/include/gtest/gtest.h" #if defined(OS_WIN) @@ -78,10 +83,14 @@ class PacPerfSuiteRunner { // |resolver_name| is the label used when logging the results. PacPerfSuiteRunner(net::ProxyResolver* resolver, const std::string& resolver_name) - : resolver_(resolver), resolver_name_(resolver_name) { + : resolver_(resolver), + resolver_name_(resolver_name), + test_server_(net::TestServer::TYPE_HTTP, + FilePath(FILE_PATH_LITERAL("net/data/proxy_resolver_perftest"))) { } void RunAllTests() { + ASSERT_TRUE(test_server_.Start()); for (size_t i = 0; i < arraysize(kPerfTests); ++i) { const PacPerfTest& test_data = kPerfTests[i]; RunTest(test_data.pac_name, @@ -95,9 +104,8 @@ class PacPerfSuiteRunner { const PacQuery* queries, int queries_len) { if (!resolver_->expects_pac_bytes()) { - InitHttpServer(); GURL pac_url = - server_->TestServerPage(std::string("files/") + script_name); + test_server_.GetURL(std::string("files/") + script_name); int rv = resolver_->SetPacScript( net::ProxyResolverScriptData::FromURL(pac_url), NULL); EXPECT_EQ(net::OK, rv); @@ -140,16 +148,6 @@ class PacPerfSuiteRunner { timer.Done(); } - // Lazily startup an HTTP server (to serve the PAC script). - void InitHttpServer() { - DCHECK(!resolver_->expects_pac_bytes()); - if (!server_) { - server_ = HTTPTestServer::CreateServer( - L"net/data/proxy_resolver_perftest", NULL); - } - ASSERT_TRUE(server_.get() != NULL); - } - // Read the PAC script from disk and initialize the proxy resolver with it. void LoadPacScriptIntoResolver(const std::string& script_name) { FilePath path; @@ -175,7 +173,7 @@ class PacPerfSuiteRunner { net::ProxyResolver* resolver_; std::string resolver_name_; - scoped_refptr<HTTPTestServer> server_; + net::TestServer test_server_; }; #if defined(OS_WIN) |