src/chrome/browser/extensions/mock_extension_special_storage_policy.h

   1 // Copyright (c) 2012 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_EXTENSIONS_MOCK_EXTENSION_SPECIAL_STORAGE_POLICY_H_
   6 #define CHROME_BROWSER_EXTENSIONS_MOCK_EXTENSION_SPECIAL_STORAGE_POLICY_H_
   7
   8 #include <set>
   9 #include <string>
  10
  11 #include "chrome/browser/extensions/extension_special_storage_policy.h"
  12 #include "url/gurl.h"
  13
  14 // This class is the same as MockSpecialStoragePolicy (in
  15 // webkit/browser/quota/mock_special_storage_policy.h), but it inherits
  16 // ExtensionSpecialStoragePolicy instead of quota::SpecialStoragePolicy.
  17 class MockExtensionSpecialStoragePolicy : public ExtensionSpecialStoragePolicy {
  18  public:
  19   MockExtensionSpecialStoragePolicy();
  20
  21   // quota::SpecialStoragePolicy:
  22   virtual bool IsStorageProtected(const GURL& origin) OVERRIDE;
  23   virtual bool IsStorageUnlimited(const GURL& origin) OVERRIDE;
  24   virtual bool IsStorageSessionOnly(const GURL& origin) OVERRIDE;
  25   virtual bool CanQueryDiskSize(const GURL& origin) OVERRIDE;
  26   virtual bool IsFileHandler(const std::string& extension_id) OVERRIDE;
  27   virtual bool HasSessionOnlyOrigins() OVERRIDE;
  28
  29   void AddProtected(const GURL& origin) {
  30     protected_.insert(origin);
  31   }
  32
  33  private:
  34   virtual ~MockExtensionSpecialStoragePolicy();
  35
  36   std::set<GURL> protected_;
  37
  38   DISALLOW_COPY_AND_ASSIGN(MockExtensionSpecialStoragePolicy);
  39 };
  40
  41 #endif  // CHROME_BROWSER_EXTENSIONS_MOCK_EXTENSION_SPECIAL_STORAGE_POLICY_H_