| OLD | NEW | 
|    1 // Copyright 2014 The Chromium Authors. All rights reserved. |    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 |    2 // Use of this source code is governed by a BSD-style license that can be | 
|    3 // found in the LICENSE file. |    3 // found in the LICENSE file. | 
|    4  |    4  | 
|    5 #ifndef CONTENT_PUBLIC_TEST_MOCK_SPECIAL_STORAGE_POLICY_H_ |    5 #ifndef CONTENT_PUBLIC_TEST_MOCK_SPECIAL_STORAGE_POLICY_H_ | 
|    6 #define CONTENT_PUBLIC_TEST_MOCK_SPECIAL_STORAGE_POLICY_H_ |    6 #define CONTENT_PUBLIC_TEST_MOCK_SPECIAL_STORAGE_POLICY_H_ | 
|    7  |    7  | 
|    8 #include <set> |    8 #include <set> | 
|    9 #include <string> |    9 #include <string> | 
|   10  |   10  | 
|   11 #include "storage/browser/quota/special_storage_policy.h" |   11 #include "storage/browser/quota/special_storage_policy.h" | 
|   12 #include "url/gurl.h" |   12 #include "url/gurl.h" | 
|   13  |   13  | 
|   14 using storage::SpecialStoragePolicy; |   14 using storage::SpecialStoragePolicy; | 
|   15  |   15  | 
|   16 namespace content { |   16 namespace content { | 
|   17  |   17  | 
|   18 class MockSpecialStoragePolicy : public storage::SpecialStoragePolicy { |   18 class MockSpecialStoragePolicy : public storage::SpecialStoragePolicy { | 
|   19  public: |   19  public: | 
|   20   MockSpecialStoragePolicy(); |   20   MockSpecialStoragePolicy(); | 
|   21  |   21  | 
|   22   bool IsStorageProtected(const GURL& origin) override; |   22   bool IsStorageProtected(const GURL& origin) override; | 
|   23   bool IsStorageUnlimited(const GURL& origin) override; |   23   bool IsStorageUnlimited(const GURL& origin) override; | 
|   24   bool IsStorageSessionOnly(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; |   25   bool HasIsolatedStorage(const GURL& origin) override; | 
|   27   bool HasSessionOnlyOrigins() override; |   26   bool HasSessionOnlyOrigins() override; | 
|   28   bool IsStorageDurable(const GURL& origin) override; |   27   bool IsStorageDurable(const GURL& origin) override; | 
|   29  |   28  | 
|   30   void AddProtected(const GURL& origin) { |   29   void AddProtected(const GURL& origin) { | 
|   31     protected_.insert(origin); |   30     protected_.insert(origin); | 
|   32   } |   31   } | 
|   33  |   32  | 
|   34   void AddUnlimited(const GURL& origin) { |   33   void AddUnlimited(const GURL& origin) { | 
|   35     unlimited_.insert(origin); |   34     unlimited_.insert(origin); | 
|   36   } |   35   } | 
|   37  |   36  | 
|   38   void RemoveUnlimited(const GURL& origin) { |   37   void RemoveUnlimited(const GURL& origin) { | 
|   39     unlimited_.erase(origin); |   38     unlimited_.erase(origin); | 
|   40   } |   39   } | 
|   41  |   40  | 
|   42   void AddSessionOnly(const GURL& origin) { |   41   void AddSessionOnly(const GURL& origin) { | 
|   43     session_only_.insert(origin); |   42     session_only_.insert(origin); | 
|   44   } |   43   } | 
|   45  |   44  | 
|   46   void GrantQueryDiskSize(const GURL& origin) { |  | 
|   47     can_query_disk_size_.insert(origin); |  | 
|   48   } |  | 
|   49  |  | 
|   50   void AddIsolated(const GURL& origin) { |   45   void AddIsolated(const GURL& origin) { | 
|   51     isolated_.insert(origin); |   46     isolated_.insert(origin); | 
|   52   } |   47   } | 
|   53  |   48  | 
|   54   void RemoveIsolated(const GURL& origin) { |   49   void RemoveIsolated(const GURL& origin) { | 
|   55     isolated_.erase(origin); |   50     isolated_.erase(origin); | 
|   56   } |   51   } | 
|   57  |   52  | 
|   58   void SetAllUnlimited(bool all_unlimited) { |   53   void SetAllUnlimited(bool all_unlimited) { | 
|   59     all_unlimited_ = all_unlimited; |   54     all_unlimited_ = all_unlimited; | 
|   60   } |   55   } | 
|   61  |   56  | 
|   62   void AddDurable(const GURL& origin) { |   57   void AddDurable(const GURL& origin) { | 
|   63     durable_.insert(origin); |   58     durable_.insert(origin); | 
|   64   } |   59   } | 
|   65  |   60  | 
|   66   void Reset() { |   61   void Reset() { | 
|   67     protected_.clear(); |   62     protected_.clear(); | 
|   68     unlimited_.clear(); |   63     unlimited_.clear(); | 
|   69     session_only_.clear(); |   64     session_only_.clear(); | 
|   70     can_query_disk_size_.clear(); |  | 
|   71     file_handlers_.clear(); |   65     file_handlers_.clear(); | 
|   72     isolated_.clear(); |   66     isolated_.clear(); | 
|   73     all_unlimited_ = false; |   67     all_unlimited_ = false; | 
|   74   } |   68   } | 
|   75  |   69  | 
|   76   void NotifyGranted(const GURL& origin, int change_flags) { |   70   void NotifyGranted(const GURL& origin, int change_flags) { | 
|   77     SpecialStoragePolicy::NotifyGranted(origin, change_flags); |   71     SpecialStoragePolicy::NotifyGranted(origin, change_flags); | 
|   78   } |   72   } | 
|   79  |   73  | 
|   80   void NotifyRevoked(const GURL& origin, int change_flags) { |   74   void NotifyRevoked(const GURL& origin, int change_flags) { | 
|   81     SpecialStoragePolicy::NotifyRevoked(origin, change_flags); |   75     SpecialStoragePolicy::NotifyRevoked(origin, change_flags); | 
|   82   } |   76   } | 
|   83  |   77  | 
|   84   void NotifyCleared() { |   78   void NotifyCleared() { | 
|   85     SpecialStoragePolicy::NotifyCleared(); |   79     SpecialStoragePolicy::NotifyCleared(); | 
|   86   } |   80   } | 
|   87  |   81  | 
|   88  protected: |   82  protected: | 
|   89   ~MockSpecialStoragePolicy() override; |   83   ~MockSpecialStoragePolicy() override; | 
|   90  |   84  | 
|   91  private: |   85  private: | 
|   92   std::set<GURL> protected_; |   86   std::set<GURL> protected_; | 
|   93   std::set<GURL> unlimited_; |   87   std::set<GURL> unlimited_; | 
|   94   std::set<GURL> session_only_; |   88   std::set<GURL> session_only_; | 
|   95   std::set<GURL> can_query_disk_size_; |  | 
|   96   std::set<GURL> isolated_; |   89   std::set<GURL> isolated_; | 
|   97   std::set<GURL> durable_; |   90   std::set<GURL> durable_; | 
|   98   std::set<std::string> file_handlers_; |   91   std::set<std::string> file_handlers_; | 
|   99  |   92  | 
|  100   bool all_unlimited_; |   93   bool all_unlimited_; | 
|  101 }; |   94 }; | 
|  102 }  // namespace content |   95 }  // namespace content | 
|  103  |   96  | 
|  104 #endif  // CONTENT_PUBLIC_TEST_MOCK_SPECIAL_STORAGE_POLICY_H_ |   97 #endif  // CONTENT_PUBLIC_TEST_MOCK_SPECIAL_STORAGE_POLICY_H_ | 
| OLD | NEW |