| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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 "content/browser/cache_storage/cache_storage_manager.h" | 5 #include "content/browser/cache_storage/cache_storage_manager.h" |
| 6 | 6 |
| 7 #include <stddef.h> | 7 #include <stddef.h> |
| 8 #include <stdint.h> | 8 #include <stdint.h> |
| 9 | 9 |
| 10 #include <utility> | 10 #include <utility> |
| 11 | 11 |
| 12 #include "base/files/file_path.h" | 12 #include "base/files/file_path.h" |
| 13 #include "base/files/file_util.h" | 13 #include "base/files/file_util.h" |
| 14 #include "base/files/scoped_temp_dir.h" | 14 #include "base/files/scoped_temp_dir.h" |
| 15 #include "base/guid.h" | 15 #include "base/guid.h" |
| 16 #include "base/macros.h" | 16 #include "base/macros.h" |
| 17 #include "base/memory/ptr_util.h" | 17 #include "base/memory/ptr_util.h" |
| 18 #include "base/run_loop.h" | 18 #include "base/run_loop.h" |
| 19 #include "base/sha1.h" | 19 #include "base/sha1.h" |
| 20 #include "base/stl_util.h" | 20 #include "base/stl_util.h" |
| 21 #include "base/strings/string_number_conversions.h" | 21 #include "base/strings/string_number_conversions.h" |
| 22 #include "base/thread_task_runner_handle.h" | 22 #include "base/thread_task_runner_handle.h" |
| 23 #include "content/browser/cache_storage/cache_storage.pb.h" | 23 #include "content/browser/cache_storage/cache_storage.pb.h" |
| 24 #include "content/browser/cache_storage/cache_storage_quota_client.h" | 24 #include "content/browser/cache_storage/cache_storage_quota_client.h" |
| 25 #include "content/browser/fileapi/chrome_blob_storage_context.h" | 25 #include "content/browser/fileapi/chrome_blob_storage_context.h" |
| 26 #include "content/browser/quota/mock_quota_manager_proxy.h" | 26 #include "content/browser/quota/mock_quota_manager_proxy.h" |
| 27 #include "content/public/browser/browser_thread.h" | 27 #include "content/public/browser/browser_thread.h" |
| 28 #include "content/public/browser/cache_storage_usage_info.h" | 28 #include "content/public/browser/cache_storage_usage_info.h" |
| 29 #include "content/public/browser/storage_partition.h" | |
| 30 #include "content/public/test/mock_special_storage_policy.h" | 29 #include "content/public/test/mock_special_storage_policy.h" |
| 31 #include "content/public/test/test_browser_context.h" | 30 #include "content/public/test/test_browser_context.h" |
| 32 #include "content/public/test/test_browser_thread_bundle.h" | 31 #include "content/public/test/test_browser_thread_bundle.h" |
| 33 #include "net/url_request/url_request_context.h" | 32 #include "net/url_request/url_request_context.h" |
| 34 #include "net/url_request/url_request_context_getter.h" | 33 #include "net/url_request/url_request_context_getter.h" |
| 35 #include "net/url_request/url_request_job_factory_impl.h" | 34 #include "net/url_request/url_request_job_factory_impl.h" |
| 36 #include "storage/browser/blob/blob_data_builder.h" | 35 #include "storage/browser/blob/blob_data_builder.h" |
| 37 #include "storage/browser/blob/blob_data_handle.h" | 36 #include "storage/browser/blob/blob_data_handle.h" |
| 38 #include "storage/browser/blob/blob_storage_context.h" | 37 #include "storage/browser/blob/blob_storage_context.h" |
| 39 #include "storage/browser/blob/blob_url_request_job_factory.h" | 38 #include "storage/browser/blob/blob_url_request_job_factory.h" |
| (...skipping 28 matching lines...) Expand all Loading... |
| 68 // Wait for ChromeBlobStorageContext to finish initializing. | 67 // Wait for ChromeBlobStorageContext to finish initializing. |
| 69 base::RunLoop().RunUntilIdle(); | 68 base::RunLoop().RunUntilIdle(); |
| 70 | 69 |
| 71 blob_storage_context_ = blob_storage_context->context(); | 70 blob_storage_context_ = blob_storage_context->context(); |
| 72 | 71 |
| 73 url_request_job_factory_.reset(new net::URLRequestJobFactoryImpl); | 72 url_request_job_factory_.reset(new net::URLRequestJobFactoryImpl); |
| 74 url_request_job_factory_->SetProtocolHandler( | 73 url_request_job_factory_->SetProtocolHandler( |
| 75 "blob", CreateMockBlobProtocolHandler(blob_storage_context->context())); | 74 "blob", CreateMockBlobProtocolHandler(blob_storage_context->context())); |
| 76 | 75 |
| 77 net::URLRequestContext* url_request_context = | 76 net::URLRequestContext* url_request_context = |
| 78 BrowserContext::GetDefaultStoragePartition(&browser_context_)-> | 77 browser_context_.GetRequestContext()->GetURLRequestContext(); |
| 79 GetURLRequestContext()->GetURLRequestContext(); | |
| 80 | 78 |
| 81 url_request_context->set_job_factory(url_request_job_factory_.get()); | 79 url_request_context->set_job_factory(url_request_job_factory_.get()); |
| 82 | 80 |
| 83 if (!MemoryOnly()) | 81 if (!MemoryOnly()) |
| 84 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); | 82 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); |
| 85 | 83 |
| 86 quota_policy_ = new MockSpecialStoragePolicy; | 84 quota_policy_ = new MockSpecialStoragePolicy; |
| 87 mock_quota_manager_ = new MockQuotaManager( | 85 mock_quota_manager_ = new MockQuotaManager( |
| 88 MemoryOnly() /* is incognito */, temp_dir_.path(), | 86 MemoryOnly() /* is incognito */, temp_dir_.path(), |
| 89 base::ThreadTaskRunnerHandle::Get().get(), | 87 base::ThreadTaskRunnerHandle::Get().get(), |
| 90 base::ThreadTaskRunnerHandle::Get().get(), quota_policy_.get()); | 88 base::ThreadTaskRunnerHandle::Get().get(), quota_policy_.get()); |
| 91 mock_quota_manager_->SetQuota( | 89 mock_quota_manager_->SetQuota( |
| 92 GURL(origin1_), storage::kStorageTypeTemporary, 1024 * 1024 * 100); | 90 GURL(origin1_), storage::kStorageTypeTemporary, 1024 * 1024 * 100); |
| 93 mock_quota_manager_->SetQuota( | 91 mock_quota_manager_->SetQuota( |
| 94 GURL(origin2_), storage::kStorageTypeTemporary, 1024 * 1024 * 100); | 92 GURL(origin2_), storage::kStorageTypeTemporary, 1024 * 1024 * 100); |
| 95 | 93 |
| 96 quota_manager_proxy_ = new MockQuotaManagerProxy( | 94 quota_manager_proxy_ = new MockQuotaManagerProxy( |
| 97 mock_quota_manager_.get(), base::ThreadTaskRunnerHandle::Get().get()); | 95 mock_quota_manager_.get(), base::ThreadTaskRunnerHandle::Get().get()); |
| 98 | 96 |
| 99 cache_manager_ = CacheStorageManager::Create( | 97 cache_manager_ = CacheStorageManager::Create( |
| 100 temp_dir_.path(), base::ThreadTaskRunnerHandle::Get(), | 98 temp_dir_.path(), base::ThreadTaskRunnerHandle::Get(), |
| 101 quota_manager_proxy_); | 99 quota_manager_proxy_); |
| 102 | 100 |
| 103 cache_manager_->SetBlobParametersForCache( | 101 cache_manager_->SetBlobParametersForCache( |
| 104 BrowserContext::GetDefaultStoragePartition(&browser_context_)-> | 102 browser_context_.GetRequestContext(), |
| 105 GetURLRequestContext(), | |
| 106 blob_storage_context->context()->AsWeakPtr()); | 103 blob_storage_context->context()->AsWeakPtr()); |
| 107 } | 104 } |
| 108 | 105 |
| 109 void TearDown() override { | 106 void TearDown() override { |
| 110 quota_manager_proxy_->SimulateQuotaManagerDestroyed(); | 107 quota_manager_proxy_->SimulateQuotaManagerDestroyed(); |
| 111 base::RunLoop().RunUntilIdle(); | 108 base::RunLoop().RunUntilIdle(); |
| 112 } | 109 } |
| 113 | 110 |
| 114 virtual bool MemoryOnly() { return false; } | 111 virtual bool MemoryOnly() { return false; } |
| 115 | 112 |
| (...skipping 216 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 332 cache_storage->Size(base::Bind(&CacheStorageManagerTest::UsageCallback, | 329 cache_storage->Size(base::Bind(&CacheStorageManagerTest::UsageCallback, |
| 333 base::Unretained(this), &loop)); | 330 base::Unretained(this), &loop)); |
| 334 loop.Run(); | 331 loop.Run(); |
| 335 return callback_usage_; | 332 return callback_usage_; |
| 336 } | 333 } |
| 337 | 334 |
| 338 protected: | 335 protected: |
| 339 // Temporary directory must be allocated first so as to be destroyed last. | 336 // Temporary directory must be allocated first so as to be destroyed last. |
| 340 base::ScopedTempDir temp_dir_; | 337 base::ScopedTempDir temp_dir_; |
| 341 | 338 |
| 339 TestBrowserContext browser_context_; |
| 342 TestBrowserThreadBundle browser_thread_bundle_; | 340 TestBrowserThreadBundle browser_thread_bundle_; |
| 343 TestBrowserContext browser_context_; | |
| 344 std::unique_ptr<net::URLRequestJobFactoryImpl> url_request_job_factory_; | 341 std::unique_ptr<net::URLRequestJobFactoryImpl> url_request_job_factory_; |
| 345 storage::BlobStorageContext* blob_storage_context_; | 342 storage::BlobStorageContext* blob_storage_context_; |
| 346 | 343 |
| 347 scoped_refptr<MockSpecialStoragePolicy> quota_policy_; | 344 scoped_refptr<MockSpecialStoragePolicy> quota_policy_; |
| 348 scoped_refptr<MockQuotaManager> mock_quota_manager_; | 345 scoped_refptr<MockQuotaManager> mock_quota_manager_; |
| 349 scoped_refptr<MockQuotaManagerProxy> quota_manager_proxy_; | 346 scoped_refptr<MockQuotaManagerProxy> quota_manager_proxy_; |
| 350 std::unique_ptr<CacheStorageManager> cache_manager_; | 347 std::unique_ptr<CacheStorageManager> cache_manager_; |
| 351 | 348 |
| 352 scoped_refptr<CacheStorageCache> callback_cache_; | 349 scoped_refptr<CacheStorageCache> callback_cache_; |
| 353 int callback_bool_; | 350 int callback_bool_; |
| (...skipping 792 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 1146 | 1143 |
| 1147 INSTANTIATE_TEST_CASE_P(CacheStorageManagerTests, | 1144 INSTANTIATE_TEST_CASE_P(CacheStorageManagerTests, |
| 1148 CacheStorageManagerTestP, | 1145 CacheStorageManagerTestP, |
| 1149 ::testing::Values(false, true)); | 1146 ::testing::Values(false, true)); |
| 1150 | 1147 |
| 1151 INSTANTIATE_TEST_CASE_P(CacheStorageQuotaClientTests, | 1148 INSTANTIATE_TEST_CASE_P(CacheStorageQuotaClientTests, |
| 1152 CacheStorageQuotaClientTestP, | 1149 CacheStorageQuotaClientTestP, |
| 1153 ::testing::Values(false, true)); | 1150 ::testing::Values(false, true)); |
| 1154 | 1151 |
| 1155 } // namespace content | 1152 } // namespace content |
| OLD | NEW |