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

Side by Side Diff: chrome/browser/engagement/site_engagement_eviction_policy_unittest.cc

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
(Empty)
1 // Copyright 2015 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 #include "chrome/browser/engagement/site_engagement_eviction_policy.h"
6
7 #include <stdint.h>
8
9 #include <memory>
10
11 #include "base/files/scoped_temp_dir.h"
12 #include "base/macros.h"
13 #include "base/memory/weak_ptr.h"
14 #include "base/run_loop.h"
15 #include "base/threading/thread_task_runner_handle.h"
16 #include "chrome/browser/engagement/site_engagement_service.h"
17 #include "content/public/test/mock_special_storage_policy.h"
18 #include "content/public/test/mock_storage_client.h"
19 #include "content/public/test/test_browser_thread_bundle.h"
20 #include "storage/browser/quota/quota_manager.h"
21 #include "storage/browser/quota/quota_manager_proxy.h"
22 #include "testing/gtest/include/gtest/gtest.h"
23
24 namespace {
25
26 const int64_t kGlobalQuota = 25 * 1024;
27
28 } // namespace
29
30 class TestSiteEngagementScoreProvider : public SiteEngagementScoreProvider {
31 public:
32 TestSiteEngagementScoreProvider() {}
33
34 virtual ~TestSiteEngagementScoreProvider() {}
35
36 double GetScore(const GURL& url) const override {
37 const auto& it = engagement_score_map_.find(url);
38 if (it != engagement_score_map_.end())
39 return it->second;
40 return 0.0;
41 }
42
43 double GetTotalEngagementPoints() const override {
44 double total = 0;
45 for (const auto& site : engagement_score_map_)
46 total += site.second;
47 return total;
48 }
49
50 void SetScore(const GURL& origin, double score) {
51 engagement_score_map_[origin] = score;
52 }
53
54 private:
55 std::map<GURL, double> engagement_score_map_;
56
57 DISALLOW_COPY_AND_ASSIGN(TestSiteEngagementScoreProvider);
58 };
59
60 class SiteEngagementEvictionPolicyTest : public testing::Test {
61 public:
62 SiteEngagementEvictionPolicyTest()
63 : score_provider_(new TestSiteEngagementScoreProvider()),
64 storage_policy_(new content::MockSpecialStoragePolicy()) {}
65
66 ~SiteEngagementEvictionPolicyTest() override {}
67
68 GURL CalculateEvictionOriginWithExceptions(
69 const std::map<GURL, int64_t>& usage,
70 const std::set<GURL>& exceptions) {
71 return SiteEngagementEvictionPolicy::CalculateEvictionOriginForTests(
72 storage_policy_, score_provider_.get(), exceptions, usage,
73 kGlobalQuota);
74 }
75
76 GURL CalculateEvictionOrigin(const std::map<GURL, int64_t>& usage) {
77 return CalculateEvictionOriginWithExceptions(usage, std::set<GURL>());
78 }
79
80 TestSiteEngagementScoreProvider* score_provider() {
81 return score_provider_.get();
82 }
83
84 content::MockSpecialStoragePolicy* storage_policy() {
85 return storage_policy_.get();
86 }
87
88 private:
89 std::unique_ptr<TestSiteEngagementScoreProvider> score_provider_;
90 scoped_refptr<content::MockSpecialStoragePolicy> storage_policy_;
91
92 DISALLOW_COPY_AND_ASSIGN(SiteEngagementEvictionPolicyTest);
93 };
94
95 TEST_F(SiteEngagementEvictionPolicyTest, GetEvictionOrigin) {
96 GURL url1("http://www.google.com");
97 GURL url2("http://www.example.com");
98 GURL url3("http://www.spam.me");
99
100 std::map<GURL, int64_t> usage;
101 usage[url1] = 10 * 1024;
102 usage[url2] = 10 * 1024;
103 usage[url3] = 10 * 1024;
104
105 score_provider()->SetScore(url1, 50);
106 score_provider()->SetScore(url2, 25);
107
108 // When 3 sites have equal usage, evict the site with the least engagement.
109 EXPECT_EQ(url3, CalculateEvictionOrigin(usage));
110
111 usage[url2] = usage[url3] + 10;
112
113 // Now |url2| has the most usage but |url3| has the least engagement score so
114 // one of them should be evicted. In this case the heuristic chooses |url3|.
115 EXPECT_EQ(url3, CalculateEvictionOrigin(usage));
116
117 // But exceeding allocated usage too much will still result in being evicted
118 // even though the engagement with |url2| is higher.
119 usage[url2] = 15 * 1024;
120 EXPECT_EQ(url2, CalculateEvictionOrigin(usage));
121
122 // When all origins have the same engagement, the origin with the highest
123 // usage is evicted.
124 score_provider()->SetScore(url1, 50);
125 score_provider()->SetScore(url2, 50);
126 score_provider()->SetScore(url3, 50);
127
128 usage[url2] = 10 * 1024;
129 usage[url3] = 20 * 1024;
130 EXPECT_EQ(url3, CalculateEvictionOrigin(usage));
131 }
132
133 // Test that durable and unlimited storage origins are exempt from eviction.
134 TEST_F(SiteEngagementEvictionPolicyTest, SpecialStoragePolicy) {
135 GURL url1("http://www.google.com");
136 GURL url2("http://www.example.com");
137
138 std::map<GURL, int64_t> usage;
139 usage[url1] = 10 * 1024;
140 usage[url2] = 10 * 1024;
141
142 score_provider()->SetScore(url1, 50);
143 score_provider()->SetScore(url2, 25);
144
145 EXPECT_EQ(url2, CalculateEvictionOrigin(usage));
146
147 // Durable storage doesn't get evicted.
148 storage_policy()->AddDurable(url2);
149 EXPECT_EQ(url1, CalculateEvictionOrigin(usage));
150
151 // Unlimited storage doesn't get evicted.
152 storage_policy()->AddUnlimited(url1);
153 EXPECT_EQ(GURL(), CalculateEvictionOrigin(usage));
154 }
155
156 TEST_F(SiteEngagementEvictionPolicyTest, Exceptions) {
157 GURL url1("http://www.google.com");
158 GURL url2("http://www.example.com");
159
160 std::map<GURL, int64_t> usage;
161 usage[url1] = 10 * 1024;
162 usage[url2] = 10 * 1024;
163
164 score_provider()->SetScore(url1, 50);
165 score_provider()->SetScore(url2, 25);
166
167 EXPECT_EQ(url2, CalculateEvictionOrigin(usage));
168
169 // The policy should respect exceptions.
170 std::set<GURL> exceptions;
171 exceptions.insert(url2);
172 EXPECT_EQ(url1, CalculateEvictionOriginWithExceptions(usage, exceptions));
173 }
OLDNEW
« no previous file with comments | « chrome/browser/engagement/site_engagement_eviction_policy.cc ('k') | chrome/browser/engagement/site_engagement_service.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698