| // Copyright 2017 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "storage/browser/test/mock_bytes_provider.h" |
| |
| #include "mojo/common/data_pipe_utils.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| namespace storage { |
| |
| MockBytesProvider::MockBytesProvider( |
| std::vector<uint8_t> data, |
| size_t* reply_request_count, |
| size_t* stream_request_count, |
| size_t* file_request_count, |
| base::Optional<base::Time> file_modification_time) |
| : data_(std::move(data)), |
| reply_request_count_(reply_request_count), |
| stream_request_count_(stream_request_count), |
| file_request_count_(file_request_count), |
| file_modification_time_(file_modification_time) {} |
| |
| MockBytesProvider::~MockBytesProvider() {} |
| |
| void MockBytesProvider::RequestAsReply(RequestAsReplyCallback callback) { |
| if (reply_request_count_) |
| ++*reply_request_count_; |
| std::move(callback).Run(data_); |
| } |
| |
| void MockBytesProvider::RequestAsStream( |
| mojo::ScopedDataPipeProducerHandle pipe) { |
| if (stream_request_count_) |
| ++*stream_request_count_; |
| mojo::common::BlockingCopyFromString( |
| std::string(reinterpret_cast<const char*>(data_.data()), data_.size()), |
| pipe); |
| } |
| |
| void MockBytesProvider::RequestAsFile(uint64_t source_offset, |
| uint64_t source_size, |
| base::File file, |
| uint64_t file_offset, |
| RequestAsFileCallback callback) { |
| if (file_request_count_) |
| ++*file_request_count_; |
| EXPECT_LE(source_offset + source_size, data_.size()); |
| EXPECT_EQ(source_size, |
| static_cast<uint64_t>(file.Write( |
| file_offset, |
| reinterpret_cast<const char*>(data_.data() + source_offset), |
| source_size))); |
| EXPECT_TRUE(file.Flush()); |
| std::move(callback).Run(file_modification_time_); |
| } |
| |
| } // namespace storage |