Index: sync/syncable/directory_unittest.cc |
diff --git a/sync/syncable/directory_unittest.cc b/sync/syncable/directory_unittest.cc |
index fabdc472a64a2998f03cc86f29c50f8824341f13..32fed39544fdcb2b58f5d4d74bf9bfdb5a5493c0 100644 |
--- a/sync/syncable/directory_unittest.cc |
+++ b/sync/syncable/directory_unittest.cc |
@@ -1719,6 +1719,53 @@ TEST_F(SyncableDirectoryTest, Directory_LastReferenceUnlinksAttachments) { |
ASSERT_FALSE(dir()->IsAttachmentLinked(attachment_id_proto)); |
} |
+TEST_F(SyncableDirectoryTest, Directory_GetAttachmentIdsToUpload) { |
+ // Create one attachment, referenced by two entries. |
+ AttachmentId attachment_id = AttachmentId::Create(); |
+ sync_pb::AttachmentIdProto attachment_id_proto = attachment_id.GetProto(); |
+ sync_pb::AttachmentMetadata attachment_metadata; |
+ sync_pb::AttachmentMetadataRecord* record = attachment_metadata.add_record(); |
+ *record->mutable_id() = attachment_id_proto; |
+ const Id id1 = TestIdFactory::FromNumber(-1); |
+ const Id id2 = TestIdFactory::FromNumber(-2); |
+ CreateEntryWithAttachmentMetadata( |
+ PREFERENCES, "some entry", id1, attachment_metadata); |
+ CreateEntryWithAttachmentMetadata( |
+ PREFERENCES, "some other entry", id2, attachment_metadata); |
+ |
+ // See that Directory reports that this attachment is not on the server. |
+ AttachmentIdSet id_set; |
+ { |
+ ReadTransaction trans(FROM_HERE, dir().get()); |
+ dir()->GetAttachmentIdsToUpload(&trans, PREFERENCES, &id_set); |
+ } |
+ ASSERT_EQ(1U, id_set.size()); |
+ ASSERT_EQ(attachment_id, *id_set.begin()); |
+ |
+ // Call again, but this time with a ModelType for which there are no entries. |
+ // See that Directory correctly reports that there are none. |
+ { |
+ ReadTransaction trans(FROM_HERE, dir().get()); |
+ dir()->GetAttachmentIdsToUpload(&trans, PASSWORDS, &id_set); |
+ } |
+ ASSERT_TRUE(id_set.empty()); |
+ |
+ // Now, mark the attachment as "on the server" via entry_1. |
+ { |
+ WriteTransaction trans(FROM_HERE, UNITTEST, dir().get()); |
+ MutableEntry entry_1(&trans, GET_BY_ID, id1); |
+ entry_1.MarkAttachmentAsOnServer(attachment_id_proto); |
+ } |
+ |
+ // See that Directory no longer reports that this attachment is not on the |
+ // server. |
+ { |
+ ReadTransaction trans(FROM_HERE, dir().get()); |
+ dir()->GetAttachmentIdsToUpload(&trans, PREFERENCES, &id_set); |
+ } |
+ ASSERT_TRUE(id_set.empty()); |
+} |
+ |
} // namespace syncable |
} // namespace syncer |