[go: nahoru, domu]
Google Git
Sign in
chromium / chromium / src / 5f11548b7ddf26564dbb1bbf5ebf001db48ac6ac / . / net / filter / mock_source_stream.cc
blob: d225c7bc7fc273de674c066a07680bd654f24694 [file] [log] [blame]
xunjieli95ea25f52016-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
Bence Béky5f115482018-06-29 18:31:55[diff] [blame^]7#include <utility>
8
xunjieli95ea25f52016-09-09 21:32:48[diff] [blame]9#include "base/logging.h"
10#include "net/base/io_buffer.h"
mmenkee66c81ae2017-01-03 20:07:58[diff] [blame]11#include "testing/gtest/include/gtest/gtest.h"
xunjieli95ea25f52016-09-09 21:32:48[diff] [blame]12
13namespace net {
14
15MockSourceStream::MockSourceStream()
16 : SourceStream(SourceStream::TYPE_NONE),
mmenkee66c81ae2017-01-03 20:07:58[diff] [blame]17 read_one_byte_at_a_time_(false),
xunjieli95ea25f52016-09-09 21:32:48[diff] [blame]18 awaiting_completion_(false),
19 dest_buffer_(nullptr),
20 dest_buffer_size_(0) {}
21
22MockSourceStream::~MockSourceStream() {
23 DCHECK(!awaiting_completion_);
mmenkee66c81ae2017-01-03 20:07:58[diff] [blame]24 // All data should have been consumed.
25 EXPECT_TRUE(results_.empty());
xunjieli95ea25f52016-09-09 21:32:48[diff] [blame]26}
27
28int MockSourceStream::Read(IOBuffer* dest_buffer,
29 int buffer_size,
Bence Béky5f115482018-06-29 18:31:55[diff] [blame^]30 CompletionOnceCallback callback) {
xunjieli95ea25f52016-09-09 21:32:48[diff] [blame]31 DCHECK(!awaiting_completion_);
32 DCHECK(!results_.empty());
33
34 if (results_.empty())
35 return ERR_UNEXPECTED;
36
37 QueuedResult r = results_.front();
38 DCHECK_GE(buffer_size, r.len);
39 if (r.mode == ASYNC) {
40 awaiting_completion_ = true;
41 dest_buffer_ = dest_buffer;
42 dest_buffer_size_ = buffer_size;
Bence Béky5f115482018-06-29 18:31:55[diff] [blame^]43 callback_ = std::move(callback);
xunjieli95ea25f52016-09-09 21:32:48[diff] [blame]44 return ERR_IO_PENDING;
45 }
46
47 results_.pop();
48 memcpy(dest_buffer->data(), r.data, r.len);
49 return r.error == OK ? r.len : r.error;
50}
51
52std::string MockSourceStream::Description() const {
53 return "";
54}
55
56MockSourceStream::QueuedResult::QueuedResult(const char* data,
57 int len,
58 Error error,
59 Mode mode)
60 : data(data), len(len), error(error), mode(mode) {}
61
62void MockSourceStream::AddReadResult(const char* data,
63 int len,
64 Error error,
65 Mode mode) {
mmenkee66c81ae2017-01-03 20:07:58[diff] [blame]66 if (error != OK) {
67 // Doesn't make any sense to have both an error and data.
68 DCHECK_EQ(len, 0);
69 } else {
70 // The read result must be between 0 and 32k (inclusive) because the read
71 // buffer used in FilterSourceStream is 32k.
72 DCHECK_GE(32 * 1024, len);
73 DCHECK_LE(0, len);
74 }
75
76 if (len > 0 && read_one_byte_at_a_time_) {
77 for (int i = 0; i < len; ++i) {
78 QueuedResult result(data + i, 1, OK, mode);
79 results_.push(result);
80 }
81 return;
82 }
83
xunjieli95ea25f52016-09-09 21:32:48[diff] [blame]84 QueuedResult result(data, len, error, mode);
85 results_.push(result);
86}
87
88void MockSourceStream::CompleteNextRead() {
89 DCHECK(awaiting_completion_);
90
91 awaiting_completion_ = false;
92 QueuedResult r = results_.front();
93 DCHECK_EQ(ASYNC, r.mode);
94 results_.pop();
95 DCHECK_GE(dest_buffer_size_, r.len);
96 memcpy(dest_buffer_->data(), r.data, r.len);
97 dest_buffer_ = nullptr;
Bence Béky5f115482018-06-29 18:31:55[diff] [blame^]98 std::move(callback_).Run(r.error == OK ? r.len : r.error);
xunjieli95ea25f52016-09-09 21:32:48[diff] [blame]99}
100
101} // namespace net