AppCache + Quota integration

webkit/appcache/appcache_database_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 "testing/gtest/include/gtest/gtest.h"
 
 #include "app/sql/connection.h"
 #include "base/file_util.h"
 #include "base/scoped_temp_dir.h"
 #include "webkit/appcache/appcache_database.h"
@@ skipping 522 matching lines @@
 
   // Ensure that we can delete from the table.
   EXPECT_TRUE(db.DeleteDeletableResponseIds(ids));
   ids.clear();
   EXPECT_TRUE(db.GetDeletableResponseIds(&ids, kint64max, 100));
   EXPECT_EQ(5U, ids.size());
   for (int i = 0; i < static_cast<int>(ids.size()); ++i)
     EXPECT_EQ(i + 5, ids[i]);
 }
 
-TEST(AppCacheDatabaseTest, Quotas) {
+TEST(AppCacheDatabaseTest, OriginUsage) {
   const GURL kManifestUrl("http://blah/manifest");
   const GURL kManifestUrl2("http://blah/manifest2");
   const GURL kOrigin(kManifestUrl.GetOrigin());
   const GURL kOtherOriginManifestUrl("http://other/manifest");
   const GURL kOtherOrigin(kOtherOriginManifestUrl.GetOrigin());
 
   const FilePath kEmptyPath;
   AppCacheDatabase db(kEmptyPath);
   EXPECT_TRUE(db.LazyOpen(true));
 
   scoped_refptr<TestErrorDelegate> error_delegate(new TestErrorDelegate);
   db.db_->set_error_delegate(error_delegate);
 
   std::vector<AppCacheDatabase::CacheRecord> cache_records;
-  EXPECT_EQ(db.GetDefaultOriginQuota(), db.GetOriginQuota(kOrigin));
   EXPECT_EQ(0, db.GetOriginUsage(kOrigin));
   EXPECT_TRUE(db.FindCachesForOrigin(kOrigin, &cache_records));
   EXPECT_TRUE(cache_records.empty());
 
   AppCacheDatabase::GroupRecord group_record;
   group_record.group_id = 1;
   group_record.manifest_url = kManifestUrl;
   group_record.origin = kOrigin;
   EXPECT_TRUE(db.InsertGroup(&group_record));
   AppCacheDatabase::CacheRecord cache_record;
@@ skipping 30 matching lines @@
   cache_record.cache_size = 5000;
   EXPECT_TRUE(db.InsertCache(&cache_record));
 
   EXPECT_EQ(5000, db.GetOriginUsage(kOtherOrigin));
 
   EXPECT_TRUE(db.FindCachesForOrigin(kOrigin, &cache_records));
   EXPECT_EQ(2U, cache_records.size());
   cache_records.clear();
   EXPECT_TRUE(db.FindCachesForOrigin(kOtherOrigin, &cache_records));
   EXPECT_EQ(1U, cache_records.size());
+
+  std::map<GURL, int64> usage_map;
+  EXPECT_TRUE(db.GetAllOriginUsage(&usage_map));
+  EXPECT_EQ(2U, usage_map.size());
+  EXPECT_EQ(1100, usage_map[kOrigin]);
+  EXPECT_EQ(5000, usage_map[kOtherOrigin]);
 }
 
 }  // namespace appcache