Index: webkit/quota/mock_quota_manager_unittest.cc |
diff --git a/webkit/quota/mock_quota_manager_unittest.cc b/webkit/quota/mock_quota_manager_unittest.cc |
index 5f1d1b61b5aff7fb3fa065d9ba713f142cd97981..30229aff7cacf957e5bb436aca9e9b72d7e593cf 100644 |
--- a/webkit/quota/mock_quota_manager_unittest.cc |
+++ b/webkit/quota/mock_quota_manager_unittest.cc |
@@ -1,4 +1,4 @@ |
-// Copyright (c) 2011 The Chromium Authors. All rights reserved. |
+// Copyright (c) 2012 The Chromium Authors. All rights reserved. |
// Use of this source code is governed by a BSD-style license that can be |
// found in the LICENSE file. |
@@ -26,6 +26,12 @@ const GURL kOrigin1(kTestOrigin1); |
const GURL kOrigin2(kTestOrigin2); |
const GURL kOrigin3(kTestOrigin3); |
+const StorageType kTemporary = kStorageTypeTemporary; |
+const StorageType kPersistent = kStorageTypePersistent; |
+ |
+const QuotaClient::ID kClientFile = QuotaClient::kFileSystem; |
+const QuotaClient::ID kClientDB = QuotaClient::kIndexedDatabase; |
+ |
class MockQuotaManagerTest : public testing::Test { |
public: |
MockQuotaManagerTest() |
@@ -62,9 +68,10 @@ class MockQuotaManagerTest : public testing::Test { |
type_ = type; |
} |
- void DeleteOriginData(const GURL& origin, StorageType type) { |
+ void DeleteOriginData(const GURL& origin, StorageType type, |
+ int quota_client_mask) { |
manager_->DeleteOriginData( |
- origin, type, |
+ origin, type, quota_client_mask, |
base::Bind(&MockQuotaManagerTest::DeletedOriginData, |
weak_factory_.GetWeakPtr())); |
} |
@@ -105,40 +112,73 @@ class MockQuotaManagerTest : public testing::Test { |
}; |
TEST_F(MockQuotaManagerTest, BasicOriginManipulation) { |
- EXPECT_FALSE(manager()->OriginHasData(kOrigin1, kStorageTypeTemporary)); |
- EXPECT_FALSE(manager()->OriginHasData(kOrigin2, kStorageTypeTemporary)); |
- EXPECT_FALSE(manager()->OriginHasData(kOrigin1, kStorageTypePersistent)); |
- EXPECT_FALSE(manager()->OriginHasData(kOrigin2, kStorageTypePersistent)); |
- |
- manager()->AddOrigin(kOrigin1, kStorageTypeTemporary, base::Time::Now()); |
- EXPECT_TRUE(manager()->OriginHasData(kOrigin1, kStorageTypeTemporary)); |
- EXPECT_FALSE(manager()->OriginHasData(kOrigin2, kStorageTypeTemporary)); |
- EXPECT_FALSE(manager()->OriginHasData(kOrigin1, kStorageTypePersistent)); |
- EXPECT_FALSE(manager()->OriginHasData(kOrigin2, kStorageTypePersistent)); |
- |
- manager()->AddOrigin(kOrigin1, kStorageTypePersistent, base::Time::Now()); |
- EXPECT_TRUE(manager()->OriginHasData(kOrigin1, kStorageTypeTemporary)); |
- EXPECT_FALSE(manager()->OriginHasData(kOrigin2, kStorageTypeTemporary)); |
- EXPECT_TRUE(manager()->OriginHasData(kOrigin1, kStorageTypePersistent)); |
- EXPECT_FALSE(manager()->OriginHasData(kOrigin2, kStorageTypePersistent)); |
- |
- manager()->AddOrigin(kOrigin2, kStorageTypeTemporary, base::Time::Now()); |
- EXPECT_TRUE(manager()->OriginHasData(kOrigin1, kStorageTypeTemporary)); |
- EXPECT_TRUE(manager()->OriginHasData(kOrigin2, kStorageTypeTemporary)); |
- EXPECT_TRUE(manager()->OriginHasData(kOrigin1, kStorageTypePersistent)); |
- EXPECT_FALSE(manager()->OriginHasData(kOrigin2, kStorageTypePersistent)); |
+ EXPECT_FALSE(manager()->OriginHasData(kOrigin1, kTemporary, kClientFile)); |
+ EXPECT_FALSE(manager()->OriginHasData(kOrigin1, kTemporary, kClientDB)); |
+ EXPECT_FALSE(manager()->OriginHasData(kOrigin1, kPersistent, kClientFile)); |
+ EXPECT_FALSE(manager()->OriginHasData(kOrigin1, kPersistent, kClientDB)); |
+ EXPECT_FALSE(manager()->OriginHasData(kOrigin2, kTemporary, kClientFile)); |
+ EXPECT_FALSE(manager()->OriginHasData(kOrigin2, kTemporary, kClientDB)); |
+ EXPECT_FALSE(manager()->OriginHasData(kOrigin2, kPersistent, kClientFile)); |
+ EXPECT_FALSE(manager()->OriginHasData(kOrigin2, kPersistent, kClientDB)); |
+ |
+ manager()->AddOrigin(kOrigin1, kTemporary, kClientFile, base::Time::Now()); |
+ EXPECT_TRUE(manager()->OriginHasData(kOrigin1, kTemporary, kClientFile)); |
+ EXPECT_FALSE(manager()->OriginHasData(kOrigin1, kTemporary, kClientDB)); |
+ EXPECT_FALSE(manager()->OriginHasData(kOrigin1, kPersistent, kClientFile)); |
+ EXPECT_FALSE(manager()->OriginHasData(kOrigin1, kPersistent, kClientDB)); |
+ EXPECT_FALSE(manager()->OriginHasData(kOrigin2, kTemporary, kClientFile)); |
+ EXPECT_FALSE(manager()->OriginHasData(kOrigin2, kTemporary, kClientDB)); |
+ EXPECT_FALSE(manager()->OriginHasData(kOrigin2, kPersistent, kClientFile)); |
+ EXPECT_FALSE(manager()->OriginHasData(kOrigin2, kPersistent, kClientDB)); |
+ |
+ manager()->AddOrigin(kOrigin1, kPersistent, kClientFile, base::Time::Now()); |
+ EXPECT_TRUE(manager()->OriginHasData(kOrigin1, kTemporary, kClientFile)); |
+ EXPECT_FALSE(manager()->OriginHasData(kOrigin1, kTemporary, kClientDB)); |
+ EXPECT_TRUE(manager()->OriginHasData(kOrigin1, kPersistent, kClientFile)); |
+ EXPECT_FALSE(manager()->OriginHasData(kOrigin1, kPersistent, kClientDB)); |
+ EXPECT_FALSE(manager()->OriginHasData(kOrigin2, kTemporary, kClientFile)); |
+ EXPECT_FALSE(manager()->OriginHasData(kOrigin2, kTemporary, kClientDB)); |
+ EXPECT_FALSE(manager()->OriginHasData(kOrigin2, kPersistent, kClientFile)); |
+ EXPECT_FALSE(manager()->OriginHasData(kOrigin2, kPersistent, kClientDB)); |
+ |
+ manager()->AddOrigin(kOrigin2, kTemporary, kClientFile | kClientDB, |
+ base::Time::Now()); |
+ EXPECT_TRUE(manager()->OriginHasData(kOrigin1, kTemporary, kClientFile)); |
+ EXPECT_FALSE(manager()->OriginHasData(kOrigin1, kTemporary, kClientDB)); |
+ EXPECT_TRUE(manager()->OriginHasData(kOrigin1, kPersistent, kClientFile)); |
+ EXPECT_FALSE(manager()->OriginHasData(kOrigin1, kPersistent, kClientDB)); |
+ EXPECT_TRUE(manager()->OriginHasData(kOrigin2, kTemporary, kClientFile)); |
+ EXPECT_TRUE(manager()->OriginHasData(kOrigin2, kTemporary, kClientDB)); |
+ EXPECT_FALSE(manager()->OriginHasData(kOrigin2, kPersistent, kClientFile)); |
+ EXPECT_FALSE(manager()->OriginHasData(kOrigin2, kPersistent, kClientDB)); |
} |
TEST_F(MockQuotaManagerTest, OriginDeletion) { |
- manager()->AddOrigin(kOrigin1, kStorageTypeTemporary, base::Time::Now()); |
- manager()->AddOrigin(kOrigin2, kStorageTypeTemporary, base::Time::Now()); |
+ manager()->AddOrigin(kOrigin1, kTemporary, kClientFile, base::Time::Now()); |
+ manager()->AddOrigin(kOrigin2, kTemporary, kClientFile | kClientDB, |
+ base::Time::Now()); |
+ manager()->AddOrigin(kOrigin3, kTemporary, kClientFile | kClientDB, |
+ base::Time::Now()); |
- DeleteOriginData(kOrigin2, kStorageTypeTemporary); |
+ DeleteOriginData(kOrigin2, kTemporary, kClientFile); |
MessageLoop::current()->RunAllPending(); |
EXPECT_EQ(1, deletion_callback_count()); |
- EXPECT_TRUE(manager()->OriginHasData(kOrigin1, kStorageTypeTemporary)); |
- EXPECT_FALSE(manager()->OriginHasData(kOrigin2, kStorageTypeTemporary)); |
+ EXPECT_TRUE(manager()->OriginHasData(kOrigin1, kTemporary, kClientFile)); |
+ EXPECT_FALSE(manager()->OriginHasData(kOrigin2, kTemporary, kClientFile)); |
+ EXPECT_TRUE(manager()->OriginHasData(kOrigin2, kTemporary, kClientDB)); |
+ EXPECT_TRUE(manager()->OriginHasData(kOrigin3, kTemporary, kClientFile)); |
+ EXPECT_TRUE(manager()->OriginHasData(kOrigin3, kTemporary, kClientDB)); |
+ |
+ DeleteOriginData(kOrigin3, kTemporary, kClientFile | kClientDB); |
+ MessageLoop::current()->RunAllPending(); |
+ |
+ EXPECT_EQ(2, deletion_callback_count()); |
+ EXPECT_TRUE(manager()->OriginHasData(kOrigin1, kTemporary, kClientFile)); |
+ EXPECT_FALSE(manager()->OriginHasData(kOrigin2, kTemporary, kClientFile)); |
+ EXPECT_TRUE(manager()->OriginHasData(kOrigin2, kTemporary, kClientDB)); |
+ EXPECT_FALSE(manager()->OriginHasData(kOrigin3, kTemporary, kClientFile)); |
+ EXPECT_FALSE(manager()->OriginHasData(kOrigin3, kTemporary, kClientDB)); |
} |
TEST_F(MockQuotaManagerTest, ModifiedOrigins) { |
@@ -147,34 +187,34 @@ TEST_F(MockQuotaManagerTest, ModifiedOrigins) { |
base::TimeDelta an_hour = base::TimeDelta::FromMilliseconds(3600000); |
base::TimeDelta a_minute = base::TimeDelta::FromMilliseconds(60000); |
- GetModifiedOrigins(kStorageTypeTemporary, then); |
+ GetModifiedOrigins(kTemporary, then); |
MessageLoop::current()->RunAllPending(); |
EXPECT_TRUE(origins().empty()); |
- manager()->AddOrigin(kOrigin1, kStorageTypeTemporary, now - an_hour); |
+ manager()->AddOrigin(kOrigin1, kTemporary, kClientFile, now - an_hour); |
- GetModifiedOrigins(kStorageTypeTemporary, then); |
+ GetModifiedOrigins(kTemporary, then); |
MessageLoop::current()->RunAllPending(); |
- EXPECT_EQ(kStorageTypeTemporary, type()); |
+ EXPECT_EQ(kTemporary, type()); |
EXPECT_EQ(1UL, origins().size()); |
EXPECT_EQ(1UL, origins().count(kOrigin1)); |
EXPECT_EQ(0UL, origins().count(kOrigin2)); |
- manager()->AddOrigin(kOrigin2, kStorageTypeTemporary, now); |
+ manager()->AddOrigin(kOrigin2, kTemporary, kClientFile, now); |
- GetModifiedOrigins(kStorageTypeTemporary, then); |
+ GetModifiedOrigins(kTemporary, then); |
MessageLoop::current()->RunAllPending(); |
- EXPECT_EQ(kStorageTypeTemporary, type()); |
+ EXPECT_EQ(kTemporary, type()); |
EXPECT_EQ(2UL, origins().size()); |
EXPECT_EQ(1UL, origins().count(kOrigin1)); |
EXPECT_EQ(1UL, origins().count(kOrigin2)); |
- GetModifiedOrigins(kStorageTypeTemporary, now - a_minute); |
+ GetModifiedOrigins(kTemporary, now - a_minute); |
MessageLoop::current()->RunAllPending(); |
- EXPECT_EQ(kStorageTypeTemporary, type()); |
+ EXPECT_EQ(kTemporary, type()); |
EXPECT_EQ(1UL, origins().size()); |
EXPECT_EQ(0UL, origins().count(kOrigin1)); |
EXPECT_EQ(1UL, origins().count(kOrigin2)); |