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 <string> | 5 #include <string> |
6 | 6 |
7 #include "base/file_util.h" | 7 #include "base/file_util.h" |
8 #include "base/message_loop/message_loop.h" | 8 #include "base/message_loop/message_loop.h" |
9 #include "base/strings/string_number_conversions.h" | 9 #include "base/strings/string_number_conversions.h" |
10 #include "base/strings/string_util.h" | 10 #include "base/strings/string_util.h" |
11 #include "base/values.h" | 11 #include "base/values.h" |
12 #include "chrome/browser/extensions/extension_info_map.h" | 12 #include "chrome/browser/extensions/extension_info_map.h" |
13 #include "chrome/browser/extensions/extension_protocols.h" | 13 #include "chrome/browser/extensions/extension_protocols.h" |
14 #include "chrome/common/chrome_paths.h" | 14 #include "chrome/common/chrome_paths.h" |
15 #include "chrome/common/extensions/extension.h" | 15 #include "chrome/common/extensions/extension.h" |
16 #include "chrome/common/url_constants.h" | 16 #include "chrome/common/url_constants.h" |
17 #include "content/public/browser/resource_request_info.h" | 17 #include "content/public/browser/resource_request_info.h" |
18 #include "content/public/test/mock_resource_context.h" | 18 #include "content/public/test/mock_resource_context.h" |
19 #include "content/public/test/test_browser_thread_bundle.h" | 19 #include "content/public/test/test_browser_thread_bundle.h" |
20 #include "extensions/common/constants.h" | 20 #include "extensions/common/constants.h" |
| 21 #include "net/base/request_priority.h" |
21 #include "net/url_request/url_request.h" | 22 #include "net/url_request/url_request.h" |
22 #include "net/url_request/url_request_job_factory_impl.h" | 23 #include "net/url_request/url_request_job_factory_impl.h" |
23 #include "net/url_request/url_request_status.h" | 24 #include "net/url_request/url_request_status.h" |
24 #include "net/url_request/url_request_test_util.h" | 25 #include "net/url_request/url_request_test_util.h" |
25 #include "testing/gtest/include/gtest/gtest.h" | 26 #include "testing/gtest/include/gtest/gtest.h" |
26 | 27 |
27 namespace extensions { | 28 namespace extensions { |
28 | 29 |
29 scoped_refptr<Extension> CreateTestExtension(const std::string& name, | 30 scoped_refptr<Extension> CreateTestExtension(const std::string& name, |
30 bool incognito_split_mode) { | 31 bool incognito_split_mode) { |
(...skipping 132 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
163 CreateTestExtension(cases[i].name, cases[i].incognito_split_mode); | 164 CreateTestExtension(cases[i].name, cases[i].incognito_split_mode); |
164 extension_info_map_->AddExtension( | 165 extension_info_map_->AddExtension( |
165 extension.get(), base::Time::Now(), cases[i].incognito_enabled); | 166 extension.get(), base::Time::Now(), cases[i].incognito_enabled); |
166 | 167 |
167 // First test a main frame request. | 168 // First test a main frame request. |
168 { | 169 { |
169 // It doesn't matter that the resource doesn't exist. If the resource | 170 // It doesn't matter that the resource doesn't exist. If the resource |
170 // is blocked, we should see ADDRESS_UNREACHABLE. Otherwise, the request | 171 // is blocked, we should see ADDRESS_UNREACHABLE. Otherwise, the request |
171 // should just fail because the file doesn't exist. | 172 // should just fail because the file doesn't exist. |
172 net::URLRequest request(extension->GetResourceURL("404.html"), | 173 net::URLRequest request(extension->GetResourceURL("404.html"), |
| 174 net::DEFAULT_PRIORITY, |
173 &test_delegate_, | 175 &test_delegate_, |
174 resource_context_.GetRequestContext()); | 176 resource_context_.GetRequestContext()); |
175 StartRequest(&request, ResourceType::MAIN_FRAME); | 177 StartRequest(&request, ResourceType::MAIN_FRAME); |
176 EXPECT_EQ(net::URLRequestStatus::FAILED, request.status().status()); | 178 EXPECT_EQ(net::URLRequestStatus::FAILED, request.status().status()); |
177 | 179 |
178 if (cases[i].should_allow_main_frame_load) { | 180 if (cases[i].should_allow_main_frame_load) { |
179 EXPECT_EQ(net::ERR_FILE_NOT_FOUND, request.status().error()) << | 181 EXPECT_EQ(net::ERR_FILE_NOT_FOUND, request.status().error()) << |
180 cases[i].name; | 182 cases[i].name; |
181 } else { | 183 } else { |
182 EXPECT_EQ(net::ERR_ADDRESS_UNREACHABLE, request.status().error()) << | 184 EXPECT_EQ(net::ERR_ADDRESS_UNREACHABLE, request.status().error()) << |
183 cases[i].name; | 185 cases[i].name; |
184 } | 186 } |
185 } | 187 } |
186 | 188 |
187 // Now do a subframe request. | 189 // Now do a subframe request. |
188 { | 190 { |
189 net::URLRequest request(extension->GetResourceURL("404.html"), | 191 net::URLRequest request(extension->GetResourceURL("404.html"), |
| 192 net::DEFAULT_PRIORITY, |
190 &test_delegate_, | 193 &test_delegate_, |
191 resource_context_.GetRequestContext()); | 194 resource_context_.GetRequestContext()); |
192 StartRequest(&request, ResourceType::SUB_FRAME); | 195 StartRequest(&request, ResourceType::SUB_FRAME); |
193 EXPECT_EQ(net::URLRequestStatus::FAILED, request.status().status()); | 196 EXPECT_EQ(net::URLRequestStatus::FAILED, request.status().status()); |
194 | 197 |
195 if (cases[i].should_allow_sub_frame_load) { | 198 if (cases[i].should_allow_sub_frame_load) { |
196 EXPECT_EQ(net::ERR_FILE_NOT_FOUND, request.status().error()) << | 199 EXPECT_EQ(net::ERR_FILE_NOT_FOUND, request.status().error()) << |
197 cases[i].name; | 200 cases[i].name; |
198 } else { | 201 } else { |
199 EXPECT_EQ(net::ERR_ADDRESS_UNREACHABLE, request.status().error()) << | 202 EXPECT_EQ(net::ERR_ADDRESS_UNREACHABLE, request.status().error()) << |
(...skipping 18 matching lines...) Expand all Loading... |
218 TEST_F(ExtensionProtocolTest, ComponentResourceRequest) { | 221 TEST_F(ExtensionProtocolTest, ComponentResourceRequest) { |
219 // Register a non-incognito extension protocol handler. | 222 // Register a non-incognito extension protocol handler. |
220 SetProtocolHandler(false); | 223 SetProtocolHandler(false); |
221 | 224 |
222 scoped_refptr<Extension> extension = CreateWebStoreExtension(); | 225 scoped_refptr<Extension> extension = CreateWebStoreExtension(); |
223 extension_info_map_->AddExtension(extension.get(), base::Time::Now(), false); | 226 extension_info_map_->AddExtension(extension.get(), base::Time::Now(), false); |
224 | 227 |
225 // First test it with the extension enabled. | 228 // First test it with the extension enabled. |
226 { | 229 { |
227 net::URLRequest request(extension->GetResourceURL("webstore_icon_16.png"), | 230 net::URLRequest request(extension->GetResourceURL("webstore_icon_16.png"), |
| 231 net::DEFAULT_PRIORITY, |
228 &test_delegate_, | 232 &test_delegate_, |
229 resource_context_.GetRequestContext()); | 233 resource_context_.GetRequestContext()); |
230 StartRequest(&request, ResourceType::MEDIA); | 234 StartRequest(&request, ResourceType::MEDIA); |
231 EXPECT_EQ(net::URLRequestStatus::SUCCESS, request.status().status()); | 235 EXPECT_EQ(net::URLRequestStatus::SUCCESS, request.status().status()); |
232 CheckForContentLengthHeader(&request); | 236 CheckForContentLengthHeader(&request); |
233 } | 237 } |
234 | 238 |
235 // And then test it with the extension disabled. | 239 // And then test it with the extension disabled. |
236 extension_info_map_->RemoveExtension(extension->id(), | 240 extension_info_map_->RemoveExtension(extension->id(), |
237 UnloadedExtensionInfo::REASON_DISABLE); | 241 UnloadedExtensionInfo::REASON_DISABLE); |
238 { | 242 { |
239 net::URLRequest request(extension->GetResourceURL("webstore_icon_16.png"), | 243 net::URLRequest request(extension->GetResourceURL("webstore_icon_16.png"), |
| 244 net::DEFAULT_PRIORITY, |
240 &test_delegate_, | 245 &test_delegate_, |
241 resource_context_.GetRequestContext()); | 246 resource_context_.GetRequestContext()); |
242 StartRequest(&request, ResourceType::MEDIA); | 247 StartRequest(&request, ResourceType::MEDIA); |
243 EXPECT_EQ(net::URLRequestStatus::SUCCESS, request.status().status()); | 248 EXPECT_EQ(net::URLRequestStatus::SUCCESS, request.status().status()); |
244 CheckForContentLengthHeader(&request); | 249 CheckForContentLengthHeader(&request); |
245 } | 250 } |
246 } | 251 } |
247 | 252 |
248 // Tests that a URL request for resource from an extension returns a few | 253 // Tests that a URL request for resource from an extension returns a few |
249 // expected response headers. | 254 // expected response headers. |
250 TEST_F(ExtensionProtocolTest, ResourceRequestResponseHeaders) { | 255 TEST_F(ExtensionProtocolTest, ResourceRequestResponseHeaders) { |
251 // Register a non-incognito extension protocol handler. | 256 // Register a non-incognito extension protocol handler. |
252 SetProtocolHandler(false); | 257 SetProtocolHandler(false); |
253 | 258 |
254 scoped_refptr<Extension> extension = CreateTestResponseHeaderExtension(); | 259 scoped_refptr<Extension> extension = CreateTestResponseHeaderExtension(); |
255 extension_info_map_->AddExtension(extension.get(), base::Time::Now(), false); | 260 extension_info_map_->AddExtension(extension.get(), base::Time::Now(), false); |
256 | 261 |
257 { | 262 { |
258 net::URLRequest request(extension->GetResourceURL("test.dat"), | 263 net::URLRequest request(extension->GetResourceURL("test.dat"), |
| 264 net::DEFAULT_PRIORITY, |
259 &test_delegate_, | 265 &test_delegate_, |
260 resource_context_.GetRequestContext()); | 266 resource_context_.GetRequestContext()); |
261 StartRequest(&request, ResourceType::MEDIA); | 267 StartRequest(&request, ResourceType::MEDIA); |
262 EXPECT_EQ(net::URLRequestStatus::SUCCESS, request.status().status()); | 268 EXPECT_EQ(net::URLRequestStatus::SUCCESS, request.status().status()); |
263 | 269 |
264 // Check that cache-related headers are set. | 270 // Check that cache-related headers are set. |
265 std::string etag; | 271 std::string etag; |
266 request.GetResponseHeaderByName("ETag", &etag); | 272 request.GetResponseHeaderByName("ETag", &etag); |
267 EXPECT_TRUE(StartsWithASCII(etag, "\"", false)); | 273 EXPECT_TRUE(StartsWithASCII(etag, "\"", false)); |
268 EXPECT_TRUE(EndsWith(etag, "\"", false)); | 274 EXPECT_TRUE(EndsWith(etag, "\"", false)); |
269 | 275 |
270 std::string revalidation_header; | 276 std::string revalidation_header; |
271 request.GetResponseHeaderByName("cache-control", &revalidation_header); | 277 request.GetResponseHeaderByName("cache-control", &revalidation_header); |
272 EXPECT_EQ("no-cache", revalidation_header); | 278 EXPECT_EQ("no-cache", revalidation_header); |
273 | 279 |
274 // We set test.dat as web-accessible, so it should have a CORS header. | 280 // We set test.dat as web-accessible, so it should have a CORS header. |
275 std::string access_control; | 281 std::string access_control; |
276 request.GetResponseHeaderByName("Access-Control-Allow-Origin", | 282 request.GetResponseHeaderByName("Access-Control-Allow-Origin", |
277 &access_control); | 283 &access_control); |
278 EXPECT_EQ("*", access_control); | 284 EXPECT_EQ("*", access_control); |
279 } | 285 } |
280 } | 286 } |
281 | 287 |
282 } // namespace extensions | 288 } // namespace extensions |
OLD | NEW |