src/content/public/test/mock_storage_client.h

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef CONTENT_PUBLIC_TEST_MOCK_STORAGE_CLIENT_H_
   6 #define CONTENT_PUBLIC_TEST_MOCK_STORAGE_CLIENT_H_
   7
   8 #include <map>
   9 #include <set>
  10 #include <string>
  11 #include <utility>
  12
  13 #include "base/compiler_specific.h"
  14 #include "base/memory/weak_ptr.h"
  15 #include "base/time/time.h"
  16 #include "storage/browser/quota/quota_client.h"
  17 #include "url/gurl.h"
  18
  19 namespace storage {
  20 class QuotaManagerProxy;
  21 }
  22
  23 using storage::QuotaClient;
  24 using storage::QuotaManagerProxy;
  25 using storage::StorageType;
  26
  27 namespace content {
  28
  29 struct MockOriginData {
  30   const char* origin;
  31   StorageType type;
  32   int64 usage;
  33 };
  34
  35 // Mock storage class for testing.
  36 class MockStorageClient : public QuotaClient {
  37  public:
  38   MockStorageClient(QuotaManagerProxy* quota_manager_proxy,
  39                     const MockOriginData* mock_data,
  40                     QuotaClient::ID id,
  41                     size_t mock_data_size);
  42   virtual ~MockStorageClient();
  43
  44   // To add or modify mock data in this client.
  45   void AddOriginAndNotify(
  46       const GURL& origin_url, StorageType type, int64 size);
  47   void ModifyOriginAndNotify(
  48       const GURL& origin_url, StorageType type, int64 delta);
  49   void TouchAllOriginsAndNotify();
  50
  51   void AddOriginToErrorSet(const GURL& origin_url, StorageType type);
  52
  53   base::Time IncrementMockTime();
  54
  55   // QuotaClient methods.
  56   virtual QuotaClient::ID id() const OVERRIDE;
  57   virtual void OnQuotaManagerDestroyed() OVERRIDE;
  58   virtual void GetOriginUsage(const GURL& origin_url,
  59                               StorageType type,
  60                               const GetUsageCallback& callback) OVERRIDE;
  61   virtual void GetOriginsForType(StorageType type,
  62                                  const GetOriginsCallback& callback) OVERRIDE;
  63   virtual void GetOriginsForHost(StorageType type, const std::string& host,
  64                                  const GetOriginsCallback& callback) OVERRIDE;
  65   virtual void DeleteOriginData(const GURL& origin,
  66                                 StorageType type,
  67                                 const DeletionCallback& callback) OVERRIDE;
  68   virtual bool DoesSupport(storage::StorageType type) const OVERRIDE;
  69
  70  private:
  71   void RunGetOriginUsage(const GURL& origin_url,
  72                          StorageType type,
  73                          const GetUsageCallback& callback);
  74   void RunGetOriginsForType(StorageType type,
  75                             const GetOriginsCallback& callback);
  76   void RunGetOriginsForHost(StorageType type,
  77                             const std::string& host,
  78                             const GetOriginsCallback& callback);
  79   void RunDeleteOriginData(const GURL& origin_url,
  80                            StorageType type,
  81                            const DeletionCallback& callback);
  82
  83   void Populate(const MockOriginData* mock_data, size_t mock_data_size);
  84
  85   scoped_refptr<QuotaManagerProxy> quota_manager_proxy_;
  86   const ID id_;
  87
  88   typedef std::map<std::pair<GURL, StorageType>, int64> OriginDataMap;
  89   OriginDataMap origin_data_;
  90   typedef std::set<std::pair<GURL, StorageType> > ErrorOriginSet;
  91   ErrorOriginSet error_origins_;
  92
  93   int mock_time_counter_;
  94
  95   base::WeakPtrFactory<MockStorageClient> weak_factory_;
  96
  97   DISALLOW_COPY_AND_ASSIGN(MockStorageClient);
  98 };
  99
 100 }  // namespace content
 101
 102 #endif  // CONTENT_PUBLIC_TEST_MOCK_STORAGE_CLIENT_H_