| 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 |