Index: components/drive/resource_metadata_unittest.cc |
diff --git a/components/drive/resource_metadata_unittest.cc b/components/drive/resource_metadata_unittest.cc |
index 04f0276516f84450ce49dc982f51828e6fb0b0fe..61358293ccddfc2017e37b090d0110cc64c58b4f 100644 |
--- a/components/drive/resource_metadata_unittest.cc |
+++ b/components/drive/resource_metadata_unittest.cc |
@@ -31,6 +31,10 @@ namespace { |
// 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) { |
@@ -522,6 +526,30 @@ TEST_F(ResourceMetadataTest, 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; |