| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 "content/browser/download/mock_download_file.h" | 5 #include "content/browser/download/mock_download_file.h" |
| 6 #include "testing/gmock/include/gmock/gmock.h" | 6 #include "testing/gmock/include/gmock/gmock.h" |
| 7 | 7 |
| 8 using ::testing::_; | 8 using ::testing::_; |
| 9 using ::testing::Return; | 9 using ::testing::Return; |
| 10 | 10 |
| 11 namespace content { | 11 namespace content { |
| 12 namespace { | 12 namespace { |
| 13 | 13 |
| 14 void SuccessRun(const DownloadFile::InitializeCallback& callback) { | 14 void SuccessRun(const DownloadFile::InitializeCallback& callback, |
| 15 const DownloadItem::ReceivedSlices& received_slices) { |
| 15 callback.Run(DOWNLOAD_INTERRUPT_REASON_NONE); | 16 callback.Run(DOWNLOAD_INTERRUPT_REASON_NONE); |
| 16 } | 17 } |
| 17 | 18 |
| 18 } // namespace | 19 } // namespace |
| 19 | 20 |
| 20 MockDownloadFile::MockDownloadFile() { | 21 MockDownloadFile::MockDownloadFile() { |
| 21 // This is here because |Initialize()| is normally called right after | 22 // This is here because |Initialize()| is normally called right after |
| 22 // construction. | 23 // construction. |
| 23 ON_CALL(*this, Initialize(_)) | 24 ON_CALL(*this, Initialize(_, _)).WillByDefault(::testing::Invoke(SuccessRun)); |
| 24 .WillByDefault(::testing::Invoke(SuccessRun)); | |
| 25 } | 25 } |
| 26 | 26 |
| 27 MockDownloadFile::~MockDownloadFile() { | 27 MockDownloadFile::~MockDownloadFile() { |
| 28 } | 28 } |
| 29 | 29 |
| 30 void MockDownloadFile::AddByteStream( | 30 void MockDownloadFile::AddByteStream( |
| 31 std::unique_ptr<ByteStreamReader> stream_reader, | 31 std::unique_ptr<ByteStreamReader> stream_reader, |
| 32 int64_t offset, | 32 int64_t offset, |
| 33 int64_t length) { | 33 int64_t length) { |
| 34 // Gmock currently can't mock method that takes move-only parameters, | 34 // Gmock currently can't mock method that takes move-only parameters, |
| 35 // delegate the EXPECT_CALL count to |DoAddByteStream|. | 35 // delegate the EXPECT_CALL count to |DoAddByteStream|. |
| 36 DoAddByteStream(stream_reader.get(), offset, length); | 36 DoAddByteStream(stream_reader.get(), offset, length); |
| 37 } | 37 } |
| 38 | 38 |
| 39 } // namespace content | 39 } // namespace content |
| OLD | NEW |