| 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 <map> | 5 #include <map> |
| 6 | 6 |
| 7 #include "base/bind.h" | 7 #include "base/bind.h" |
| 8 #include "base/files/file_path.h" | 8 #include "base/files/file_path.h" |
| 9 #include "base/location.h" | 9 #include "base/message_loop/message_loop_proxy.h" |
| 10 #include "base/run_loop.h" | 10 #include "base/run_loop.h" |
| 11 #include "base/single_thread_task_runner.h" | |
| 12 #include "base/strings/utf_string_conversions.h" | 11 #include "base/strings/utf_string_conversions.h" |
| 13 #include "base/thread_task_runner_handle.h" | |
| 14 #include "net/base/completion_callback.h" | 12 #include "net/base/completion_callback.h" |
| 15 #include "net/base/net_errors.h" | 13 #include "net/base/net_errors.h" |
| 16 #include "storage/browser/database/database_quota_client.h" | 14 #include "storage/browser/database/database_quota_client.h" |
| 17 #include "storage/browser/database/database_tracker.h" | 15 #include "storage/browser/database/database_tracker.h" |
| 18 #include "storage/browser/database/database_util.h" | 16 #include "storage/browser/database/database_util.h" |
| 19 #include "storage/common/database/database_identifier.h" | 17 #include "storage/common/database/database_identifier.h" |
| 20 #include "testing/gtest/include/gtest/gtest.h" | 18 #include "testing/gtest/include/gtest/gtest.h" |
| 21 | 19 |
| 22 using storage::DatabaseQuotaClient; | 20 using storage::DatabaseQuotaClient; |
| 23 using storage::DatabaseTracker; | 21 using storage::DatabaseTracker; |
| (...skipping 43 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 67 ++iter) { | 65 ++iter) { |
| 68 origins_info->push_back(OriginInfo(iter->second)); | 66 origins_info->push_back(OriginInfo(iter->second)); |
| 69 } | 67 } |
| 70 return true; | 68 return true; |
| 71 } | 69 } |
| 72 | 70 |
| 73 int DeleteDataForOrigin(const std::string& origin_identifier, | 71 int DeleteDataForOrigin(const std::string& origin_identifier, |
| 74 const net::CompletionCallback& callback) override { | 72 const net::CompletionCallback& callback) override { |
| 75 ++delete_called_count_; | 73 ++delete_called_count_; |
| 76 if (async_delete()) { | 74 if (async_delete()) { |
| 77 base::ThreadTaskRunnerHandle::Get()->PostTask( | 75 base::MessageLoopProxy::current()->PostTask( |
| 78 FROM_HERE, base::Bind(&MockDatabaseTracker::AsyncDeleteDataForOrigin, | 76 FROM_HERE, |
| 79 this, callback)); | 77 base::Bind(&MockDatabaseTracker::AsyncDeleteDataForOrigin, this, |
| 78 callback)); |
| 80 return net::ERR_IO_PENDING; | 79 return net::ERR_IO_PENDING; |
| 81 } | 80 } |
| 82 return net::OK; | 81 return net::OK; |
| 83 } | 82 } |
| 84 | 83 |
| 85 void AsyncDeleteDataForOrigin(const net::CompletionCallback& callback) { | 84 void AsyncDeleteDataForOrigin(const net::CompletionCallback& callback) { |
| 86 callback.Run(net::OK); | 85 callback.Run(net::OK); |
| 87 } | 86 } |
| 88 | 87 |
| 89 void AddMockDatabase(const GURL& origin, const char* name, int size) { | 88 void AddMockDatabase(const GURL& origin, const char* name, int size) { |
| (...skipping 111 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 201 base::MessageLoop message_loop_; | 200 base::MessageLoop message_loop_; |
| 202 int64 usage_; | 201 int64 usage_; |
| 203 std::set<GURL> origins_; | 202 std::set<GURL> origins_; |
| 204 storage::QuotaStatusCode delete_status_; | 203 storage::QuotaStatusCode delete_status_; |
| 205 scoped_refptr<MockDatabaseTracker> mock_tracker_; | 204 scoped_refptr<MockDatabaseTracker> mock_tracker_; |
| 206 base::WeakPtrFactory<DatabaseQuotaClientTest> weak_factory_; | 205 base::WeakPtrFactory<DatabaseQuotaClientTest> weak_factory_; |
| 207 }; | 206 }; |
| 208 | 207 |
| 209 | 208 |
| 210 TEST_F(DatabaseQuotaClientTest, GetOriginUsage) { | 209 TEST_F(DatabaseQuotaClientTest, GetOriginUsage) { |
| 211 DatabaseQuotaClient client(base::ThreadTaskRunnerHandle::Get().get(), | 210 DatabaseQuotaClient client(base::MessageLoopProxy::current().get(), |
| 212 mock_tracker()); | 211 mock_tracker()); |
| 213 | 212 |
| 214 EXPECT_EQ(0, GetOriginUsage(&client, kOriginA, kTemp)); | 213 EXPECT_EQ(0, GetOriginUsage(&client, kOriginA, kTemp)); |
| 215 EXPECT_EQ(0, GetOriginUsage(&client, kOriginA, kPerm)); | 214 EXPECT_EQ(0, GetOriginUsage(&client, kOriginA, kPerm)); |
| 216 | 215 |
| 217 mock_tracker()->AddMockDatabase(kOriginA, "fooDB", 1000); | 216 mock_tracker()->AddMockDatabase(kOriginA, "fooDB", 1000); |
| 218 EXPECT_EQ(1000, GetOriginUsage(&client, kOriginA, kTemp)); | 217 EXPECT_EQ(1000, GetOriginUsage(&client, kOriginA, kTemp)); |
| 219 EXPECT_EQ(0, GetOriginUsage(&client, kOriginA, kPerm)); | 218 EXPECT_EQ(0, GetOriginUsage(&client, kOriginA, kPerm)); |
| 220 | 219 |
| 221 EXPECT_EQ(0, GetOriginUsage(&client, kOriginB, kPerm)); | 220 EXPECT_EQ(0, GetOriginUsage(&client, kOriginB, kPerm)); |
| 222 EXPECT_EQ(0, GetOriginUsage(&client, kOriginB, kTemp)); | 221 EXPECT_EQ(0, GetOriginUsage(&client, kOriginB, kTemp)); |
| 223 } | 222 } |
| 224 | 223 |
| 225 TEST_F(DatabaseQuotaClientTest, GetOriginsForHost) { | 224 TEST_F(DatabaseQuotaClientTest, GetOriginsForHost) { |
| 226 DatabaseQuotaClient client(base::ThreadTaskRunnerHandle::Get().get(), | 225 DatabaseQuotaClient client(base::MessageLoopProxy::current().get(), |
| 227 mock_tracker()); | 226 mock_tracker()); |
| 228 | 227 |
| 229 EXPECT_EQ(kOriginA.host(), kOriginB.host()); | 228 EXPECT_EQ(kOriginA.host(), kOriginB.host()); |
| 230 EXPECT_NE(kOriginA.host(), kOriginOther.host()); | 229 EXPECT_NE(kOriginA.host(), kOriginOther.host()); |
| 231 | 230 |
| 232 std::set<GURL> origins = GetOriginsForHost(&client, kTemp, kOriginA.host()); | 231 std::set<GURL> origins = GetOriginsForHost(&client, kTemp, kOriginA.host()); |
| 233 EXPECT_TRUE(origins.empty()); | 232 EXPECT_TRUE(origins.empty()); |
| 234 | 233 |
| 235 mock_tracker()->AddMockDatabase(kOriginA, "fooDB", 1000); | 234 mock_tracker()->AddMockDatabase(kOriginA, "fooDB", 1000); |
| 236 origins = GetOriginsForHost(&client, kTemp, kOriginA.host()); | 235 origins = GetOriginsForHost(&client, kTemp, kOriginA.host()); |
| 237 EXPECT_EQ(origins.size(), 1ul); | 236 EXPECT_EQ(origins.size(), 1ul); |
| 238 EXPECT_TRUE(origins.find(kOriginA) != origins.end()); | 237 EXPECT_TRUE(origins.find(kOriginA) != origins.end()); |
| 239 | 238 |
| 240 mock_tracker()->AddMockDatabase(kOriginB, "barDB", 1000); | 239 mock_tracker()->AddMockDatabase(kOriginB, "barDB", 1000); |
| 241 origins = GetOriginsForHost(&client, kTemp, kOriginA.host()); | 240 origins = GetOriginsForHost(&client, kTemp, kOriginA.host()); |
| 242 EXPECT_EQ(origins.size(), 2ul); | 241 EXPECT_EQ(origins.size(), 2ul); |
| 243 EXPECT_TRUE(origins.find(kOriginA) != origins.end()); | 242 EXPECT_TRUE(origins.find(kOriginA) != origins.end()); |
| 244 EXPECT_TRUE(origins.find(kOriginB) != origins.end()); | 243 EXPECT_TRUE(origins.find(kOriginB) != origins.end()); |
| 245 | 244 |
| 246 EXPECT_TRUE(GetOriginsForHost(&client, kPerm, kOriginA.host()).empty()); | 245 EXPECT_TRUE(GetOriginsForHost(&client, kPerm, kOriginA.host()).empty()); |
| 247 EXPECT_TRUE(GetOriginsForHost(&client, kTemp, kOriginOther.host()).empty()); | 246 EXPECT_TRUE(GetOriginsForHost(&client, kTemp, kOriginOther.host()).empty()); |
| 248 } | 247 } |
| 249 | 248 |
| 250 TEST_F(DatabaseQuotaClientTest, GetOriginsForType) { | 249 TEST_F(DatabaseQuotaClientTest, GetOriginsForType) { |
| 251 DatabaseQuotaClient client(base::ThreadTaskRunnerHandle::Get().get(), | 250 DatabaseQuotaClient client(base::MessageLoopProxy::current().get(), |
| 252 mock_tracker()); | 251 mock_tracker()); |
| 253 | 252 |
| 254 EXPECT_TRUE(GetOriginsForType(&client, kTemp).empty()); | 253 EXPECT_TRUE(GetOriginsForType(&client, kTemp).empty()); |
| 255 EXPECT_TRUE(GetOriginsForType(&client, kPerm).empty()); | 254 EXPECT_TRUE(GetOriginsForType(&client, kPerm).empty()); |
| 256 | 255 |
| 257 mock_tracker()->AddMockDatabase(kOriginA, "fooDB", 1000); | 256 mock_tracker()->AddMockDatabase(kOriginA, "fooDB", 1000); |
| 258 std::set<GURL> origins = GetOriginsForType(&client, kTemp); | 257 std::set<GURL> origins = GetOriginsForType(&client, kTemp); |
| 259 EXPECT_EQ(origins.size(), 1ul); | 258 EXPECT_EQ(origins.size(), 1ul); |
| 260 EXPECT_TRUE(origins.find(kOriginA) != origins.end()); | 259 EXPECT_TRUE(origins.find(kOriginA) != origins.end()); |
| 261 | 260 |
| 262 EXPECT_TRUE(GetOriginsForType(&client, kPerm).empty()); | 261 EXPECT_TRUE(GetOriginsForType(&client, kPerm).empty()); |
| 263 } | 262 } |
| 264 | 263 |
| 265 TEST_F(DatabaseQuotaClientTest, DeleteOriginData) { | 264 TEST_F(DatabaseQuotaClientTest, DeleteOriginData) { |
| 266 DatabaseQuotaClient client(base::ThreadTaskRunnerHandle::Get().get(), | 265 DatabaseQuotaClient client(base::MessageLoopProxy::current().get(), |
| 267 mock_tracker()); | 266 mock_tracker()); |
| 268 | 267 |
| 269 // Perm deletions are short circuited in the Client and | 268 // Perm deletions are short circuited in the Client and |
| 270 // should not reach the DatabaseTracker. | 269 // should not reach the DatabaseTracker. |
| 271 EXPECT_TRUE(DeleteOriginData(&client, kPerm, kOriginA)); | 270 EXPECT_TRUE(DeleteOriginData(&client, kPerm, kOriginA)); |
| 272 EXPECT_EQ(0, mock_tracker()->delete_called_count()); | 271 EXPECT_EQ(0, mock_tracker()->delete_called_count()); |
| 273 | 272 |
| 274 mock_tracker()->set_async_delete(false); | 273 mock_tracker()->set_async_delete(false); |
| 275 EXPECT_TRUE(DeleteOriginData(&client, kTemp, kOriginA)); | 274 EXPECT_TRUE(DeleteOriginData(&client, kTemp, kOriginA)); |
| 276 EXPECT_EQ(1, mock_tracker()->delete_called_count()); | 275 EXPECT_EQ(1, mock_tracker()->delete_called_count()); |
| 277 | 276 |
| 278 mock_tracker()->set_async_delete(true); | 277 mock_tracker()->set_async_delete(true); |
| 279 EXPECT_TRUE(DeleteOriginData(&client, kTemp, kOriginA)); | 278 EXPECT_TRUE(DeleteOriginData(&client, kTemp, kOriginA)); |
| 280 EXPECT_EQ(2, mock_tracker()->delete_called_count()); | 279 EXPECT_EQ(2, mock_tracker()->delete_called_count()); |
| 281 } | 280 } |
| 282 | 281 |
| 283 } // namespace content | 282 } // namespace content |
| OLD | NEW |