chrome/browser/browsing_data/mock_browsing_data_cache_storage_helper.h

   1 // Copyright 2015 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 CHROME_BROWSER_BROWSING_DATA_MOCK_BROWSING_DATA_CACHE_STORAGE_HELPER_H_
   6 #define CHROME_BROWSER_BROWSING_DATA_MOCK_BROWSING_DATA_CACHE_STORAGE_HELPER_H_
   7
   8 #include <list>
   9 #include <map>
  10
  11 #include "base/callback.h"
  12 #include "chrome/browser/browsing_data/browsing_data_cache_storage_helper.h"
  13
  14 class Profile;
  15
  16 // Mock for BrowsingDataCacheStorageHelper.
  17 // Use AddCacheStorageSamples() or add directly to response_ list, then
  18 // call Notify().
  19 class MockBrowsingDataCacheStorageHelper
  20     : public BrowsingDataCacheStorageHelper {
  21  public:
  22   explicit MockBrowsingDataCacheStorageHelper(Profile* profile);
  23
  24   // Adds some CacheStorageUsageInfo samples.
  25   void AddCacheStorageSamples();
  26
  27   // Notifies the callback.
  28   void Notify();
  29
  30   // Marks all cache storage files as existing.
  31   void Reset();
  32
  33   // Returns true if all cache storage files were deleted since the last
  34   // Reset() invokation.
  35   bool AllDeleted();
  36
  37   // BrowsingDataCacheStorageHelper.
  38   void StartFetching(const base::Callback<
  39                      void(const std::list<content::CacheStorageUsageInfo>&)>&
  40                          callback) override;
  41   void DeleteCacheStorage(const GURL& origin) override;
  42
  43  private:
  44   ~MockBrowsingDataCacheStorageHelper() override;
  45
  46   FetchCallback callback_;
  47   std::map<GURL, bool> origins_;
  48   std::list<content::CacheStorageUsageInfo> response_;
  49
  50   DISALLOW_COPY_AND_ASSIGN(MockBrowsingDataCacheStorageHelper);
  51 };
  52
  53 #endif  // CHROME_BROWSER_BROWSING_DATA_MOCK_BROWSING_DATA_CACHE_STORAGE_HELPER_H_