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

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

Issue 2368913003: Populate storage_unittests target. (Closed)
Patch Set: Removed unnecessary include from storage/browser/blob/blob_storage_context_unittest.cc. Created 4 years, 2 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
(Empty)
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 bool IsStorageDurable(const GURL& origin) override;
29
30 void AddProtected(const GURL& origin) {
31 protected_.insert(origin);
32 }
33
34 void AddUnlimited(const GURL& origin) {
35 unlimited_.insert(origin);
36 }
37
38 void RemoveUnlimited(const GURL& origin) {
39 unlimited_.erase(origin);
40 }
41
42 void AddSessionOnly(const GURL& origin) {
43 session_only_.insert(origin);
44 }
45
46 void GrantQueryDiskSize(const GURL& origin) {
47 can_query_disk_size_.insert(origin);
48 }
49
50 void AddIsolated(const GURL& origin) {
51 isolated_.insert(origin);
52 }
53
54 void RemoveIsolated(const GURL& origin) {
55 isolated_.erase(origin);
56 }
57
58 void SetAllUnlimited(bool all_unlimited) {
59 all_unlimited_ = all_unlimited;
60 }
61
62 void AddDurable(const GURL& origin) {
63 durable_.insert(origin);
64 }
65
66 void Reset() {
67 protected_.clear();
68 unlimited_.clear();
69 session_only_.clear();
70 can_query_disk_size_.clear();
71 file_handlers_.clear();
72 isolated_.clear();
73 all_unlimited_ = false;
74 }
75
76 void NotifyGranted(const GURL& origin, int change_flags) {
77 SpecialStoragePolicy::NotifyGranted(origin, change_flags);
78 }
79
80 void NotifyRevoked(const GURL& origin, int change_flags) {
81 SpecialStoragePolicy::NotifyRevoked(origin, change_flags);
82 }
83
84 void NotifyCleared() {
85 SpecialStoragePolicy::NotifyCleared();
86 }
87
88 protected:
89 ~MockSpecialStoragePolicy() override;
90
91 private:
92 std::set<GURL> protected_;
93 std::set<GURL> unlimited_;
94 std::set<GURL> session_only_;
95 std::set<GURL> can_query_disk_size_;
96 std::set<GURL> isolated_;
97 std::set<GURL> durable_;
98 std::set<std::string> file_handlers_;
99
100 bool all_unlimited_;
101 };
102 } // namespace content
103
104 #endif // CONTENT_PUBLIC_TEST_MOCK_SPECIAL_STORAGE_POLICY_H_
OLDNEW
« no previous file with comments | « content/public/test/mock_blob_url_request_context.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