| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 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 | 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 WEBKIT_QUOTA_MOCK_SPECIAL_STORAGE_POLICY_H_ | 5 #ifndef WEBKIT_QUOTA_MOCK_SPECIAL_STORAGE_POLICY_H_ |
| 6 #define WEBKIT_QUOTA_MOCK_SPECIAL_STORAGE_POLICY_H_ | 6 #define WEBKIT_QUOTA_MOCK_SPECIAL_STORAGE_POLICY_H_ |
| 7 | 7 |
| 8 #include <set> | 8 #include <set> |
| 9 #include <string> | 9 #include <string> |
| 10 #include "googleurl/src/gurl.h" | 10 #include "googleurl/src/gurl.h" |
| 11 #include "webkit/quota/special_storage_policy.h" | 11 #include "webkit/quota/special_storage_policy.h" |
| 12 | 12 |
| 13 namespace quota { | 13 namespace quota { |
| 14 | 14 |
| 15 class MockSpecialStoragePolicy : public quota::SpecialStoragePolicy { | 15 class MockSpecialStoragePolicy : public quota::SpecialStoragePolicy { |
| 16 public: | 16 public: |
| 17 MockSpecialStoragePolicy(); | 17 MockSpecialStoragePolicy(); |
| 18 virtual ~MockSpecialStoragePolicy(); | 18 virtual ~MockSpecialStoragePolicy(); |
| 19 | 19 |
| 20 virtual bool IsStorageProtected(const GURL& origin); | 20 virtual bool IsStorageProtected(const GURL& origin); |
| 21 virtual bool IsStorageUnlimited(const GURL& origin); | 21 virtual bool IsStorageUnlimited(const GURL& origin); |
| 22 virtual bool IsStorageSessionOnly(const GURL& origin); |
| 22 virtual bool IsFileHandler(const std::string& extension_id); | 23 virtual bool IsFileHandler(const std::string& extension_id); |
| 23 | 24 |
| 24 void AddProtected(const GURL& origin) { | 25 void AddProtected(const GURL& origin) { |
| 25 protected_.insert(origin); | 26 protected_.insert(origin); |
| 26 } | 27 } |
| 27 | 28 |
| 28 void AddUnlimited(const GURL& origin) { | 29 void AddUnlimited(const GURL& origin) { |
| 29 unlimited_.insert(origin); | 30 unlimited_.insert(origin); |
| 30 } | 31 } |
| 31 | 32 |
| 32 void AddFileHandler(const std::string& id) { | 33 void AddFileHandler(const std::string& id) { |
| 33 file_handlers_.insert(id); | 34 file_handlers_.insert(id); |
| 34 } | 35 } |
| 35 | 36 |
| 36 void Reset() { | 37 void Reset() { |
| 37 protected_.clear(); | 38 protected_.clear(); |
| 38 unlimited_.clear(); | 39 unlimited_.clear(); |
| 39 file_handlers_.clear(); | 40 file_handlers_.clear(); |
| 40 } | 41 } |
| 41 | 42 |
| 42 private: | 43 private: |
| 43 std::set<GURL> protected_; | 44 std::set<GURL> protected_; |
| 44 std::set<GURL> unlimited_; | 45 std::set<GURL> unlimited_; |
| 46 std::set<GURL> session_only_; |
| 45 std::set<std::string> file_handlers_; | 47 std::set<std::string> file_handlers_; |
| 46 }; | 48 }; |
| 47 } // namespace quota | 49 } // namespace quota |
| 48 | 50 |
| 49 #endif // WEBKIT_QUOTA_MOCK_SPECIAL_STORAGE_POLICY_H_ | 51 #endif // WEBKIT_QUOTA_MOCK_SPECIAL_STORAGE_POLICY_H_ |
| OLD | NEW |