drive: DriveIntegrationService owns ResourceMetadataStorage

chrome/browser/chromeos/drive/resource_metadata_storage_unittest.cc

 // Copyright 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/chromeos/drive/resource_metadata_storage.h"
 
 #include <algorithm>
 
 #include "base/file_util.h"
 #include "base/files/scoped_temp_dir.h"
 #include "chrome/browser/chromeos/drive/drive.pb.h"
+#include "chrome/browser/chromeos/drive/test_util.h"
+#include "content/public/test/test_browser_thread_bundle.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "third_party/leveldatabase/src/include/leveldb/db.h"
 
 namespace drive {
+namespace internal {
 
 class ResourceMetadataStorageTest : public testing::Test {
  protected:
   virtual void SetUp() OVERRIDE {
     ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
 
-    storage_.reset(new ResourceMetadataStorage(temp_dir_.path()));
+    storage_.reset(new ResourceMetadataStorage(
+        temp_dir_.path(), base::MessageLoopProxy::current()));
     ASSERT_TRUE(storage_->Initialize());
   }
 
-  virtual void TearDown() OVERRIDE {
-  }
-
   // Overwrites |storage_|'s version.
   void SetDBVersion(int version) {
     ResourceMetadataHeader header;
     ASSERT_TRUE(storage_->GetHeader(&header));
     header.set_version(version);
     EXPECT_TRUE(storage_->PutHeader(header));
   }
 
   bool CheckValidity() {
     return storage_->CheckValidity();
@@ skipping 12 matching lines @@
 
   // Removes a child entry.
   void RemoveChild(const std::string& parent_resource_id,
                    const std::string& child_base_name) {
     storage_->resource_map_->Delete(
         leveldb::WriteOptions(),
         ResourceMetadataStorage::GetChildEntryKey(parent_resource_id,
                                                   child_base_name));
   }
 
+  content::TestBrowserThreadBundle thread_bundle_;
   base::ScopedTempDir temp_dir_;
-  scoped_ptr<ResourceMetadataStorage> storage_;
+  scoped_ptr<ResourceMetadataStorage,
+             test_util::DestroyHelperForTests> storage_;
 };
 
 TEST_F(ResourceMetadataStorageTest, LargestChangestamp) {
   const int64 kLargestChangestamp = 1234567890;
   EXPECT_TRUE(storage_->SetLargestChangestamp(kLargestChangestamp));
   EXPECT_EQ(kLargestChangestamp, storage_->GetLargestChangestamp());
 }
 
 TEST_F(ResourceMetadataStorageTest, PutEntry) {
   const std::string key1 = "abcdefg";
@@ skipping 228 matching lines @@
   ResourceEntry entry2;
   entry2.set_resource_id(child_id1);
   entry2.set_parent_resource_id(parent_id1);
   entry2.set_base_name(child_name1);
 
   // Put some data.
   EXPECT_TRUE(storage_->PutEntry(entry1));
   EXPECT_TRUE(storage_->PutEntry(entry2));
 
   // Close DB and reopen.
-  storage_.reset(new ResourceMetadataStorage(temp_dir_.path()));
+  storage_.reset(new ResourceMetadataStorage(
+      temp_dir_.path(), base::MessageLoopProxy::current()));
   ASSERT_TRUE(storage_->Initialize());
 
   // Can read data.
   ResourceEntry result;
   ASSERT_TRUE(storage_->GetEntry(parent_id1, &result));
   EXPECT_EQ(parent_id1, result.resource_id());
 
   ASSERT_TRUE(storage_->GetEntry(child_id1, &result));
   EXPECT_EQ(child_id1, result.resource_id());
   EXPECT_EQ(parent_id1, result.parent_resource_id());
@@ skipping 10 matching lines @@
   entry.set_resource_id(key1);
 
   // Put some data.
   ResourceEntry result;
   EXPECT_TRUE(storage_->SetLargestChangestamp(kLargestChangestamp));
   EXPECT_TRUE(storage_->PutEntry(entry));
   EXPECT_TRUE(storage_->GetEntry(key1, &result));
 
   // Set incompatible version and reopen DB.
   SetDBVersion(ResourceMetadataStorage::kDBVersion - 1);
-  storage_.reset(new ResourceMetadataStorage(temp_dir_.path()));
+  storage_.reset(new ResourceMetadataStorage(
+      temp_dir_.path(), base::MessageLoopProxy::current()));
   ASSERT_TRUE(storage_->Initialize());
 
   // Data is erased because of the incompatible version.
   EXPECT_EQ(0, storage_->GetLargestChangestamp());
   EXPECT_FALSE(storage_->GetEntry(key1, &result));
 }
 
 TEST_F(ResourceMetadataStorageTest, WrongPath) {
   // Create a file.
   base::FilePath path;
   ASSERT_TRUE(file_util::CreateTemporaryFileInDir(temp_dir_.path(), &path));
 
-  storage_.reset(new ResourceMetadataStorage(path));
+  storage_.reset(new ResourceMetadataStorage(
+      path, base::MessageLoopProxy::current()));
   // Cannot initialize DB beacause the path does not point a directory.
   ASSERT_FALSE(storage_->Initialize());
 }
 
 TEST_F(ResourceMetadataStorageTest, CheckValidity) {
   const std::string key1 = "foo";
   const std::string name1 = "hoge";
   const std::string key2 = "bar";
   const std::string name2 = "fuga";
   const std::string key3 = "boo";
@@ skipping 61 matching lines @@
   RemoveChild(key2, name3);
   EXPECT_FALSE(CheckValidity());
   EXPECT_TRUE(storage_->RemoveEntry(key3));
   EXPECT_TRUE(CheckValidity());
 
   // Remove key1.
   EXPECT_TRUE(storage_->RemoveEntry(key1));
   EXPECT_TRUE(CheckValidity());
 }
 
+}  // namespace internal
 }  // namespace drive