OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "sync/internal_api/public/attachments/attachment_service_impl.h" | 5 #include "sync/internal_api/public/attachments/attachment_service_impl.h" |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/memory/weak_ptr.h" | 8 #include "base/memory/weak_ptr.h" |
9 #include "base/message_loop/message_loop.h" | 9 #include "base/message_loop/message_loop.h" |
10 #include "base/run_loop.h" | 10 #include "base/run_loop.h" |
(...skipping 66 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
77 write_attachments.pop_back(); | 77 write_attachments.pop_back(); |
78 base::MessageLoop::current()->PostTask(FROM_HERE, | 78 base::MessageLoop::current()->PostTask(FROM_HERE, |
79 base::Bind(callback, result)); | 79 base::Bind(callback, result)); |
80 } | 80 } |
81 | 81 |
82 std::vector<AttachmentIdList> read_ids; | 82 std::vector<AttachmentIdList> read_ids; |
83 std::vector<ReadCallback> read_callbacks; | 83 std::vector<ReadCallback> read_callbacks; |
84 std::vector<AttachmentList> write_attachments; | 84 std::vector<AttachmentList> write_attachments; |
85 std::vector<WriteCallback> write_callbacks; | 85 std::vector<WriteCallback> write_callbacks; |
86 | 86 |
| 87 private: |
| 88 virtual ~MockAttachmentStore() {} |
| 89 |
87 DISALLOW_COPY_AND_ASSIGN(MockAttachmentStore); | 90 DISALLOW_COPY_AND_ASSIGN(MockAttachmentStore); |
88 }; | 91 }; |
89 | 92 |
90 class MockAttachmentDownloader | 93 class MockAttachmentDownloader |
91 : public AttachmentDownloader, | 94 : public AttachmentDownloader, |
92 public base::SupportsWeakPtr<MockAttachmentDownloader> { | 95 public base::SupportsWeakPtr<MockAttachmentDownloader> { |
93 public: | 96 public: |
94 MockAttachmentDownloader() {} | 97 MockAttachmentDownloader() {} |
95 | 98 |
96 virtual void DownloadAttachment(const AttachmentId& id, | 99 virtual void DownloadAttachment(const AttachmentId& id, |
(...skipping 70 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
167 // AttachmentService::Delegate implementation. | 170 // AttachmentService::Delegate implementation. |
168 virtual void OnAttachmentUploaded( | 171 virtual void OnAttachmentUploaded( |
169 const AttachmentId& attachment_id) OVERRIDE { | 172 const AttachmentId& attachment_id) OVERRIDE { |
170 on_attachment_uploaded_list_.push_back(attachment_id); | 173 on_attachment_uploaded_list_.push_back(attachment_id); |
171 } | 174 } |
172 | 175 |
173 void InitializeAttachmentService( | 176 void InitializeAttachmentService( |
174 scoped_ptr<MockAttachmentUploader> uploader, | 177 scoped_ptr<MockAttachmentUploader> uploader, |
175 scoped_ptr<MockAttachmentDownloader> downloader, | 178 scoped_ptr<MockAttachmentDownloader> downloader, |
176 AttachmentService::Delegate* delegate) { | 179 AttachmentService::Delegate* delegate) { |
177 scoped_ptr<MockAttachmentStore> attachment_store(new MockAttachmentStore()); | 180 scoped_refptr<MockAttachmentStore> attachment_store( |
| 181 new MockAttachmentStore()); |
178 attachment_store_ = attachment_store->AsWeakPtr(); | 182 attachment_store_ = attachment_store->AsWeakPtr(); |
179 | 183 |
180 if (uploader.get()) { | 184 if (uploader.get()) { |
181 attachment_uploader_ = uploader->AsWeakPtr(); | 185 attachment_uploader_ = uploader->AsWeakPtr(); |
182 } | 186 } |
183 if (downloader.get()) { | 187 if (downloader.get()) { |
184 attachment_downloader_ = downloader->AsWeakPtr(); | 188 attachment_downloader_ = downloader->AsWeakPtr(); |
185 } | 189 } |
186 attachment_service_.reset( | 190 attachment_service_.reset( |
187 new AttachmentServiceImpl(attachment_store.PassAs<AttachmentStore>(), | 191 new AttachmentServiceImpl(attachment_store, |
188 uploader.PassAs<AttachmentUploader>(), | 192 uploader.PassAs<AttachmentUploader>(), |
189 downloader.PassAs<AttachmentDownloader>(), | 193 downloader.PassAs<AttachmentDownloader>(), |
190 delegate)); | 194 delegate)); |
191 } | 195 } |
192 | 196 |
193 AttachmentService* attachment_service() { return attachment_service_.get(); } | 197 AttachmentService* attachment_service() { return attachment_service_.get(); } |
194 | 198 |
195 AttachmentService::GetOrDownloadCallback download_callback() { | 199 AttachmentService::GetOrDownloadCallback download_callback() { |
196 return base::Bind(&AttachmentServiceImplTest::DownloadDone, | 200 return base::Bind(&AttachmentServiceImplTest::DownloadDone, |
197 base::Unretained(this)); | 201 base::Unretained(this)); |
(...skipping 281 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
479 AttachmentUploader::UPLOAD_UNSPECIFIED_ERROR); | 483 AttachmentUploader::UPLOAD_UNSPECIFIED_ERROR); |
480 uploader()->RespondToUpload(uploader()->upload_requests.begin()->first, | 484 uploader()->RespondToUpload(uploader()->upload_requests.begin()->first, |
481 AttachmentUploader::UPLOAD_SUCCESS); | 485 AttachmentUploader::UPLOAD_SUCCESS); |
482 EXPECT_EQ(0U, uploader()->upload_requests.size()); | 486 EXPECT_EQ(0U, uploader()->upload_requests.size()); |
483 RunLoop(); | 487 RunLoop(); |
484 | 488 |
485 EXPECT_EQ(2U, on_attachment_uploaded_list().size()); | 489 EXPECT_EQ(2U, on_attachment_uploaded_list().size()); |
486 } | 490 } |
487 | 491 |
488 } // namespace syncer | 492 } // namespace syncer |
OLD | NEW |