Index: components/drive/resource_metadata_storage_unittest.cc |
diff --git a/components/drive/resource_metadata_storage_unittest.cc b/components/drive/resource_metadata_storage_unittest.cc |
index 5f4a0fa089cc59e84d4a830b86346bd8b858afec..886834dbe2c78acf330cfb13505f09bf149aae6a 100644 |
--- a/components/drive/resource_metadata_storage_unittest.cc |
+++ b/components/drive/resource_metadata_storage_unittest.cc |
@@ -43,6 +43,21 @@ class ResourceMetadataStorageTest : public testing::Test { |
EXPECT_EQ(FILE_ERROR_OK, storage_->PutHeader(header)); |
} |
+ // Overwrites |storage_|'s has_starred. |
+ void SetHasStarred(bool has_starred) { |
+ ResourceMetadataHeader header; |
+ ASSERT_EQ(FILE_ERROR_OK, storage_->GetHeader(&header)); |
+ header.set_has_starred(has_starred); |
+ EXPECT_EQ(FILE_ERROR_OK, storage_->PutHeader(header)); |
+ } |
+ |
+ // Get |storage_|'s has_starred. |
+ void GetHasStarred(bool* has_starred) { |
+ ResourceMetadataHeader header; |
+ EXPECT_EQ(FILE_ERROR_OK, storage_->GetHeader(&header)); |
+ *has_starred = header.has_starred(); |
+ } |
+ |
bool CheckValidity() { |
return storage_->CheckValidity(); |
} |
@@ -634,5 +649,19 @@ TEST_F(ResourceMetadataStorageTest, CheckValidity) { |
EXPECT_TRUE(CheckValidity()); |
} |
+TEST_F(ResourceMetadataStorageTest, ChangeHasStarred) { |
+ // Suppose 'Starred' property has not loaded. |
+ bool has_starred = false; |
+ SetHasStarred(has_starred); |
+ |
+ // Close DB and reopen. |
+ storage_.reset(new ResourceMetadataStorage( |
+ temp_dir_.GetPath(), base::ThreadTaskRunnerHandle::Get().get())); |
+ ASSERT_TRUE(storage_->Initialize()); |
+ |
+ GetHasStarred(&has_starred); |
+ EXPECT_TRUE(has_starred); |
+} |
+ |
} // namespace internal |
} // namespace drive |