chrome/browser/mock_browsing_data_quota_helper.h

   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
   3 // found in the LICENSE file.
   4
   5 #ifndef CHROME_BROWSER_MOCK_BROWSING_DATA_QUOTA_HELPER_H_
   6 #define CHROME_BROWSER_MOCK_BROWSING_DATA_QUOTA_HELPER_H_
   7 #pragma once
   8
   9 #include <list>
  10 #include <string>
  11
  12 #include "base/compiler_specific.h"
  13 #include "chrome/browser/browsing_data_quota_helper.h"
  14
  15 class MockBrowsingDataQuotaHelper : public BrowsingDataQuotaHelper {
  16  public:
  17   explicit MockBrowsingDataQuotaHelper(Profile* profile);
  18
  19   virtual void StartFetching(const FetchResultCallback& callback) OVERRIDE;
  20   virtual void CancelNotification() OVERRIDE;
  21   virtual void RevokeHostQuota(const std::string& host) OVERRIDE;
  22
  23   void AddHost(const std::string& host,
  24                int64 temporary_usage,
  25                int64 persistent_usage);
  26   void AddQuotaSamples();
  27   void Notify();
  28
  29  private:
  30   virtual ~MockBrowsingDataQuotaHelper();
  31
  32   FetchResultCallback callback_;
  33   std::list<QuotaInfo> response_;
  34   Profile* profile_;
  35 };
  36
  37 #endif  // CHROME_BROWSER_MOCK_BROWSING_DATA_QUOTA_HELPER_H_