Populate storage_unittests target.

content/public/test/mock_storage_client.cc

-// Copyright 2014 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 "content/public/test/mock_storage_client.h"
-
-#include <memory>
-
-#include "base/bind.h"
-#include "base/location.h"
-#include "base/memory/singleton.h"
-#include "base/single_thread_task_runner.h"
-#include "base/stl_util.h"
-#include "base/threading/thread_task_runner_handle.h"
-#include "net/base/url_util.h"
-#include "storage/browser/quota/quota_manager_proxy.h"
-
-using storage::kQuotaErrorInvalidModification;
-using storage::kQuotaStatusOk;
-
-namespace content {
-
-using std::make_pair;
-
-MockStorageClient::MockStorageClient(
-    QuotaManagerProxy* quota_manager_proxy,
-    const MockOriginData* mock_data, QuotaClient::ID id, size_t mock_data_size)
-    : quota_manager_proxy_(quota_manager_proxy),
-      id_(id),
-      mock_time_counter_(0),
-      weak_factory_(this) {
-  Populate(mock_data, mock_data_size);
-}
-
-void MockStorageClient::Populate(
-    const MockOriginData* mock_data,
-    size_t mock_data_size) {
-  for (size_t i = 0; i < mock_data_size; ++i) {
-    origin_data_[make_pair(GURL(mock_data[i].origin), mock_data[i].type)] =
-        mock_data[i].usage;
-  }
-}
-
-MockStorageClient::~MockStorageClient() {}
-
-void MockStorageClient::AddOriginAndNotify(const GURL& origin_url,
-                                           StorageType type,
-                                           int64_t size) {
-  DCHECK(origin_data_.find(make_pair(origin_url, type)) == origin_data_.end());
-  DCHECK_GE(size, 0);
-  origin_data_[make_pair(origin_url, type)] = size;
-  quota_manager_proxy_->quota_manager()->NotifyStorageModifiedInternal(
-      id(), origin_url, type, size, IncrementMockTime());
-}
-
-void MockStorageClient::ModifyOriginAndNotify(const GURL& origin_url,
-                                              StorageType type,
-                                              int64_t delta) {
-  OriginDataMap::iterator find = origin_data_.find(make_pair(origin_url, type));
-  DCHECK(find != origin_data_.end());
-  find->second += delta;
-  DCHECK_GE(find->second, 0);
-
-  // TODO(tzik): Check quota to prevent usage exceed
-  quota_manager_proxy_->quota_manager()->NotifyStorageModifiedInternal(
-      id(), origin_url, type, delta, IncrementMockTime());
-}
-
-void MockStorageClient::TouchAllOriginsAndNotify() {
-  for (OriginDataMap::const_iterator itr = origin_data_.begin();
-       itr != origin_data_.end();
-       ++itr) {
-    quota_manager_proxy_->quota_manager()->NotifyStorageModifiedInternal(
-        id(), itr->first.first, itr->first.second, 0, IncrementMockTime());
-  }
-}
-
-void MockStorageClient::AddOriginToErrorSet(
-    const GURL& origin_url, StorageType type) {
-  error_origins_.insert(make_pair(origin_url, type));
-}
-
-base::Time MockStorageClient::IncrementMockTime() {
-  ++mock_time_counter_;
-  return base::Time::FromDoubleT(mock_time_counter_ * 10.0);
-}
-
-QuotaClient::ID MockStorageClient::id() const {
-  return id_;
-}
-
-void MockStorageClient::OnQuotaManagerDestroyed() {
-  delete this;
-}
-
-void MockStorageClient::GetOriginUsage(const GURL& origin_url,
-                                       StorageType type,
-                                       const GetUsageCallback& callback) {
-  base::ThreadTaskRunnerHandle::Get()->PostTask(
-      FROM_HERE,
-      base::Bind(&MockStorageClient::RunGetOriginUsage,
-                 weak_factory_.GetWeakPtr(), origin_url, type, callback));
-}
-
-void MockStorageClient::GetOriginsForType(
-    StorageType type, const GetOriginsCallback& callback) {
-  base::ThreadTaskRunnerHandle::Get()->PostTask(
-      FROM_HERE, base::Bind(&MockStorageClient::RunGetOriginsForType,
-                            weak_factory_.GetWeakPtr(), type, callback));
-}
-
-void MockStorageClient::GetOriginsForHost(
-    StorageType type, const std::string& host,
-    const GetOriginsCallback& callback) {
-  base::ThreadTaskRunnerHandle::Get()->PostTask(
-      FROM_HERE, base::Bind(&MockStorageClient::RunGetOriginsForHost,
-                            weak_factory_.GetWeakPtr(), type, host, callback));
-}
-
-void MockStorageClient::DeleteOriginData(
-    const GURL& origin, StorageType type,
-    const DeletionCallback& callback) {
-  base::ThreadTaskRunnerHandle::Get()->PostTask(
-      FROM_HERE,
-      base::Bind(&MockStorageClient::RunDeleteOriginData,
-                 weak_factory_.GetWeakPtr(), origin, type, callback));
-}
-
-bool MockStorageClient::DoesSupport(storage::StorageType type) const {
-  return true;
-}
-
-void MockStorageClient::RunGetOriginUsage(
-    const GURL& origin_url, StorageType type,
-    const GetUsageCallback& callback) {
-  OriginDataMap::iterator find = origin_data_.find(make_pair(origin_url, type));
-  if (find == origin_data_.end()) {
-    callback.Run(0);
-  } else {
-    callback.Run(find->second);
-  }
-}
-
-void MockStorageClient::RunGetOriginsForType(
-    StorageType type, const GetOriginsCallback& callback) {
-  std::set<GURL> origins;
-  for (OriginDataMap::iterator iter = origin_data_.begin();
-       iter != origin_data_.end(); ++iter) {
-    if (type == iter->first.second)
-      origins.insert(iter->first.first);
-  }
-  callback.Run(origins);
-}
-
-void MockStorageClient::RunGetOriginsForHost(
-    StorageType type, const std::string& host,
-    const GetOriginsCallback& callback) {
-  std::set<GURL> origins;
-  for (OriginDataMap::iterator iter = origin_data_.begin();
-       iter != origin_data_.end(); ++iter) {
-    std::string host_or_spec = net::GetHostOrSpecFromURL(iter->first.first);
-    if (type == iter->first.second && host == host_or_spec)
-      origins.insert(iter->first.first);
-  }
-  callback.Run(origins);
-}
-
-void MockStorageClient::RunDeleteOriginData(
-    const GURL& origin_url,
-    StorageType type,
-    const DeletionCallback& callback) {
-  ErrorOriginSet::iterator itr_error =
-      error_origins_.find(make_pair(origin_url, type));
-  if (itr_error != error_origins_.end()) {
-    callback.Run(kQuotaErrorInvalidModification);
-    return;
-  }
-
-  OriginDataMap::iterator itr =
-      origin_data_.find(make_pair(origin_url, type));
-  if (itr != origin_data_.end()) {
-    int64_t delta = itr->second;
-    quota_manager_proxy_->
-        NotifyStorageModified(id(), origin_url, type, -delta);
-    origin_data_.erase(itr);
-  }
-
-  callback.Run(kQuotaStatusOk);
-}
-
-}  // namespace content