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 |