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 |