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

Side by Side Diff: storage/browser/database/database_quota_client_unittest.cc

Issue 2776253002: Move some database tests next to the files they cover. (Closed)
Patch Set: Created 3 years, 9 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
« no previous file with comments | « storage/browser/BUILD.gn ('k') | storage/browser/database/database_util_unittest.cc » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
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 #include <stdint.h> 5 #include <stdint.h>
6 6
7 #include <map> 7 #include <map>
8 8
9 #include "base/bind.h" 9 #include "base/bind.h"
10 #include "base/files/file_path.h" 10 #include "base/files/file_path.h"
(...skipping 36 matching lines...) Expand 10 before | Expand all | Expand 10 after
47 storage::GetOriginFromIdentifier(origin_identifier)); 47 storage::GetOriginFromIdentifier(origin_identifier));
48 if (found == mock_origin_infos_.end()) 48 if (found == mock_origin_infos_.end())
49 return false; 49 return false;
50 *info = OriginInfo(found->second); 50 *info = OriginInfo(found->second);
51 return true; 51 return true;
52 } 52 }
53 53
54 bool GetAllOriginIdentifiers( 54 bool GetAllOriginIdentifiers(
55 std::vector<std::string>* origins_identifiers) override { 55 std::vector<std::string>* origins_identifiers) override {
56 std::map<GURL, MockOriginInfo>::const_iterator iter; 56 std::map<GURL, MockOriginInfo>::const_iterator iter;
57 for (iter = mock_origin_infos_.begin(); 57 for (iter = mock_origin_infos_.begin(); iter != mock_origin_infos_.end();
58 iter != mock_origin_infos_.end();
59 ++iter) { 58 ++iter) {
60 origins_identifiers->push_back(iter->second.GetOriginIdentifier()); 59 origins_identifiers->push_back(iter->second.GetOriginIdentifier());
61 } 60 }
62 return true; 61 return true;
63 } 62 }
64 63
65 bool GetAllOriginsInfo(std::vector<OriginInfo>* origins_info) override { 64 bool GetAllOriginsInfo(std::vector<OriginInfo>* origins_info) override {
66 std::map<GURL, MockOriginInfo>::const_iterator iter; 65 std::map<GURL, MockOriginInfo>::const_iterator iter;
67 for (iter = mock_origin_infos_.begin(); 66 for (iter = mock_origin_infos_.begin(); iter != mock_origin_infos_.end();
68 iter != mock_origin_infos_.end();
69 ++iter) { 67 ++iter) {
70 origins_info->push_back(OriginInfo(iter->second)); 68 origins_info->push_back(OriginInfo(iter->second));
71 } 69 }
72 return true; 70 return true;
73 } 71 }
74 72
75 int DeleteDataForOrigin(const std::string& origin_identifier, 73 int DeleteDataForOrigin(const std::string& origin_identifier,
76 const net::CompletionCallback& callback) override { 74 const net::CompletionCallback& callback) override {
77 ++delete_called_count_; 75 ++delete_called_count_;
78 if (async_delete()) { 76 if (async_delete()) {
79 base::ThreadTaskRunnerHandle::Get()->PostTask( 77 base::ThreadTaskRunnerHandle::Get()->PostTask(
80 FROM_HERE, base::Bind(&MockDatabaseTracker::AsyncDeleteDataForOrigin, 78 FROM_HERE, base::Bind(&MockDatabaseTracker::AsyncDeleteDataForOrigin,
81 this, callback)); 79 this, callback));
82 return net::ERR_IO_PENDING; 80 return net::ERR_IO_PENDING;
83 } 81 }
84 return net::OK; 82 return net::OK;
85 } 83 }
86 84
87 void AsyncDeleteDataForOrigin(const net::CompletionCallback& callback) { 85 void AsyncDeleteDataForOrigin(const net::CompletionCallback& callback) {
88 callback.Run(net::OK); 86 callback.Run(net::OK);
89 } 87 }
90 88
91 void AddMockDatabase(const GURL& origin, const char* name, int size) { 89 void AddMockDatabase(const GURL& origin, const char* name, int size) {
92 MockOriginInfo& info = mock_origin_infos_[origin]; 90 MockOriginInfo& info = mock_origin_infos_[origin];
93 info.set_origin(storage::GetIdentifierFromOrigin(origin)); 91 info.set_origin(storage::GetIdentifierFromOrigin(origin));
94 info.AddMockDatabase(base::ASCIIToUTF16(name), size); 92 info.AddMockDatabase(base::ASCIIToUTF16(name), size);
95 } 93 }
96 94
97 int delete_called_count() { return delete_called_count_; } 95 int delete_called_count() { return delete_called_count_; }
98 bool async_delete() { return async_delete_; } 96 bool async_delete() { return async_delete_; }
99 void set_async_delete(bool async) { async_delete_ = async; } 97 void set_async_delete(bool async) { async_delete_ = async; }
100 98
101 protected: 99 protected:
(...skipping 11 matching lines...) Expand all
113 database_info_[name].first = size; 111 database_info_[name].first = size;
114 total_size_ += size; 112 total_size_ += size;
115 } 113 }
116 }; 114 };
117 115
118 int delete_called_count_; 116 int delete_called_count_;
119 bool async_delete_; 117 bool async_delete_;
120 std::map<GURL, MockOriginInfo> mock_origin_infos_; 118 std::map<GURL, MockOriginInfo> mock_origin_infos_;
121 }; 119 };
122 120
123
124 // Base class for our test fixtures. 121 // Base class for our test fixtures.
125 class DatabaseQuotaClientTest : public testing::Test { 122 class DatabaseQuotaClientTest : public testing::Test {
126 public: 123 public:
127 const GURL kOriginA; 124 const GURL kOriginA;
128 const GURL kOriginB; 125 const GURL kOriginB;
129 const GURL kOriginOther; 126 const GURL kOriginOther;
130 127
131 DatabaseQuotaClientTest() 128 DatabaseQuotaClientTest()
132 : kOriginA("http://host"), 129 : kOriginA("http://host"),
133 kOriginB("http://host:8000"), 130 kOriginB("http://host:8000"),
134 kOriginOther("http://other"), 131 kOriginOther("http://other"),
135 usage_(0), 132 usage_(0),
136 mock_tracker_(new MockDatabaseTracker), 133 mock_tracker_(new MockDatabaseTracker),
137 weak_factory_(this) { 134 weak_factory_(this) {}
138 }
139 135
140 int64_t GetOriginUsage(storage::QuotaClient* client, 136 int64_t GetOriginUsage(storage::QuotaClient* client,
141 const GURL& origin, 137 const GURL& origin,
142 storage::StorageType type) { 138 storage::StorageType type) {
143 usage_ = 0; 139 usage_ = 0;
144 client->GetOriginUsage( 140 client->GetOriginUsage(
145 origin, type, 141 origin, type,
146 base::Bind(&DatabaseQuotaClientTest::OnGetOriginUsageComplete, 142 base::Bind(&DatabaseQuotaClientTest::OnGetOriginUsageComplete,
147 weak_factory_.GetWeakPtr())); 143 weak_factory_.GetWeakPtr()));
148 base::RunLoop().RunUntilIdle(); 144 base::RunLoop().RunUntilIdle();
149 return usage_; 145 return usage_;
150 } 146 }
151 147
152 const std::set<GURL>& GetOriginsForType(storage::QuotaClient* client, 148 const std::set<GURL>& GetOriginsForType(storage::QuotaClient* client,
153 storage::StorageType type) { 149 storage::StorageType type) {
154 origins_.clear(); 150 origins_.clear();
155 client->GetOriginsForType( 151 client->GetOriginsForType(
156 type, 152 type, base::Bind(&DatabaseQuotaClientTest::OnGetOriginsComplete,
157 base::Bind(&DatabaseQuotaClientTest::OnGetOriginsComplete, 153 weak_factory_.GetWeakPtr()));
158 weak_factory_.GetWeakPtr()));
159 base::RunLoop().RunUntilIdle(); 154 base::RunLoop().RunUntilIdle();
160 return origins_; 155 return origins_;
161 } 156 }
162 157
163 const std::set<GURL>& GetOriginsForHost(storage::QuotaClient* client, 158 const std::set<GURL>& GetOriginsForHost(storage::QuotaClient* client,
164 storage::StorageType type, 159 storage::StorageType type,
165 const std::string& host) { 160 const std::string& host) {
166 origins_.clear(); 161 origins_.clear();
167 client->GetOriginsForHost( 162 client->GetOriginsForHost(
168 type, host, 163 type, host,
(...skipping 10 matching lines...) Expand all
179 client->DeleteOriginData( 174 client->DeleteOriginData(
180 origin, type, 175 origin, type,
181 base::Bind(&DatabaseQuotaClientTest::OnDeleteOriginDataComplete, 176 base::Bind(&DatabaseQuotaClientTest::OnDeleteOriginDataComplete,
182 weak_factory_.GetWeakPtr())); 177 weak_factory_.GetWeakPtr()));
183 base::RunLoop().RunUntilIdle(); 178 base::RunLoop().RunUntilIdle();
184 return delete_status_ == storage::kQuotaStatusOk; 179 return delete_status_ == storage::kQuotaStatusOk;
185 } 180 }
186 181
187 MockDatabaseTracker* mock_tracker() { return mock_tracker_.get(); } 182 MockDatabaseTracker* mock_tracker() { return mock_tracker_.get(); }
188 183
189
190 private: 184 private:
191 void OnGetOriginUsageComplete(int64_t usage) { usage_ = usage; } 185 void OnGetOriginUsageComplete(int64_t usage) { usage_ = usage; }
192 186
193 void OnGetOriginsComplete(const std::set<GURL>& origins) { 187 void OnGetOriginsComplete(const std::set<GURL>& origins) {
194 origins_ = origins; 188 origins_ = origins;
195 } 189 }
196 190
197 void OnDeleteOriginDataComplete(storage::QuotaStatusCode status) { 191 void OnDeleteOriginDataComplete(storage::QuotaStatusCode status) {
198 delete_status_ = status; 192 delete_status_ = status;
199 } 193 }
200 194
201 base::MessageLoop message_loop_; 195 base::MessageLoop message_loop_;
202 int64_t usage_; 196 int64_t usage_;
203 std::set<GURL> origins_; 197 std::set<GURL> origins_;
204 storage::QuotaStatusCode delete_status_; 198 storage::QuotaStatusCode delete_status_;
205 scoped_refptr<MockDatabaseTracker> mock_tracker_; 199 scoped_refptr<MockDatabaseTracker> mock_tracker_;
206 base::WeakPtrFactory<DatabaseQuotaClientTest> weak_factory_; 200 base::WeakPtrFactory<DatabaseQuotaClientTest> weak_factory_;
207 }; 201 };
208 202
209
210 TEST_F(DatabaseQuotaClientTest, GetOriginUsage) { 203 TEST_F(DatabaseQuotaClientTest, GetOriginUsage) {
211 DatabaseQuotaClient client(base::ThreadTaskRunnerHandle::Get().get(), 204 DatabaseQuotaClient client(base::ThreadTaskRunnerHandle::Get().get(),
212 mock_tracker()); 205 mock_tracker());
213 206
214 EXPECT_EQ(0, GetOriginUsage(&client, kOriginA, kTemp)); 207 EXPECT_EQ(0, GetOriginUsage(&client, kOriginA, kTemp));
215 EXPECT_EQ(0, GetOriginUsage(&client, kOriginA, kPerm)); 208 EXPECT_EQ(0, GetOriginUsage(&client, kOriginA, kPerm));
216 209
217 mock_tracker()->AddMockDatabase(kOriginA, "fooDB", 1000); 210 mock_tracker()->AddMockDatabase(kOriginA, "fooDB", 1000);
218 EXPECT_EQ(1000, GetOriginUsage(&client, kOriginA, kTemp)); 211 EXPECT_EQ(1000, GetOriginUsage(&client, kOriginA, kTemp));
219 EXPECT_EQ(0, GetOriginUsage(&client, kOriginA, kPerm)); 212 EXPECT_EQ(0, GetOriginUsage(&client, kOriginA, kPerm));
(...skipping 54 matching lines...) Expand 10 before | Expand all | Expand 10 after
274 mock_tracker()->set_async_delete(false); 267 mock_tracker()->set_async_delete(false);
275 EXPECT_TRUE(DeleteOriginData(&client, kTemp, kOriginA)); 268 EXPECT_TRUE(DeleteOriginData(&client, kTemp, kOriginA));
276 EXPECT_EQ(1, mock_tracker()->delete_called_count()); 269 EXPECT_EQ(1, mock_tracker()->delete_called_count());
277 270
278 mock_tracker()->set_async_delete(true); 271 mock_tracker()->set_async_delete(true);
279 EXPECT_TRUE(DeleteOriginData(&client, kTemp, kOriginA)); 272 EXPECT_TRUE(DeleteOriginData(&client, kTemp, kOriginA));
280 EXPECT_EQ(2, mock_tracker()->delete_called_count()); 273 EXPECT_EQ(2, mock_tracker()->delete_called_count());
281 } 274 }
282 275
283 } // namespace content 276 } // namespace content
OLDNEW
« no previous file with comments | « storage/browser/BUILD.gn ('k') | storage/browser/database/database_util_unittest.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698