Quota: Add quota::StorageType to the GetOriginsCallback definition.

webkit/quota/mock_quota_manager_unittest.cc

 // Copyright (c) 2011 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.
 
 
 #include <set>
 
 #include "base/file_util.h"
 #include "base/memory/scoped_callback_factory.h"
 #include "base/memory/scoped_ptr.h"
@@ skipping 38 matching lines @@
     manager_ = NULL;
     MessageLoop::current()->RunAllPending();
   }
 
   void GetModifiedOrigins(StorageType type, base::Time since) {
     manager_->GetOriginsModifiedSince(type, since,
         callback_factory_.NewCallback(
             &MockQuotaManagerTest::GotModifiedOrigins));
   }
 
-  void GotModifiedOrigins(const std::set<GURL>& origins) {
+  void GotModifiedOrigins(const std::set<GURL>& origins, StorageType type) {
     origins_ = origins;
+    type_ = type;
   }
 
   void DeleteOriginData(const GURL& origin, StorageType type) {
     manager_->DeleteOriginData(origin, type,
         callback_factory_.NewCallback(
             &MockQuotaManagerTest::DeletedOriginData));
   }
 
   void DeletedOriginData(QuotaStatusCode status) {
     ++deletion_callback_count_;
     EXPECT_EQ(quota::kQuotaStatusOk, status);
   }
 
-  int GetDeletionCallbackCount() {
+  int deletion_callback_count() const {
     return deletion_callback_count_;
   }
 
-  MockQuotaManager* GetManager() {
+  MockQuotaManager* manager() const {
     return manager_.get();
   }
 
-  const std::set<GURL> GetOrigins() {
+  const std::set<GURL>& origins() const {
     return origins_;
   }
 
+  const StorageType type() const {
+    return type_;
+  }
+
  private:
   ScopedTempDir data_dir_;
   base::ScopedCallbackFactory<MockQuotaManagerTest> callback_factory_;
   scoped_refptr<MockQuotaManager> manager_;
   scoped_refptr<MockSpecialStoragePolicy> policy_;
 
   int deletion_callback_count_;
 
   std::set<GURL> origins_;
+  StorageType type_;
 
   DISALLOW_COPY_AND_ASSIGN(MockQuotaManagerTest);
 };
 
 TEST_F(MockQuotaManagerTest, BasicOriginManipulation) {
-  EXPECT_FALSE(GetManager()->OriginHasData(kOrigin1, kStorageTypeTemporary));
-  EXPECT_FALSE(GetManager()->OriginHasData(kOrigin2, kStorageTypeTemporary));
-  EXPECT_FALSE(GetManager()->OriginHasData(kOrigin1, kStorageTypePersistent));
-  EXPECT_FALSE(GetManager()->OriginHasData(kOrigin2, kStorageTypePersistent));
+  EXPECT_FALSE(manager()->OriginHasData(kOrigin1, kStorageTypeTemporary));
+  EXPECT_FALSE(manager()->OriginHasData(kOrigin2, kStorageTypeTemporary));
+  EXPECT_FALSE(manager()->OriginHasData(kOrigin1, kStorageTypePersistent));
+  EXPECT_FALSE(manager()->OriginHasData(kOrigin2, kStorageTypePersistent));
 
-  GetManager()->AddOrigin(kOrigin1, kStorageTypeTemporary, base::Time::Now());
-  EXPECT_TRUE(GetManager()->OriginHasData(kOrigin1, kStorageTypeTemporary));
-  EXPECT_FALSE(GetManager()->OriginHasData(kOrigin2, kStorageTypeTemporary));
-  EXPECT_FALSE(GetManager()->OriginHasData(kOrigin1, kStorageTypePersistent));
-  EXPECT_FALSE(GetManager()->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));
 
-  GetManager()->AddOrigin(kOrigin1, kStorageTypePersistent, base::Time::Now());
-  EXPECT_TRUE(GetManager()->OriginHasData(kOrigin1, kStorageTypeTemporary));
-  EXPECT_FALSE(GetManager()->OriginHasData(kOrigin2, kStorageTypeTemporary));
-  EXPECT_TRUE(GetManager()->OriginHasData(kOrigin1, kStorageTypePersistent));
-  EXPECT_FALSE(GetManager()->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));
 
-  GetManager()->AddOrigin(kOrigin2, kStorageTypeTemporary, base::Time::Now());
-  EXPECT_TRUE(GetManager()->OriginHasData(kOrigin1, kStorageTypeTemporary));
-  EXPECT_TRUE(GetManager()->OriginHasData(kOrigin2, kStorageTypeTemporary));
-  EXPECT_TRUE(GetManager()->OriginHasData(kOrigin1, kStorageTypePersistent));
-  EXPECT_FALSE(GetManager()->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));
 }
 
 TEST_F(MockQuotaManagerTest, OriginDeletion) {
-  GetManager()->AddOrigin(kOrigin1, kStorageTypeTemporary, base::Time::Now());
-  GetManager()->AddOrigin(kOrigin2, kStorageTypeTemporary, base::Time::Now());
+  manager()->AddOrigin(kOrigin1, kStorageTypeTemporary, base::Time::Now());
+  manager()->AddOrigin(kOrigin2, kStorageTypeTemporary, base::Time::Now());
 
   DeleteOriginData(kOrigin2, kStorageTypeTemporary);
   MessageLoop::current()->RunAllPending();
 
-  EXPECT_EQ(1, GetDeletionCallbackCount());
-  EXPECT_TRUE(GetManager()->OriginHasData(kOrigin1, kStorageTypeTemporary));
-  EXPECT_FALSE(GetManager()->OriginHasData(kOrigin2, kStorageTypeTemporary));
+  EXPECT_EQ(1, deletion_callback_count());
+  EXPECT_TRUE(manager()->OriginHasData(kOrigin1, kStorageTypeTemporary));
+  EXPECT_FALSE(manager()->OriginHasData(kOrigin2, kStorageTypeTemporary));
 }
 
 TEST_F(MockQuotaManagerTest, ModifiedOrigins) {
   base::Time now = base::Time::Now();
   base::Time then = base::Time();
   base::TimeDelta an_hour = base::TimeDelta::FromMilliseconds(3600000);
   base::TimeDelta a_minute = base::TimeDelta::FromMilliseconds(60000);
 
   GetModifiedOrigins(kStorageTypeTemporary, then);
   MessageLoop::current()->RunAllPending();
-  EXPECT_TRUE(GetOrigins().empty());
+  EXPECT_TRUE(origins().empty());
 
-  GetManager()->AddOrigin(kOrigin1, kStorageTypeTemporary, now - an_hour);
+  manager()->AddOrigin(kOrigin1, kStorageTypeTemporary, now - an_hour);
 
   GetModifiedOrigins(kStorageTypeTemporary, then);
   MessageLoop::current()->RunAllPending();
 
-  EXPECT_EQ(1UL, GetOrigins().size());
-  EXPECT_EQ(1UL, GetOrigins().count(kOrigin1));
-  EXPECT_EQ(0UL, GetOrigins().count(kOrigin2));
+  EXPECT_EQ(kStorageTypeTemporary, type());
+  EXPECT_EQ(1UL, origins().size());
+  EXPECT_EQ(1UL, origins().count(kOrigin1));
+  EXPECT_EQ(0UL, origins().count(kOrigin2));
 
-  GetManager()->AddOrigin(kOrigin2, kStorageTypeTemporary, now);
+  manager()->AddOrigin(kOrigin2, kStorageTypeTemporary, now);
 
   GetModifiedOrigins(kStorageTypeTemporary, then);
   MessageLoop::current()->RunAllPending();
 
-  EXPECT_EQ(2UL, GetOrigins().size());
-  EXPECT_EQ(1UL, GetOrigins().count(kOrigin1));
-  EXPECT_EQ(1UL, GetOrigins().count(kOrigin2));
+  EXPECT_EQ(kStorageTypeTemporary, type());
+  EXPECT_EQ(2UL, origins().size());
+  EXPECT_EQ(1UL, origins().count(kOrigin1));
+  EXPECT_EQ(1UL, origins().count(kOrigin2));
 
   GetModifiedOrigins(kStorageTypeTemporary, now - a_minute);
   MessageLoop::current()->RunAllPending();
 
-  EXPECT_EQ(1UL, GetOrigins().size());
-  EXPECT_EQ(0UL, GetOrigins().count(kOrigin1));
-  EXPECT_EQ(1UL, GetOrigins().count(kOrigin2));
+  EXPECT_EQ(kStorageTypeTemporary, type());
+  EXPECT_EQ(1UL, origins().size());
+  EXPECT_EQ(0UL, origins().count(kOrigin1));
+  EXPECT_EQ(1UL, origins().count(kOrigin2));
 }
 }  // Namespace quota