OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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 "chrome/browser/chromeos/drive/resource_metadata_storage.h" | 5 #include "chrome/browser/chromeos/drive/resource_metadata_storage.h" |
6 | 6 |
7 #include <algorithm> | 7 #include <algorithm> |
8 | 8 |
9 #include "base/file_util.h" | 9 #include "base/file_util.h" |
10 #include "base/files/scoped_temp_dir.h" | 10 #include "base/files/scoped_temp_dir.h" |
11 #include "chrome/browser/chromeos/drive/drive.pb.h" | 11 #include "chrome/browser/chromeos/drive/drive.pb.h" |
12 #include "chrome/browser/chromeos/drive/test_util.h" | 12 #include "chrome/browser/chromeos/drive/test_util.h" |
13 #include "content/public/test/test_browser_thread_bundle.h" | 13 #include "content/public/test/test_browser_thread_bundle.h" |
14 #include "testing/gtest/include/gtest/gtest.h" | 14 #include "testing/gtest/include/gtest/gtest.h" |
15 #include "third_party/leveldatabase/src/include/leveldb/db.h" | 15 #include "third_party/leveldatabase/src/include/leveldb/db.h" |
16 | 16 |
17 namespace drive { | 17 namespace drive { |
18 namespace internal { | 18 namespace internal { |
19 | 19 |
20 class ResourceMetadataStorageTest : public testing::Test { | 20 class ResourceMetadataStorageTest : public testing::Test { |
21 protected: | 21 protected: |
22 virtual void SetUp() OVERRIDE { | 22 virtual void SetUp() OVERRIDE { |
23 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); | 23 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); |
24 | 24 |
25 storage_.reset(new ResourceMetadataStorage( | 25 storage_.reset(new ResourceMetadataStorage( |
26 temp_dir_.path(), base::MessageLoopProxy::current())); | 26 temp_dir_.path(), base::MessageLoopProxy::current().get())); |
27 ASSERT_TRUE(storage_->Initialize()); | 27 ASSERT_TRUE(storage_->Initialize()); |
28 } | 28 } |
29 | 29 |
30 // Overwrites |storage_|'s version. | 30 // Overwrites |storage_|'s version. |
31 void SetDBVersion(int version) { | 31 void SetDBVersion(int version) { |
32 ResourceMetadataHeader header; | 32 ResourceMetadataHeader header; |
33 ASSERT_TRUE(storage_->GetHeader(&header)); | 33 ASSERT_TRUE(storage_->GetHeader(&header)); |
34 header.set_version(version); | 34 header.set_version(version); |
35 EXPECT_TRUE(storage_->PutHeader(header)); | 35 EXPECT_TRUE(storage_->PutHeader(header)); |
36 } | 36 } |
(...skipping 268 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
305 entry2.set_resource_id(child_id1); | 305 entry2.set_resource_id(child_id1); |
306 entry2.set_parent_resource_id(parent_id1); | 306 entry2.set_parent_resource_id(parent_id1); |
307 entry2.set_base_name(child_name1); | 307 entry2.set_base_name(child_name1); |
308 | 308 |
309 // Put some data. | 309 // Put some data. |
310 EXPECT_TRUE(storage_->PutEntry(entry1)); | 310 EXPECT_TRUE(storage_->PutEntry(entry1)); |
311 EXPECT_TRUE(storage_->PutEntry(entry2)); | 311 EXPECT_TRUE(storage_->PutEntry(entry2)); |
312 | 312 |
313 // Close DB and reopen. | 313 // Close DB and reopen. |
314 storage_.reset(new ResourceMetadataStorage( | 314 storage_.reset(new ResourceMetadataStorage( |
315 temp_dir_.path(), base::MessageLoopProxy::current())); | 315 temp_dir_.path(), base::MessageLoopProxy::current().get())); |
316 ASSERT_TRUE(storage_->Initialize()); | 316 ASSERT_TRUE(storage_->Initialize()); |
317 | 317 |
318 // Can read data. | 318 // Can read data. |
319 ResourceEntry result; | 319 ResourceEntry result; |
320 ASSERT_TRUE(storage_->GetEntry(parent_id1, &result)); | 320 ASSERT_TRUE(storage_->GetEntry(parent_id1, &result)); |
321 EXPECT_EQ(parent_id1, result.resource_id()); | 321 EXPECT_EQ(parent_id1, result.resource_id()); |
322 | 322 |
323 ASSERT_TRUE(storage_->GetEntry(child_id1, &result)); | 323 ASSERT_TRUE(storage_->GetEntry(child_id1, &result)); |
324 EXPECT_EQ(child_id1, result.resource_id()); | 324 EXPECT_EQ(child_id1, result.resource_id()); |
325 EXPECT_EQ(parent_id1, result.parent_resource_id()); | 325 EXPECT_EQ(parent_id1, result.parent_resource_id()); |
(...skipping 11 matching lines...) Expand all Loading... |
337 | 337 |
338 // Put some data. | 338 // Put some data. |
339 ResourceEntry result; | 339 ResourceEntry result; |
340 EXPECT_TRUE(storage_->SetLargestChangestamp(kLargestChangestamp)); | 340 EXPECT_TRUE(storage_->SetLargestChangestamp(kLargestChangestamp)); |
341 EXPECT_TRUE(storage_->PutEntry(entry)); | 341 EXPECT_TRUE(storage_->PutEntry(entry)); |
342 EXPECT_TRUE(storage_->GetEntry(key1, &result)); | 342 EXPECT_TRUE(storage_->GetEntry(key1, &result)); |
343 | 343 |
344 // Set incompatible version and reopen DB. | 344 // Set incompatible version and reopen DB. |
345 SetDBVersion(ResourceMetadataStorage::kDBVersion - 1); | 345 SetDBVersion(ResourceMetadataStorage::kDBVersion - 1); |
346 storage_.reset(new ResourceMetadataStorage( | 346 storage_.reset(new ResourceMetadataStorage( |
347 temp_dir_.path(), base::MessageLoopProxy::current())); | 347 temp_dir_.path(), base::MessageLoopProxy::current().get())); |
348 ASSERT_TRUE(storage_->Initialize()); | 348 ASSERT_TRUE(storage_->Initialize()); |
349 | 349 |
350 // Data is erased because of the incompatible version. | 350 // Data is erased because of the incompatible version. |
351 EXPECT_EQ(0, storage_->GetLargestChangestamp()); | 351 EXPECT_EQ(0, storage_->GetLargestChangestamp()); |
352 EXPECT_FALSE(storage_->GetEntry(key1, &result)); | 352 EXPECT_FALSE(storage_->GetEntry(key1, &result)); |
353 } | 353 } |
354 | 354 |
355 TEST_F(ResourceMetadataStorageTest, WrongPath) { | 355 TEST_F(ResourceMetadataStorageTest, WrongPath) { |
356 // Create a file. | 356 // Create a file. |
357 base::FilePath path; | 357 base::FilePath path; |
358 ASSERT_TRUE(file_util::CreateTemporaryFileInDir(temp_dir_.path(), &path)); | 358 ASSERT_TRUE(file_util::CreateTemporaryFileInDir(temp_dir_.path(), &path)); |
359 | 359 |
360 storage_.reset(new ResourceMetadataStorage( | 360 storage_.reset(new ResourceMetadataStorage( |
361 path, base::MessageLoopProxy::current())); | 361 path, base::MessageLoopProxy::current().get())); |
362 // Cannot initialize DB beacause the path does not point a directory. | 362 // Cannot initialize DB beacause the path does not point a directory. |
363 ASSERT_FALSE(storage_->Initialize()); | 363 ASSERT_FALSE(storage_->Initialize()); |
364 } | 364 } |
365 | 365 |
366 TEST_F(ResourceMetadataStorageTest, CheckValidity) { | 366 TEST_F(ResourceMetadataStorageTest, CheckValidity) { |
367 const std::string key1 = "foo"; | 367 const std::string key1 = "foo"; |
368 const std::string name1 = "hoge"; | 368 const std::string name1 = "hoge"; |
369 const std::string key2 = "bar"; | 369 const std::string key2 = "bar"; |
370 const std::string name2 = "fuga"; | 370 const std::string name2 = "fuga"; |
371 const std::string key3 = "boo"; | 371 const std::string key3 = "boo"; |
(...skipping 63 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
435 EXPECT_TRUE(storage_->RemoveEntry(key3)); | 435 EXPECT_TRUE(storage_->RemoveEntry(key3)); |
436 EXPECT_TRUE(CheckValidity()); | 436 EXPECT_TRUE(CheckValidity()); |
437 | 437 |
438 // Remove key1. | 438 // Remove key1. |
439 EXPECT_TRUE(storage_->RemoveEntry(key1)); | 439 EXPECT_TRUE(storage_->RemoveEntry(key1)); |
440 EXPECT_TRUE(CheckValidity()); | 440 EXPECT_TRUE(CheckValidity()); |
441 } | 441 } |
442 | 442 |
443 } // namespace internal | 443 } // namespace internal |
444 } // namespace drive | 444 } // namespace drive |
OLD | NEW |