| xunjieli | 95ea25f5 | 2016-09-09 21:32:48 | [diff] [blame^] | 1 | // Copyright 2015 The Chromium Authors. All rights reserved. |
| 2 | // Use of this source code is governed by a BSD-style license that can be |
| 3 | // found in the LICENSE file. |
| 4 | |
| 5 | #include "net/filter/mock_source_stream.h" |
| 6 | |
| 7 | #include "base/logging.h" |
| 8 | #include "net/base/io_buffer.h" |
| 9 | |
| 10 | namespace net { |
| 11 | |
| 12 | MockSourceStream::MockSourceStream() |
| 13 | : SourceStream(SourceStream::TYPE_NONE), |
| 14 | awaiting_completion_(false), |
| 15 | dest_buffer_(nullptr), |
| 16 | dest_buffer_size_(0) {} |
| 17 | |
| 18 | MockSourceStream::~MockSourceStream() { |
| 19 | DCHECK(!awaiting_completion_); |
| 20 | DCHECK(results_.empty()); |
| 21 | } |
| 22 | |
| 23 | int MockSourceStream::Read(IOBuffer* dest_buffer, |
| 24 | int buffer_size, |
| 25 | const CompletionCallback& callback) { |
| 26 | DCHECK(!awaiting_completion_); |
| 27 | DCHECK(!results_.empty()); |
| 28 | |
| 29 | if (results_.empty()) |
| 30 | return ERR_UNEXPECTED; |
| 31 | |
| 32 | QueuedResult r = results_.front(); |
| 33 | DCHECK_GE(buffer_size, r.len); |
| 34 | if (r.mode == ASYNC) { |
| 35 | awaiting_completion_ = true; |
| 36 | dest_buffer_ = dest_buffer; |
| 37 | dest_buffer_size_ = buffer_size; |
| 38 | callback_ = callback; |
| 39 | return ERR_IO_PENDING; |
| 40 | } |
| 41 | |
| 42 | results_.pop(); |
| 43 | memcpy(dest_buffer->data(), r.data, r.len); |
| 44 | return r.error == OK ? r.len : r.error; |
| 45 | } |
| 46 | |
| 47 | std::string MockSourceStream::Description() const { |
| 48 | return ""; |
| 49 | } |
| 50 | |
| 51 | MockSourceStream::QueuedResult::QueuedResult(const char* data, |
| 52 | int len, |
| 53 | Error error, |
| 54 | Mode mode) |
| 55 | : data(data), len(len), error(error), mode(mode) {} |
| 56 | |
| 57 | void MockSourceStream::AddReadResult(const char* data, |
| 58 | int len, |
| 59 | Error error, |
| 60 | Mode mode) { |
| 61 | QueuedResult result(data, len, error, mode); |
| 62 | results_.push(result); |
| 63 | } |
| 64 | |
| 65 | void MockSourceStream::CompleteNextRead() { |
| 66 | DCHECK(awaiting_completion_); |
| 67 | |
| 68 | awaiting_completion_ = false; |
| 69 | QueuedResult r = results_.front(); |
| 70 | DCHECK_EQ(ASYNC, r.mode); |
| 71 | results_.pop(); |
| 72 | DCHECK_GE(dest_buffer_size_, r.len); |
| 73 | memcpy(dest_buffer_->data(), r.data, r.len); |
| 74 | dest_buffer_ = nullptr; |
| 75 | callback_.Run(r.error == OK ? r.len : r.error); |
| 76 | } |
| 77 | |
| 78 | } // namespace net |