[SyncFS] Add MetadataDatabase::PopulateInitialData

chrome/browser/sync_file_system/drive_backend/metadata_database_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/sync_file_system/drive_backend/metadata_database.h"
 
 #include "base/bind.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/message_loop/message_loop.h"
 #include "base/message_loop/message_loop_proxy.h"
@@ skipping 373 matching lines @@
   }
 
   TrackedFile CreateTrackedFolder(const TrackedFile& parent,
                                   const std::string& title) {
     TrackedFile folder;
     folder.metadata = CreateFolderMetadata(parent.metadata, title);
     folder.tracker = CreateTracker(parent.tracker, folder.metadata);
     return folder;
   }
 
-  scoped_ptr<google_apis::ChangeResource> CreateChangeResourceFromMetadata(
+  scoped_ptr<google_apis::FileResource> CreateFileResourceFromMetadata(
       const FileMetadata& file) {
-    scoped_ptr<google_apis::ChangeResource> change(
-        new google_apis::ChangeResource);
-    change->set_change_id(file.details().change_id());
-    change->set_file_id(file.file_id());
-    change->set_deleted(file.details().deleted());
-    if (change->is_deleted())
-      return change.Pass();
-
     scoped_ptr<google_apis::FileResource> file_resource(
         new google_apis::FileResource);
     ScopedVector<google_apis::ParentReference> parents;
     for (int i = 0; i < file.details().parent_folder_ids_size(); ++i) {
       scoped_ptr<google_apis::ParentReference> parent(
           new google_apis::ParentReference);
       parent->set_file_id(file.details().parent_folder_ids(i));
       parents.push_back(parent.release());
     }
 
     file_resource->set_file_id(file.file_id());
     file_resource->set_parents(&parents);
     file_resource->set_title(file.details().title());
     if (file.details().file_kind() == FILE_KIND_FOLDER)
       file_resource->set_mime_type("application/vnd.google-apps.folder");
     else if (file.details().file_kind() == FILE_KIND_FILE)
       file_resource->set_mime_type("text/plain");
     else
       file_resource->set_mime_type("application/vnd.google-apps.document");
     file_resource->set_md5_checksum(file.details().md5());
     file_resource->set_etag(file.details().etag());
     file_resource->set_created_date(base::Time::FromInternalValue(
         file.details().creation_time()));
     file_resource->set_modified_date(base::Time::FromInternalValue(
         file.details().modification_time()));
 
-    change->set_file(file_resource.Pass());
+    return file_resource.Pass();
+  }
+
+  scoped_ptr<google_apis::ChangeResource> CreateChangeResourceFromMetadata(
+      const FileMetadata& file) {
+    scoped_ptr<google_apis::ChangeResource> change(
+        new google_apis::ChangeResource);
+    change->set_change_id(file.details().change_id());
+    change->set_file_id(file.file_id());
+    change->set_deleted(file.details().deleted());
+    if (change->is_deleted())
+      return change.Pass();
+
+    change->set_file(CreateFileResourceFromMetadata(file));
     return change.Pass();
   }
 
   void ApplyRenameChangeToMetadata(const std::string& new_title,
                                    FileMetadata* file) {
     FileDetails* details = file->mutable_details();
     details->set_title(new_title);
     details->set_change_id(next_change_id_++);
   }
 
@@ skipping 157 matching lines @@
 
   SyncStatusCode UpdateTracker(const FileTracker& tracker) {
     SyncStatusCode status = SYNC_STATUS_UNKNOWN;
     metadata_database_->UpdateTracker(
         tracker.tracker_id(), tracker.synced_details(),
         base::Bind(&SyncStatusResultCallback, &status));
     message_loop_.RunUntilIdle();
     return status;
   }
 
+  SyncStatusCode PopulateInitialData(
+      int64 largest_change_id,
+      const google_apis::FileResource& sync_root_folder,
+      const ScopedVector<google_apis::FileResource>& app_root_folders) {
+    SyncStatusCode status = SYNC_STATUS_UNKNOWN;
+    metadata_database_->PopulateInitialData(
+        largest_change_id,
+        sync_root_folder,
+        app_root_folders,
+        base::Bind(&SyncStatusResultCallback, &status));
+    message_loop_.RunUntilIdle();
+    return status;
+  }
+
+  void ResetTrackerID(FileTracker* tracker) {
+    tracker->set_tracker_id(GetTrackerIDByFileID(tracker->file_id()));
+  }
+
  private:
   base::ScopedTempDir database_dir_;
   base::MessageLoop message_loop_;
 
   scoped_ptr<MetadataDatabase> metadata_database_;
 
   int64 next_change_id_;
   int64 next_tracker_id_;
   int64 next_file_id_number_;
   int64 next_md5_sequence_number_;
@@ skipping 177 matching lines @@
       CreateChangeResourceFromMetadata(new_file.metadata), &changes);
   EXPECT_EQ(SYNC_STATUS_OK, UpdateByChangeList(changes.Pass()));
 
   renamed_file.tracker.set_dirty(true);
   reorganized_file.tracker.set_dirty(true);
   updated_file.tracker.set_dirty(true);
   noop_file.tracker.set_dirty(true);
   new_file.tracker.clear_synced_details();
   new_file.tracker.set_active(false);
   new_file.tracker.set_dirty(true);
