| 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 |