QuotaManager::DeleteOriginData now allows deletion of specific QuotaClients

webkit/quota/mock_quota_manager.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 "webkit/quota/mock_quota_manager.h"
 
 #include <set>
 #include <string>
 #include <vector>
 
@@ skipping 57 matching lines @@
 
  private:
   scoped_ptr<StatusCallback> callback_;
 
   DISALLOW_COPY_AND_ASSIGN(DeleteOriginDataTask);
 };
 
 MockQuotaManager::OriginInfo::OriginInfo(
     const GURL& origin,
     StorageType type,
+    int client_mask,

[jochen (gone - plz use gerrit), 2011/09/12 11:23:00]

in the header this is quota_client_mask

     base::Time modified)
     : origin(origin),
       type(type),
+      mask(client_mask),
       modified(modified) {
 }
 
 MockQuotaManager::OriginInfo::~OriginInfo() {}
 
 MockQuotaManager::MockQuotaManager(bool is_incognito,
     const FilePath& profile_path, base::MessageLoopProxy* io_thread,
     base::MessageLoopProxy* db_thread,
     SpecialStoragePolicy* special_storage_policy)
     : QuotaManager(is_incognito, profile_path, io_thread, db_thread,
         special_storage_policy) {
 }
 
 MockQuotaManager::~MockQuotaManager() {}
 
 bool MockQuotaManager::AddOrigin(const GURL& origin, StorageType type,
-    base::Time modified) {
-  origins_.push_back(OriginInfo(origin, type, modified));
+    int quota_client_mask, base::Time modified) {
+  origins_.push_back(OriginInfo(origin, type, quota_client_mask, modified));
   return true;
 }
 
 bool MockQuotaManager::OriginHasData(const GURL& origin,
-    StorageType type) const {
+    StorageType type, QuotaClient::ID quota_client) const {
   for (std::vector<OriginInfo>::const_iterator current = origins_.begin();
        current != origins_.end();
        ++current) {
-    if (current->origin == origin && current->type == type)
+    if (current->origin == origin && current->type == type &&
+        (current->mask & quota_client))
       return true;
   }
   return false;
 }
 
 void MockQuotaManager::GetOriginsModifiedSince(StorageType type,
     base::Time modified_since, GetOriginsCallback* callback) {
   std::set<GURL> origins_to_return;
   for (std::vector<OriginInfo>::const_iterator current = origins_.begin();
        current != origins_.end();
        ++current) {
     if (current->type == type && current->modified >= modified_since)
       origins_to_return.insert(current->origin);
   }
   make_scoped_refptr(new GetModifiedSinceTask(this, origins_to_return, type,
       callback))->Start();
 }
 
 void MockQuotaManager::DeleteOriginData(const GURL& origin, StorageType type,
-    StatusCallback* callback) {
+    int quota_client_mask, StatusCallback* callback) {
   for (std::vector<OriginInfo>::iterator current = origins_.begin();
        current != origins_.end();
        ++current) {
     if (current->origin == origin && current->type == type) {
-      origins_.erase(current);
+      // Modify the current bitmask, and if it's 0 after "deletion", remove the
+      // origin entirely.
+      current->mask &= ~quota_client_mask;
+      if (current->mask == 0)
+        origins_.erase(current);
       break;
     }
   }
   make_scoped_refptr(new DeleteOriginDataTask(this, callback))->Start();
 }
 
 }  // namespace quota