Chromium Code Reviews| Index: sync/internal_api/attachments/attachment_service_impl_unittest.cc |
| diff --git a/sync/internal_api/attachments/attachment_service_impl_unittest.cc b/sync/internal_api/attachments/attachment_service_impl_unittest.cc |
| index db8369989c4cecb00f301620959256c870877e93..56dab22f7ca1b554ea5aa5a7a831d7ff0b650cfb 100644 |
| --- a/sync/internal_api/attachments/attachment_service_impl_unittest.cc |
| +++ b/sync/internal_api/attachments/attachment_service_impl_unittest.cc |
| @@ -260,6 +260,13 @@ class AttachmentServiceImplTest : public testing::Test, |
| } |
| } |
| + AttachmentIdSet AttachmentIdSetFromList(const AttachmentIdList& id_list) { |
|
maniscalco
2015/03/25 00:42:39
Could be static?
pavely
2015/03/25 00:52:11
Done.
|
| + AttachmentIdSet id_set; |
| + std::copy(id_list.begin(), id_list.end(), |
| + std::inserter(id_set, id_set.end())); |
| + return id_set; |
| + } |
| + |
| const std::vector<AttachmentService::GetOrDownloadResult>& |
| download_results() const { |
| return download_results_; |
| @@ -420,10 +427,10 @@ TEST_F(AttachmentServiceImplTest, GetOrDownload_NoDownloader) { |
| } |
| TEST_F(AttachmentServiceImplTest, UploadAttachments_Success) { |
| - AttachmentIdSet attachment_ids; |
| + AttachmentIdList attachment_ids; |
| const unsigned num_attachments = 3; |
| for (unsigned i = 0; i < num_attachments; ++i) { |
| - attachment_ids.insert(AttachmentId::Create(0, 0)); |
| + attachment_ids.push_back(AttachmentId::Create(0, 0)); |
| } |
| attachment_service()->UploadAttachments(attachment_ids); |
| @@ -432,7 +439,7 @@ TEST_F(AttachmentServiceImplTest, UploadAttachments_Success) { |
| // See that the service has issued a read for at least one of the |
| // attachments. |
| ASSERT_GE(store()->read_ids.size(), 1U); |
| - store()->RespondToRead(attachment_ids); |
| + store()->RespondToRead(AttachmentIdSetFromList(attachment_ids)); |
| RunLoop(); |
| ASSERT_GE(uploader()->upload_requests.size(), 1U); |
| uploader()->RespondToUpload(uploader()->upload_requests.begin()->first, |
| @@ -444,9 +451,8 @@ TEST_F(AttachmentServiceImplTest, UploadAttachments_Success) { |
| // See that all the attachments were uploaded. |
| ASSERT_EQ(attachment_ids.size(), on_attachment_uploaded_list().size()); |
| - AttachmentIdSet::const_iterator iter = attachment_ids.begin(); |
| - const AttachmentIdSet::const_iterator end = attachment_ids.end(); |
| - for (iter = attachment_ids.begin(); iter != end; ++iter) { |
| + for (auto iter = attachment_ids.begin(); iter != attachment_ids.end(); |
| + ++iter) { |
| EXPECT_THAT(on_attachment_uploaded_list(), testing::Contains(*iter)); |
| } |
| } |
| @@ -456,13 +462,16 @@ TEST_F(AttachmentServiceImplTest, UploadAttachments_Success_NoDelegate) { |
| make_scoped_ptr(new MockAttachmentDownloader()), |
| NULL); // No delegate. |
| - AttachmentIdSet attachment_ids; |
| - attachment_ids.insert(AttachmentId::Create(0, 0)); |
| + AttachmentIdList attachment_ids; |
| + attachment_ids.push_back(AttachmentId::Create(0, 0)); |
| attachment_service()->UploadAttachments(attachment_ids); |
| RunLoopAndFireTimer(); |
| ASSERT_EQ(1U, store()->read_ids.size()); |
| ASSERT_EQ(0U, uploader()->upload_requests.size()); |
| - store()->RespondToRead(attachment_ids); |
| + AttachmentIdSet id_set; |
| + std::copy(attachment_ids.begin(), attachment_ids.end(), |
|
maniscalco
2015/03/25 00:42:39
Can you use AttachmentIdSetFromList here?
pavely
2015/03/25 00:52:11
Done.
|
| + std::inserter(id_set, id_set.end())); |
| + store()->RespondToRead(id_set); |
| RunLoop(); |
| ASSERT_EQ(0U, store()->read_ids.size()); |
| ASSERT_EQ(1U, uploader()->upload_requests.size()); |
| @@ -473,15 +482,15 @@ TEST_F(AttachmentServiceImplTest, UploadAttachments_Success_NoDelegate) { |
| } |
| TEST_F(AttachmentServiceImplTest, UploadAttachments_SomeMissingFromStore) { |
| - AttachmentIdSet attachment_ids; |
| - attachment_ids.insert(AttachmentId::Create(0, 0)); |
| - attachment_ids.insert(AttachmentId::Create(0, 0)); |
| + AttachmentIdList attachment_ids; |
| + attachment_ids.push_back(AttachmentId::Create(0, 0)); |
| + attachment_ids.push_back(AttachmentId::Create(0, 0)); |
| attachment_service()->UploadAttachments(attachment_ids); |
| RunLoopAndFireTimer(); |
| ASSERT_GE(store()->read_ids.size(), 1U); |
| ASSERT_EQ(0U, uploader()->upload_requests.size()); |
| - store()->RespondToRead(attachment_ids); |
| + store()->RespondToRead(AttachmentIdSetFromList(attachment_ids)); |
| RunLoop(); |
| ASSERT_EQ(1U, uploader()->upload_requests.size()); |
| @@ -498,10 +507,10 @@ TEST_F(AttachmentServiceImplTest, UploadAttachments_SomeMissingFromStore) { |
| } |
| TEST_F(AttachmentServiceImplTest, UploadAttachments_AllMissingFromStore) { |
| - AttachmentIdSet attachment_ids; |
| + AttachmentIdList attachment_ids; |
| const unsigned num_attachments = 2; |
| for (unsigned i = 0; i < num_attachments; ++i) { |
| - attachment_ids.insert(AttachmentId::Create(0, 0)); |
| + attachment_ids.push_back(AttachmentId::Create(0, 0)); |
| } |
| attachment_service()->UploadAttachments(attachment_ids); |
| @@ -524,8 +533,8 @@ TEST_F(AttachmentServiceImplTest, UploadAttachments_NoUploader) { |
| make_scoped_ptr(new MockAttachmentDownloader()), |
| this); |
| - AttachmentIdSet attachment_ids; |
| - attachment_ids.insert(AttachmentId::Create(0, 0)); |
| + AttachmentIdList attachment_ids; |
| + attachment_ids.push_back(AttachmentId::Create(0, 0)); |
| attachment_service()->UploadAttachments(attachment_ids); |
| RunLoop(); |
| EXPECT_EQ(0U, store()->read_ids.size()); |
| @@ -534,17 +543,17 @@ TEST_F(AttachmentServiceImplTest, UploadAttachments_NoUploader) { |
| // Upload three attachments. For one of them, server responds with error. |
| TEST_F(AttachmentServiceImplTest, UploadAttachments_OneUploadFails) { |
| - AttachmentIdSet attachment_ids; |
| + AttachmentIdList attachment_ids; |
| const unsigned num_attachments = 3; |
| for (unsigned i = 0; i < num_attachments; ++i) { |
| - attachment_ids.insert(AttachmentId::Create(0, 0)); |
| + attachment_ids.push_back(AttachmentId::Create(0, 0)); |
| } |
| attachment_service()->UploadAttachments(attachment_ids); |
| for (unsigned i = 0; i < 3; ++i) { |
| RunLoopAndFireTimer(); |
| ASSERT_GE(store()->read_ids.size(), 1U); |
| - store()->RespondToRead(attachment_ids); |
| + store()->RespondToRead(AttachmentIdSetFromList(attachment_ids)); |
| RunLoop(); |
| ASSERT_EQ(1U, uploader()->upload_requests.size()); |
| AttachmentUploader::UploadResult result = |
| @@ -566,13 +575,13 @@ TEST_F(AttachmentServiceImplTest, UploadAttachments_OneUploadFails) { |
| // network disconnect/connect events and see that backoff is cleared. |
| TEST_F(AttachmentServiceImplTest, |
| UploadAttachments_ResetBackoffAfterNetworkChange) { |
| - AttachmentIdSet attachment_ids; |
| - attachment_ids.insert(AttachmentId::Create(0, 0)); |
| + AttachmentIdList attachment_ids; |
| + attachment_ids.push_back(AttachmentId::Create(0, 0)); |
| attachment_service()->UploadAttachments(attachment_ids); |
| RunLoopAndFireTimer(); |
| ASSERT_EQ(1U, store()->read_ids.size()); |
| - store()->RespondToRead(attachment_ids); |
| + store()->RespondToRead(AttachmentIdSetFromList(attachment_ids)); |
| RunLoop(); |
| ASSERT_EQ(1U, uploader()->upload_requests.size()); |