| 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 <stdint.h> |
| 6 |
| 5 #include <map> | 7 #include <map> |
| 6 | 8 |
| 7 #include "base/bind.h" | 9 #include "base/bind.h" |
| 8 #include "base/files/file_path.h" | 10 #include "base/files/file_path.h" |
| 9 #include "base/location.h" | 11 #include "base/location.h" |
| 10 #include "base/run_loop.h" | 12 #include "base/run_loop.h" |
| 11 #include "base/single_thread_task_runner.h" | 13 #include "base/single_thread_task_runner.h" |
| 12 #include "base/strings/utf_string_conversions.h" | 14 #include "base/strings/utf_string_conversions.h" |
| 13 #include "base/thread_task_runner_handle.h" | 15 #include "base/thread_task_runner_handle.h" |
| 14 #include "net/base/completion_callback.h" | 16 #include "net/base/completion_callback.h" |
| (...skipping 113 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 128 | 130 |
| 129 DatabaseQuotaClientTest() | 131 DatabaseQuotaClientTest() |
| 130 : kOriginA("http://host"), | 132 : kOriginA("http://host"), |
| 131 kOriginB("http://host:8000"), | 133 kOriginB("http://host:8000"), |
| 132 kOriginOther("http://other"), | 134 kOriginOther("http://other"), |
| 133 usage_(0), | 135 usage_(0), |
| 134 mock_tracker_(new MockDatabaseTracker), | 136 mock_tracker_(new MockDatabaseTracker), |
| 135 weak_factory_(this) { | 137 weak_factory_(this) { |
| 136 } | 138 } |
| 137 | 139 |
| 138 int64 GetOriginUsage(storage::QuotaClient* client, | 140 int64_t GetOriginUsage(storage::QuotaClient* client, |
| 139 const GURL& origin, | 141 const GURL& origin, |
| 140 storage::StorageType type) { | 142 storage::StorageType type) { |
| 141 usage_ = 0; | 143 usage_ = 0; |
| 142 client->GetOriginUsage( | 144 client->GetOriginUsage( |
| 143 origin, type, | 145 origin, type, |
| 144 base::Bind(&DatabaseQuotaClientTest::OnGetOriginUsageComplete, | 146 base::Bind(&DatabaseQuotaClientTest::OnGetOriginUsageComplete, |
| 145 weak_factory_.GetWeakPtr())); | 147 weak_factory_.GetWeakPtr())); |
| 146 base::RunLoop().RunUntilIdle(); | 148 base::RunLoop().RunUntilIdle(); |
| 147 return usage_; | 149 return usage_; |
| 148 } | 150 } |
| 149 | 151 |
| 150 const std::set<GURL>& GetOriginsForType(storage::QuotaClient* client, | 152 const std::set<GURL>& GetOriginsForType(storage::QuotaClient* client, |
| (...skipping 28 matching lines...) Expand all Loading... |
| 179 base::Bind(&DatabaseQuotaClientTest::OnDeleteOriginDataComplete, | 181 base::Bind(&DatabaseQuotaClientTest::OnDeleteOriginDataComplete, |
| 180 weak_factory_.GetWeakPtr())); | 182 weak_factory_.GetWeakPtr())); |
| 181 base::RunLoop().RunUntilIdle(); | 183 base::RunLoop().RunUntilIdle(); |
| 182 return delete_status_ == storage::kQuotaStatusOk; | 184 return delete_status_ == storage::kQuotaStatusOk; |
| 183 } | 185 } |
| 184 | 186 |
| 185 MockDatabaseTracker* mock_tracker() { return mock_tracker_.get(); } | 187 MockDatabaseTracker* mock_tracker() { return mock_tracker_.get(); } |
| 186 | 188 |
| 187 | 189 |
| 188 private: | 190 private: |
| 189 void OnGetOriginUsageComplete(int64 usage) { | 191 void OnGetOriginUsageComplete(int64_t usage) { usage_ = usage; } |
| 190 usage_ = usage; | |
| 191 } | |
| 192 | 192 |
| 193 void OnGetOriginsComplete(const std::set<GURL>& origins) { | 193 void OnGetOriginsComplete(const std::set<GURL>& origins) { |
| 194 origins_ = origins; | 194 origins_ = origins; |
| 195 } | 195 } |
| 196 | 196 |
| 197 void OnDeleteOriginDataComplete(storage::QuotaStatusCode status) { | 197 void OnDeleteOriginDataComplete(storage::QuotaStatusCode status) { |
| 198 delete_status_ = status; | 198 delete_status_ = status; |
| 199 } | 199 } |
| 200 | 200 |
| 201 base::MessageLoop message_loop_; | 201 base::MessageLoop message_loop_; |
| 202 int64 usage_; | 202 int64_t usage_; |
| 203 std::set<GURL> origins_; | 203 std::set<GURL> origins_; |
| 204 storage::QuotaStatusCode delete_status_; | 204 storage::QuotaStatusCode delete_status_; |
| 205 scoped_refptr<MockDatabaseTracker> mock_tracker_; | 205 scoped_refptr<MockDatabaseTracker> mock_tracker_; |
| 206 base::WeakPtrFactory<DatabaseQuotaClientTest> weak_factory_; | 206 base::WeakPtrFactory<DatabaseQuotaClientTest> weak_factory_; |
| 207 }; | 207 }; |
| 208 | 208 |
| 209 | 209 |
| 210 TEST_F(DatabaseQuotaClientTest, GetOriginUsage) { | 210 TEST_F(DatabaseQuotaClientTest, GetOriginUsage) { |
| 211 DatabaseQuotaClient client(base::ThreadTaskRunnerHandle::Get().get(), | 211 DatabaseQuotaClient client(base::ThreadTaskRunnerHandle::Get().get(), |
| 212 mock_tracker()); | 212 mock_tracker()); |
| (...skipping 61 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 274 mock_tracker()->set_async_delete(false); | 274 mock_tracker()->set_async_delete(false); |
| 275 EXPECT_TRUE(DeleteOriginData(&client, kTemp, kOriginA)); | 275 EXPECT_TRUE(DeleteOriginData(&client, kTemp, kOriginA)); |
| 276 EXPECT_EQ(1, mock_tracker()->delete_called_count()); | 276 EXPECT_EQ(1, mock_tracker()->delete_called_count()); |
| 277 | 277 |
| 278 mock_tracker()->set_async_delete(true); | 278 mock_tracker()->set_async_delete(true); |
| 279 EXPECT_TRUE(DeleteOriginData(&client, kTemp, kOriginA)); | 279 EXPECT_TRUE(DeleteOriginData(&client, kTemp, kOriginA)); |
| 280 EXPECT_EQ(2, mock_tracker()->delete_called_count()); | 280 EXPECT_EQ(2, mock_tracker()->delete_called_count()); |
| 281 } | 281 } |
| 282 | 282 |
| 283 } // namespace content | 283 } // namespace content |
| OLD | NEW |