Process TeamDrive change in change list.

components/drive/resource_metadata_unittest.cc

 // Copyright (c) 2012 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 "components/drive/chromeos/resource_metadata.h"
 
 #include <stddef.h>
 #include <stdint.h>
 
 #include <algorithm>
@@ skipping 13 matching lines @@
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace drive {
 namespace internal {
 namespace {
 
 // The changestamp of the resource metadata used in
 // ResourceMetadataTest.
 const int64_t kTestChangestamp = 100;
 
+const char kTeamDriveId[] = "TEAM_DRIVE_ID";
+const char kTeamDriveName[] = "Name of a Team Drive";
+const char kTeamDriveName2[] = "New Name of a Team Drive";
+
 // Returns the sorted base names from |entries|.
 std::vector<std::string> GetSortedBaseNames(
     const ResourceEntryVector& entries) {
   std::vector<std::string> base_names;
   for (size_t i = 0; i < entries.size(); ++i)
     base_names.push_back(entries[i].base_name());
   std::sort(base_names.begin(), base_names.end());
 
   return base_names;
 }
@@ skipping 471 matching lines @@
       dir3_local_id, &entry));
 
   // Remove unknown local_id using RemoveEntry.
   EXPECT_EQ(FILE_ERROR_NOT_FOUND, resource_metadata_->RemoveEntry("foo"));
 
   // Try removing root. This should fail.
   EXPECT_EQ(FILE_ERROR_ACCESS_DENIED, resource_metadata_->RemoveEntry(
       util::kDriveGrandRootLocalId));
 }
 
+TEST_F(ResourceMetadataTest, AddTeamDrive) {
+  ResourceEntry entry;
+  EXPECT_EQ(FILE_ERROR_NOT_FOUND,
+            resource_metadata_->GetResourceEntryById(kTeamDriveId, &entry));
+  EXPECT_EQ(FILE_ERROR_OK,
+            resource_metadata_->AddTeamDrive(kTeamDriveId, kTeamDriveName));
+  EXPECT_EQ(FILE_ERROR_OK,
+            resource_metadata_->GetResourceEntryById(kTeamDriveId, &entry));
+  EXPECT_EQ(kTeamDriveName, entry.base_name());
+}
+
+TEST_F(ResourceMetadataTest, RefreshTeamDrive) {
+  ResourceEntry entry;
+  EXPECT_EQ(FILE_ERROR_NOT_FOUND,
+            resource_metadata_->GetResourceEntryById(kTeamDriveId, &entry));
+  EXPECT_EQ(FILE_ERROR_OK,
+            resource_metadata_->AddTeamDrive(kTeamDriveId, kTeamDriveName));
+  EXPECT_EQ(FILE_ERROR_OK, resource_metadata_->RefreshTeamDrive(
+                               kTeamDriveId, kTeamDriveName2));
+  EXPECT_EQ(FILE_ERROR_OK,
+            resource_metadata_->GetResourceEntryById(kTeamDriveId, &entry));
+  EXPECT_EQ(kTeamDriveName2, entry.base_name());
+}
+
 TEST_F(ResourceMetadataTest, GetResourceEntryById_RootDirectory) {
   // Look up the root directory by its ID.
   ResourceEntry entry;
   EXPECT_EQ(FILE_ERROR_OK, resource_metadata_->GetResourceEntryById(
       util::kDriveGrandRootLocalId, &entry));
   EXPECT_EQ("drive", entry.base_name());
 }
 
 TEST_F(ResourceMetadataTest, GetResourceEntryById) {
   // Get file4 by path.
@@ skipping 168 matching lines @@
 
   // The "trash" directory should be empty.
   ASSERT_EQ(FILE_ERROR_OK,
             resource_metadata_->ReadDirectoryByPath(
                 base::FilePath::FromUTF8Unsafe("drive/trash"), &entries));
   EXPECT_TRUE(entries.empty());
 }
 
 }  // namespace internal
 }  // namespace drive