OLD | NEW |
1 // Copyright (c) 2009 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2009 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/perftimer.h" | 5 #include "base/perftimer.h" |
6 #include "net/base/mock_host_resolver.h" | 6 #include "net/base/mock_host_resolver.h" |
7 #include "net/proxy/proxy_resolver_v8.h" | 7 #include "net/proxy/proxy_resolver_v8.h" |
8 #include "net/url_request/url_request_unittest.h" | 8 #include "net/url_request/url_request_unittest.h" |
9 #include "testing/gtest/include/gtest/gtest.h" | 9 #include "testing/gtest/include/gtest/gtest.h" |
10 | 10 |
(...skipping 74 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
85 RunTest(test_data.pac_name, | 85 RunTest(test_data.pac_name, |
86 test_data.queries, | 86 test_data.queries, |
87 test_data.NumQueries()); | 87 test_data.NumQueries()); |
88 } | 88 } |
89 } | 89 } |
90 | 90 |
91 private: | 91 private: |
92 void RunTest(const std::string& script_name, | 92 void RunTest(const std::string& script_name, |
93 const PacQuery* queries, | 93 const PacQuery* queries, |
94 int queries_len) { | 94 int queries_len) { |
95 GURL pac_url; | 95 if (!resolver_->expects_pac_bytes()) { |
96 | |
97 if (resolver_->does_fetch()) { | |
98 InitHttpServer(); | 96 InitHttpServer(); |
99 pac_url = server_->TestServerPage(std::string("files/") + script_name); | 97 GURL pac_url = |
| 98 server_->TestServerPage(std::string("files/") + script_name); |
| 99 resolver_->SetPacScriptByUrl(pac_url); |
100 } else { | 100 } else { |
101 LoadPacScriptIntoResolver(script_name); | 101 LoadPacScriptIntoResolver(script_name); |
102 } | 102 } |
103 | 103 |
104 // Do a query to warm things up. In the case of internal-fetch proxy | 104 // Do a query to warm things up. In the case of internal-fetch proxy |
105 // resolvers, the first resolve will be slow since it has to download | 105 // resolvers, the first resolve will be slow since it has to download |
106 // the PAC script. | 106 // the PAC script. |
107 { | 107 { |
108 net::ProxyInfo proxy_info; | 108 net::ProxyInfo proxy_info; |
109 int result = resolver_->GetProxyForURL( | 109 int result = resolver_->GetProxyForURL( |
110 GURL("http://www.warmup.com"), pac_url, &proxy_info); | 110 GURL("http://www.warmup.com"), &proxy_info, NULL, NULL); |
111 ASSERT_EQ(net::OK, result); | 111 ASSERT_EQ(net::OK, result); |
112 } | 112 } |
113 | 113 |
114 // Start the perf timer. | 114 // Start the perf timer. |
115 std::string perf_test_name = resolver_name_ + "_" + script_name; | 115 std::string perf_test_name = resolver_name_ + "_" + script_name; |
116 PerfTimeLogger timer(perf_test_name.c_str()); | 116 PerfTimeLogger timer(perf_test_name.c_str()); |
117 | 117 |
118 for (int i = 0; i < kNumIterations; ++i) { | 118 for (int i = 0; i < kNumIterations; ++i) { |
119 // Round-robin between URLs to resolve. | 119 // Round-robin between URLs to resolve. |
120 const PacQuery& query = queries[i % queries_len]; | 120 const PacQuery& query = queries[i % queries_len]; |
121 | 121 |
122 // Resolve. | 122 // Resolve. |
123 net::ProxyInfo proxy_info; | 123 net::ProxyInfo proxy_info; |
124 int result = resolver_->GetProxyForURL(GURL(query.query_url), | 124 int result = resolver_->GetProxyForURL(GURL(query.query_url), |
125 pac_url, | 125 &proxy_info, NULL, NULL); |
126 &proxy_info); | |
127 | 126 |
128 // Check that the result was correct. Note that ToPacString() and | 127 // Check that the result was correct. Note that ToPacString() and |
129 // ASSERT_EQ() are fast, so they won't skew the results. | 128 // ASSERT_EQ() are fast, so they won't skew the results. |
130 ASSERT_EQ(net::OK, result); | 129 ASSERT_EQ(net::OK, result); |
131 ASSERT_EQ(query.expected_result, proxy_info.ToPacString()); | 130 ASSERT_EQ(query.expected_result, proxy_info.ToPacString()); |
132 } | 131 } |
133 | 132 |
134 // Print how long the test ran for. | 133 // Print how long the test ran for. |
135 timer.Done(); | 134 timer.Done(); |
136 } | 135 } |
137 | 136 |
138 // Lazily startup an HTTP server (to serve the PAC script). | 137 // Lazily startup an HTTP server (to serve the PAC script). |
139 void InitHttpServer() { | 138 void InitHttpServer() { |
140 DCHECK(resolver_->does_fetch()); | 139 DCHECK(!resolver_->expects_pac_bytes()); |
141 if (!server_) { | 140 if (!server_) { |
142 server_ = HTTPTestServer::CreateServer( | 141 server_ = HTTPTestServer::CreateServer( |
143 L"net/data/proxy_resolver_perftest", NULL); | 142 L"net/data/proxy_resolver_perftest", NULL); |
144 } | 143 } |
145 ASSERT_TRUE(server_.get() != NULL); | 144 ASSERT_TRUE(server_.get() != NULL); |
146 } | 145 } |
147 | 146 |
148 // Read the PAC script from disk and initialize the proxy resolver with it. | 147 // Read the PAC script from disk and initialize the proxy resolver with it. |
149 void LoadPacScriptIntoResolver(const std::string& script_name) { | 148 void LoadPacScriptIntoResolver(const std::string& script_name) { |
150 FilePath path; | 149 FilePath path; |
151 PathService::Get(base::DIR_SOURCE_ROOT, &path); | 150 PathService::Get(base::DIR_SOURCE_ROOT, &path); |
152 path = path.AppendASCII("net"); | 151 path = path.AppendASCII("net"); |
153 path = path.AppendASCII("data"); | 152 path = path.AppendASCII("data"); |
154 path = path.AppendASCII("proxy_resolver_perftest"); | 153 path = path.AppendASCII("proxy_resolver_perftest"); |
155 path = path.AppendASCII(script_name); | 154 path = path.AppendASCII(script_name); |
156 | 155 |
157 // Try to read the file from disk. | 156 // Try to read the file from disk. |
158 std::string file_contents; | 157 std::string file_contents; |
159 bool ok = file_util::ReadFileToString(path, &file_contents); | 158 bool ok = file_util::ReadFileToString(path, &file_contents); |
160 | 159 |
161 // If we can't load the file from disk, something is misconfigured. | 160 // If we can't load the file from disk, something is misconfigured. |
162 LOG_IF(ERROR, !ok) << "Failed to read file: " << path.value(); | 161 LOG_IF(ERROR, !ok) << "Failed to read file: " << path.value(); |
163 ASSERT_TRUE(ok); | 162 ASSERT_TRUE(ok); |
164 | 163 |
165 // Load the PAC script into the ProxyResolver. | 164 // Load the PAC script into the ProxyResolver. |
166 resolver_->SetPacScript(file_contents); | 165 resolver_->SetPacScriptByData(file_contents); |
167 } | 166 } |
168 | 167 |
169 net::ProxyResolver* resolver_; | 168 net::ProxyResolver* resolver_; |
170 std::string resolver_name_; | 169 std::string resolver_name_; |
171 scoped_refptr<HTTPTestServer> server_; | 170 scoped_refptr<HTTPTestServer> server_; |
172 }; | 171 }; |
173 | 172 |
174 #if defined(OS_WIN) | 173 #if defined(OS_WIN) |
175 TEST(ProxyResolverPerfTest, ProxyResolverWinHttp) { | 174 TEST(ProxyResolverPerfTest, ProxyResolverWinHttp) { |
176 net::ProxyResolverWinHttp resolver; | 175 net::ProxyResolverWinHttp resolver; |
(...skipping 10 matching lines...) Expand all Loading... |
187 | 186 |
188 TEST(ProxyResolverPerfTest, ProxyResolverV8) { | 187 TEST(ProxyResolverPerfTest, ProxyResolverV8) { |
189 net::ProxyResolverV8::JSBindings* js_bindings = | 188 net::ProxyResolverV8::JSBindings* js_bindings = |
190 net::ProxyResolverV8::CreateDefaultBindings( | 189 net::ProxyResolverV8::CreateDefaultBindings( |
191 new net::MockHostResolver, NULL); | 190 new net::MockHostResolver, NULL); |
192 | 191 |
193 net::ProxyResolverV8 resolver(js_bindings); | 192 net::ProxyResolverV8 resolver(js_bindings); |
194 PacPerfSuiteRunner runner(&resolver, "ProxyResolverV8"); | 193 PacPerfSuiteRunner runner(&resolver, "ProxyResolverV8"); |
195 runner.RunAllTests(); | 194 runner.RunAllTests(); |
196 } | 195 } |
OLD | NEW |