-  new_file.tracker.set_tracker_id(
-      GetTrackerIDByFileID(new_file.metadata.file_id()));
+  ResetTrackerID(&new_file.tracker);
   EXPECT_NE(0, new_file.tracker.tracker_id());
 
   new_file.should_be_absent = false;
 
   VerifyTrackedFiles(tracked_files, arraysize(tracked_files));
   VerifyReloadConsistency();
 }
 
 TEST_F(MetadataDatabaseTest, PopulateFolderTest_RegularFolder) {
   TrackedFile sync_root(CreateTrackedSyncRoot());
@@ skipping 20 matching lines @@
   FileIDList listed_children;
   listed_children.push_back(known_file.metadata.file_id());
   listed_children.push_back(new_file.metadata.file_id());
 
   EXPECT_EQ(SYNC_STATUS_OK,
             PopulateFolder(folder_to_populate.metadata.file_id(),
                            listed_children));
 
   folder_to_populate.tracker.set_dirty(false);
   folder_to_populate.tracker.set_needs_folder_listing(false);
-  new_file.tracker.set_tracker_id(
-      GetTrackerIDByFileID(new_file.metadata.file_id()));
+  ResetTrackerID(&new_file.tracker);
   new_file.tracker.set_dirty(true);
   new_file.tracker.set_active(false);
   new_file.tracker.clear_synced_details();
   new_file.should_be_absent = false;
   new_file.tracker_only = true;
   VerifyTrackedFiles(tracked_files, arraysize(tracked_files));
   VerifyReloadConsistency();
 }
 
 TEST_F(MetadataDatabaseTest, PopulateFolderTest_InactiveFolder) {
@@ skipping 42 matching lines @@
   };
 
   SetUpDatabaseByTrackedFiles(tracked_files, arraysize(tracked_files));
   EXPECT_EQ(SYNC_STATUS_OK, InitializeMetadataDatabase());
   VerifyTrackedFiles(tracked_files, arraysize(tracked_files));
 
   FileIDList disabled_app_children;
   disabled_app_children.push_back(file.metadata.file_id());
   EXPECT_EQ(SYNC_STATUS_OK, PopulateFolder(
       disabled_app_root.metadata.file_id(), disabled_app_children));
-  file.tracker.set_tracker_id(GetTrackerIDByFileID(file.metadata.file_id()));
+  ResetTrackerID(&file.tracker);
   file.tracker.clear_synced_details();
   file.tracker.set_dirty(true);
   file.tracker.set_active(false);
   file.should_be_absent = false;
   file.tracker_only = true;
 
   disabled_app_root.tracker.set_dirty(false);
   disabled_app_root.tracker.set_needs_folder_listing(false);
   VerifyTrackedFiles(tracked_files, arraysize(tracked_files));
   VerifyReloadConsistency();
@@ skipping 43 matching lines @@
        new_conflict.metadata.details();
   new_conflict.tracker.set_dirty(false);
   new_conflict.tracker.set_active(true);
   file.tracker.set_dirty(true);
   file.tracker.set_active(false);
   EXPECT_EQ(SYNC_STATUS_OK, UpdateTracker(new_conflict.tracker));
   VerifyTrackedFiles(tracked_files, arraysize(tracked_files));
   VerifyReloadConsistency();
 }
 
+TEST_F(MetadataDatabaseTest, PopulateInitialDataTest) {
+  TrackedFile sync_root(CreateTrackedSyncRoot());
+  TrackedFile app_root(CreateTrackedFolder(sync_root, "app_root"));
+  app_root.tracker.set_active(false);
+
+  const TrackedFile* tracked_files[] = {
+    &sync_root, &app_root
+  };
+
+  int64 largest_change_id = 42;
+  scoped_ptr<google_apis::FileResource> sync_root_folder(
+      CreateFileResourceFromMetadata(sync_root.metadata));
+  scoped_ptr<google_apis::FileResource> app_root_folder(
+      CreateFileResourceFromMetadata(app_root.metadata));
+
+  ScopedVector<google_apis::FileResource> app_root_folders;
+  app_root_folders.push_back(app_root_folder.release());
+
+  EXPECT_EQ(SYNC_STATUS_OK, InitializeMetadataDatabase());
+  EXPECT_EQ(SYNC_STATUS_OK, PopulateInitialData(
+      largest_change_id,
+      *sync_root_folder,
+      app_root_folders));
+
+  ResetTrackerID(&sync_root.tracker);
+  ResetTrackerID(&app_root.tracker);
+  app_root.tracker.set_parent_tracker_id(sync_root.tracker.tracker_id());
+
+  VerifyTrackedFiles(tracked_files, arraysize(tracked_files));
+  VerifyReloadConsistency();
+}
+
 }  // namespace drive_backend
 }  // namespace sync_file_system