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()); |