Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(94)

Side by Side Diff: content/public/test/mock_special_storage_policy.h

Issue 1782053004: Change how the quota system computes the total poolsize for temporary storage (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: rebase Created 3 years, 10 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
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_
OLDNEW
« no previous file with comments | « content/public/browser/content_browser_client.cc ('k') | content/public/test/mock_special_storage_policy.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698