OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 "base/bind.h" | 5 #include "base/bind.h" |
6 #include "base/file_util.h" | 6 #include "base/file_util.h" |
7 #include "base/files/scoped_temp_dir.h" | 7 #include "base/files/scoped_temp_dir.h" |
8 #include "base/message_loop.h" | 8 #include "base/message_loop.h" |
9 #include "base/message_loop_proxy.h" | 9 #include "base/message_loop_proxy.h" |
10 #include "base/threading/sequenced_worker_pool.h" | 10 #include "base/threading/sequenced_worker_pool.h" |
11 #include "base/time.h" | 11 #include "base/time.h" |
12 #include "base/utf_string_conversions.h" | 12 #include "base/utf_string_conversions.h" |
13 #include "testing/gtest/include/gtest/gtest.h" | 13 #include "testing/gtest/include/gtest/gtest.h" |
14 #include "webkit/dom_storage/dom_storage_area.h" | 14 #include "webkit/browser/dom_storage/dom_storage_area.h" |
15 #include "webkit/dom_storage/dom_storage_context.h" | 15 #include "webkit/browser/dom_storage/dom_storage_context.h" |
16 #include "webkit/dom_storage/dom_storage_namespace.h" | 16 #include "webkit/browser/dom_storage/dom_storage_namespace.h" |
17 #include "webkit/dom_storage/dom_storage_task_runner.h" | 17 #include "webkit/browser/dom_storage/dom_storage_task_runner.h" |
18 #include "webkit/dom_storage/dom_storage_types.h" | 18 #include "webkit/browser/quota/mock_special_storage_policy.h" |
19 #include "webkit/quota/mock_special_storage_policy.h" | 19 #include "webkit/common/dom_storage/dom_storage_types.h" |
20 | 20 |
21 namespace dom_storage { | 21 namespace dom_storage { |
22 | 22 |
23 class DomStorageContextTest : public testing::Test { | 23 class DomStorageContextTest : public testing::Test { |
24 public: | 24 public: |
25 DomStorageContextTest() | 25 DomStorageContextTest() |
26 : kOrigin(GURL("http://dom_storage/")), | 26 : kOrigin(GURL("http://dom_storage/")), |
27 kKey(ASCIIToUTF16("key")), | 27 kKey(ASCIIToUTF16("key")), |
28 kValue(ASCIIToUTF16("value")), | 28 kValue(ASCIIToUTF16("value")), |
29 kDontIncludeFileInfo(false), | 29 kDontIncludeFileInfo(false), |
30 kDoIncludeFileInfo(true) { | 30 kDoIncludeFileInfo(true) { |
31 } | 31 } |
32 | 32 |
33 const GURL kOrigin; | 33 const GURL kOrigin; |
34 const base::string16 kKey; | 34 const base::string16 kKey; |
35 const base::string16 kValue; | 35 const base::string16 kValue; |
36 const bool kDontIncludeFileInfo; | 36 const bool kDontIncludeFileInfo; |
37 const bool kDoIncludeFileInfo; | 37 const bool kDoIncludeFileInfo; |
38 | 38 |
39 virtual void SetUp() { | 39 virtual void SetUp() { |
40 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); | 40 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); |
41 storage_policy_ = new quota::MockSpecialStoragePolicy; | 41 storage_policy_ = new quota::MockSpecialStoragePolicy; |
42 task_runner_ = new MockDomStorageTaskRunner( | 42 task_runner_ = new MockDomStorageTaskRunner( |
43 base::MessageLoopProxy::current()); | 43 base::MessageLoopProxy::current()); |
44 context_ = new DomStorageContext(temp_dir_.path(), | 44 context_ = new DomStorageContext(temp_dir_.path(), |
45 base::FilePath(), | 45 base::FilePath(), |
46 storage_policy_, | 46 storage_policy_.get(), |
47 task_runner_); | 47 task_runner_.get()); |
48 } | 48 } |
49 | 49 |
50 virtual void TearDown() { | 50 virtual void TearDown() { |
51 base::MessageLoop::current()->RunUntilIdle(); | 51 base::MessageLoop::current()->RunUntilIdle(); |
52 } | 52 } |
53 | 53 |
54 void VerifySingleOriginRemains(const GURL& origin) { | 54 void VerifySingleOriginRemains(const GURL& origin) { |
55 // Use a new instance to examine the contexts of temp_dir_. | 55 // Use a new instance to examine the contexts of temp_dir_. |
56 scoped_refptr<DomStorageContext> context = | 56 scoped_refptr<DomStorageContext> context = |
57 new DomStorageContext(temp_dir_.path(), base::FilePath(), NULL, NULL); | 57 new DomStorageContext(temp_dir_.path(), base::FilePath(), NULL, NULL); |
(...skipping 127 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
185 cloned_dom_namespace->persistent_namespace_id()); | 185 cloned_dom_namespace->persistent_namespace_id()); |
186 // Verify that the areas inherit the persistent ID. | 186 // Verify that the areas inherit the persistent ID. |
187 DomStorageArea* cloned_area = cloned_dom_namespace->OpenStorageArea(kOrigin); | 187 DomStorageArea* cloned_area = cloned_dom_namespace->OpenStorageArea(kOrigin); |
188 EXPECT_EQ(kClonedPersistentId, cloned_area->persistent_namespace_id_); | 188 EXPECT_EQ(kClonedPersistentId, cloned_area->persistent_namespace_id_); |
189 } | 189 } |
190 | 190 |
191 TEST_F(DomStorageContextTest, DeleteSessionStorage) { | 191 TEST_F(DomStorageContextTest, DeleteSessionStorage) { |
192 // Create a DomStorageContext which will save sessionStorage on disk. | 192 // Create a DomStorageContext which will save sessionStorage on disk. |
193 context_ = new DomStorageContext(temp_dir_.path(), | 193 context_ = new DomStorageContext(temp_dir_.path(), |
194 temp_dir_.path(), | 194 temp_dir_.path(), |
195 storage_policy_, | 195 storage_policy_.get(), |
196 task_runner_); | 196 task_runner_.get()); |
197 context_->SetSaveSessionStorageOnDisk(); | 197 context_->SetSaveSessionStorageOnDisk(); |
198 ASSERT_EQ(temp_dir_.path(), context_->sessionstorage_directory()); | 198 ASSERT_EQ(temp_dir_.path(), context_->sessionstorage_directory()); |
199 | 199 |
200 // Write data. | 200 // Write data. |
201 const int kSessionStorageNamespaceId = 1; | 201 const int kSessionStorageNamespaceId = 1; |
202 const std::string kPersistentId = "persistent"; | 202 const std::string kPersistentId = "persistent"; |
203 context_->CreateSessionNamespace(kSessionStorageNamespaceId, | 203 context_->CreateSessionNamespace(kSessionStorageNamespaceId, |
204 kPersistentId); | 204 kPersistentId); |
205 DomStorageNamespace* dom_namespace = | 205 DomStorageNamespace* dom_namespace = |
206 context_->GetStorageNamespace(kSessionStorageNamespaceId); | 206 context_->GetStorageNamespace(kSessionStorageNamespaceId); |
207 DomStorageArea* area = dom_namespace->OpenStorageArea(kOrigin); | 207 DomStorageArea* area = dom_namespace->OpenStorageArea(kOrigin); |
208 const base::string16 kKey(ASCIIToUTF16("foo")); | 208 const base::string16 kKey(ASCIIToUTF16("foo")); |
209 const base::string16 kValue(ASCIIToUTF16("bar")); | 209 const base::string16 kValue(ASCIIToUTF16("bar")); |
210 NullableString16 old_nullable_value; | 210 NullableString16 old_nullable_value; |
211 area->SetItem(kKey, kValue, &old_nullable_value); | 211 area->SetItem(kKey, kValue, &old_nullable_value); |
212 dom_namespace->CloseStorageArea(area); | 212 dom_namespace->CloseStorageArea(area); |
213 | 213 |
214 // Destroy and recreate the DomStorageContext. | 214 // Destroy and recreate the DomStorageContext. |
215 context_->Shutdown(); | 215 context_->Shutdown(); |
216 context_ = NULL; | 216 context_ = NULL; |
217 base::MessageLoop::current()->RunUntilIdle(); | 217 base::MessageLoop::current()->RunUntilIdle(); |
218 context_ = new DomStorageContext( | 218 context_ = new DomStorageContext( |
219 temp_dir_.path(), temp_dir_.path(), storage_policy_, task_runner_); | 219 temp_dir_.path(), temp_dir_.path(), |
| 220 storage_policy_.get(), task_runner_.get()); |
220 context_->SetSaveSessionStorageOnDisk(); | 221 context_->SetSaveSessionStorageOnDisk(); |
221 | 222 |
222 // Read the data back. | 223 // Read the data back. |
223 context_->CreateSessionNamespace(kSessionStorageNamespaceId, | 224 context_->CreateSessionNamespace(kSessionStorageNamespaceId, |
224 kPersistentId); | 225 kPersistentId); |
225 dom_namespace = context_->GetStorageNamespace(kSessionStorageNamespaceId); | 226 dom_namespace = context_->GetStorageNamespace(kSessionStorageNamespaceId); |
226 area = dom_namespace->OpenStorageArea(kOrigin); | 227 area = dom_namespace->OpenStorageArea(kOrigin); |
227 NullableString16 read_value; | 228 NullableString16 read_value; |
228 read_value = area->GetItem(kKey); | 229 read_value = area->GetItem(kKey); |
229 EXPECT_EQ(kValue, read_value.string()); | 230 EXPECT_EQ(kValue, read_value.string()); |
230 dom_namespace->CloseStorageArea(area); | 231 dom_namespace->CloseStorageArea(area); |
231 | 232 |
232 SessionStorageUsageInfo info; | 233 SessionStorageUsageInfo info; |
233 info.origin = kOrigin; | 234 info.origin = kOrigin; |
234 info.persistent_namespace_id = kPersistentId; | 235 info.persistent_namespace_id = kPersistentId; |
235 context_->DeleteSessionStorage(info); | 236 context_->DeleteSessionStorage(info); |
236 | 237 |
237 // Destroy and recreate again. | 238 // Destroy and recreate again. |
238 context_->Shutdown(); | 239 context_->Shutdown(); |
239 context_ = NULL; | 240 context_ = NULL; |
240 base::MessageLoop::current()->RunUntilIdle(); | 241 base::MessageLoop::current()->RunUntilIdle(); |
241 context_ = new DomStorageContext( | 242 context_ = new DomStorageContext( |
242 temp_dir_.path(), temp_dir_.path(), storage_policy_, task_runner_); | 243 temp_dir_.path(), temp_dir_.path(), |
| 244 storage_policy_.get(), task_runner_.get()); |
243 context_->SetSaveSessionStorageOnDisk(); | 245 context_->SetSaveSessionStorageOnDisk(); |
244 | 246 |
245 // Now there should be no data. | 247 // Now there should be no data. |
246 context_->CreateSessionNamespace(kSessionStorageNamespaceId, | 248 context_->CreateSessionNamespace(kSessionStorageNamespaceId, |
247 kPersistentId); | 249 kPersistentId); |
248 dom_namespace = context_->GetStorageNamespace(kSessionStorageNamespaceId); | 250 dom_namespace = context_->GetStorageNamespace(kSessionStorageNamespaceId); |
249 area = dom_namespace->OpenStorageArea(kOrigin); | 251 area = dom_namespace->OpenStorageArea(kOrigin); |
250 read_value = area->GetItem(kKey); | 252 read_value = area->GetItem(kKey); |
251 EXPECT_TRUE(read_value.is_null()); | 253 EXPECT_TRUE(read_value.is_null()); |
252 dom_namespace->CloseStorageArea(area); | 254 dom_namespace->CloseStorageArea(area); |
253 context_->Shutdown(); | 255 context_->Shutdown(); |
254 context_ = NULL; | 256 context_ = NULL; |
255 base::MessageLoop::current()->RunUntilIdle(); | 257 base::MessageLoop::current()->RunUntilIdle(); |
256 } | 258 } |
257 | 259 |
258 } // namespace dom_storage | 260 } // namespace dom_storage |
OLD | NEW |