| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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 "webkit/quota/mock_storage_client.h" | 5 #include "webkit/quota/mock_storage_client.h" |
| 6 | 6 |
| 7 #include "base/atomic_sequence_num.h" | 7 #include "base/atomic_sequence_num.h" |
| 8 #include "base/basictypes.h" | 8 #include "base/basictypes.h" |
| 9 #include "base/memory/scoped_ptr.h" | 9 #include "base/memory/scoped_ptr.h" |
| 10 #include "base/memory/singleton.h" | 10 #include "base/memory/singleton.h" |
| (...skipping 145 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 156 void MockStorageClient::RunGetOriginsForType( | 156 void MockStorageClient::RunGetOriginsForType( |
| 157 StorageType type, GetOriginsCallback* callback_ptr) { | 157 StorageType type, GetOriginsCallback* callback_ptr) { |
| 158 scoped_ptr<GetOriginsCallback> callback(callback_ptr); | 158 scoped_ptr<GetOriginsCallback> callback(callback_ptr); |
| 159 origins_callbacks_.erase(callback_ptr); | 159 origins_callbacks_.erase(callback_ptr); |
| 160 std::set<GURL> origins; | 160 std::set<GURL> origins; |
| 161 for (OriginDataMap::iterator iter = origin_data_.begin(); | 161 for (OriginDataMap::iterator iter = origin_data_.begin(); |
| 162 iter != origin_data_.end(); ++iter) { | 162 iter != origin_data_.end(); ++iter) { |
| 163 if (type == iter->first.second) | 163 if (type == iter->first.second) |
| 164 origins.insert(iter->first.first); | 164 origins.insert(iter->first.first); |
| 165 } | 165 } |
| 166 callback->Run(origins); | 166 callback->Run(origins, type); |
| 167 } | 167 } |
| 168 | 168 |
| 169 void MockStorageClient::RunGetOriginsForHost( | 169 void MockStorageClient::RunGetOriginsForHost( |
| 170 StorageType type, const std::string& host, | 170 StorageType type, const std::string& host, |
| 171 GetOriginsCallback* callback_ptr) { | 171 GetOriginsCallback* callback_ptr) { |
| 172 scoped_ptr<GetOriginsCallback> callback(callback_ptr); | 172 scoped_ptr<GetOriginsCallback> callback(callback_ptr); |
| 173 origins_callbacks_.erase(callback_ptr); | 173 origins_callbacks_.erase(callback_ptr); |
| 174 std::set<GURL> origins; | 174 std::set<GURL> origins; |
| 175 for (OriginDataMap::iterator iter = origin_data_.begin(); | 175 for (OriginDataMap::iterator iter = origin_data_.begin(); |
| 176 iter != origin_data_.end(); ++iter) { | 176 iter != origin_data_.end(); ++iter) { |
| 177 std::string host_or_spec = net::GetHostOrSpecFromURL(iter->first.first); | 177 std::string host_or_spec = net::GetHostOrSpecFromURL(iter->first.first); |
| 178 if (type == iter->first.second && host == host_or_spec) | 178 if (type == iter->first.second && host == host_or_spec) |
| 179 origins.insert(iter->first.first); | 179 origins.insert(iter->first.first); |
| 180 } | 180 } |
| 181 callback->Run(origins); | 181 callback->Run(origins, type); |
| 182 } | 182 } |
| 183 | 183 |
| 184 void MockStorageClient::RunDeleteOriginData( | 184 void MockStorageClient::RunDeleteOriginData( |
| 185 const GURL& origin_url, | 185 const GURL& origin_url, |
| 186 StorageType type, | 186 StorageType type, |
| 187 DeletionCallback* callback_ptr) { | 187 DeletionCallback* callback_ptr) { |
| 188 scoped_ptr<DeletionCallback> callback(callback_ptr); | 188 scoped_ptr<DeletionCallback> callback(callback_ptr); |
| 189 ErrorOriginSet::iterator itr_error = | 189 ErrorOriginSet::iterator itr_error = |
| 190 error_origins_.find(make_pair(origin_url, type)); | 190 error_origins_.find(make_pair(origin_url, type)); |
| 191 if (itr_error != error_origins_.end()) { | 191 if (itr_error != error_origins_.end()) { |
| 192 deletion_callbacks_.erase(callback_ptr); | 192 deletion_callbacks_.erase(callback_ptr); |
| 193 callback->Run(kQuotaErrorInvalidModification); | 193 callback->Run(kQuotaErrorInvalidModification); |
| 194 return; | 194 return; |
| 195 } | 195 } |
| 196 | 196 |
| 197 OriginDataMap::iterator itr = | 197 OriginDataMap::iterator itr = |
| 198 origin_data_.find(make_pair(origin_url, type)); | 198 origin_data_.find(make_pair(origin_url, type)); |
| 199 if (itr != origin_data_.end()) { | 199 if (itr != origin_data_.end()) { |
| 200 int64 delta = itr->second; | 200 int64 delta = itr->second; |
| 201 quota_manager_proxy_-> | 201 quota_manager_proxy_-> |
| 202 NotifyStorageModified(id(), origin_url, type, -delta); | 202 NotifyStorageModified(id(), origin_url, type, -delta); |
| 203 origin_data_.erase(itr); | 203 origin_data_.erase(itr); |
| 204 } | 204 } |
| 205 | 205 |
| 206 deletion_callbacks_.erase(callback_ptr); | 206 deletion_callbacks_.erase(callback_ptr); |
| 207 callback->Run(kQuotaStatusOk); | 207 callback->Run(kQuotaStatusOk); |
| 208 } | 208 } |
| 209 | 209 |
| 210 } // namespace quota | 210 } // namespace quota |
| OLD | NEW |