content/public/test/mock_special_storage_policy.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_SPECIAL_STORAGE_POLICY_H_
   6 #define CONTENT_PUBLIC_TEST_MOCK_SPECIAL_STORAGE_POLICY_H_
   7
   8 #include <set>
   9 #include <string>
  10
  11 #include "storage/browser/quota/special_storage_policy.h"
  12 #include "url/gurl.h"
  13
  14 using storage::SpecialStoragePolicy;
  15
  16 namespace content {
  17
  18 class MockSpecialStoragePolicy : public storage::SpecialStoragePolicy {
  19  public:
  20   MockSpecialStoragePolicy();
  21
  22   bool IsStorageProtected(const GURL& origin) override;
  23   bool IsStorageUnlimited(const GURL& origin) override;
  24   bool IsStorageSessionOnly(const GURL& origin) override;
  25   bool CanQueryDiskSize(const GURL& origin) override;
  26   bool HasIsolatedStorage(const GURL& origin) override;
  27   bool HasSessionOnlyOrigins() override;
  28
  29   void AddProtected(const GURL& origin) {
  30     protected_.insert(origin);
  31   }
  32
  33   void AddUnlimited(const GURL& origin) {
  34     unlimited_.insert(origin);
  35   }
  36
  37   void RemoveUnlimited(const GURL& origin) {
  38     unlimited_.erase(origin);
  39   }
  40
  41   void AddSessionOnly(const GURL& origin) {
  42     session_only_.insert(origin);
  43   }
  44
  45   void GrantQueryDiskSize(const GURL& origin) {
  46     can_query_disk_size_.insert(origin);
  47   }
  48
  49   void AddIsolated(const GURL& origin) {
  50     isolated_.insert(origin);
  51   }
  52
  53   void RemoveIsolated(const GURL& origin) {
  54     isolated_.erase(origin);
  55   }
  56
  57   void SetAllUnlimited(bool all_unlimited) {
  58     all_unlimited_ = all_unlimited;
  59   }
  60
  61   void Reset() {
  62     protected_.clear();
  63     unlimited_.clear();
  64     session_only_.clear();
  65     can_query_disk_size_.clear();
  66     file_handlers_.clear();
  67     isolated_.clear();
  68     all_unlimited_ = false;
  69   }
  70
  71   void NotifyGranted(const GURL& origin, int change_flags) {
  72     SpecialStoragePolicy::NotifyGranted(origin, change_flags);
  73   }
  74
  75   void NotifyRevoked(const GURL& origin, int change_flags) {
  76     SpecialStoragePolicy::NotifyRevoked(origin, change_flags);
  77   }
  78
  79   void NotifyCleared() {
  80     SpecialStoragePolicy::NotifyCleared();
  81   }
  82
  83  protected:
  84   ~MockSpecialStoragePolicy() override;
  85
  86  private:
  87   std::set<GURL> protected_;
  88   std::set<GURL> unlimited_;
  89   std::set<GURL> session_only_;
  90   std::set<GURL> can_query_disk_size_;
  91   std::set<GURL> isolated_;
  92   std::set<std::string> file_handlers_;
  93
  94   bool all_unlimited_;
  95 };
  96 }  // namespace content
  97
  98 #endif  // CONTENT_PUBLIC_TEST_MOCK_SPECIAL_STORAGE_POLICY_H_