OLD | NEW |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | 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 | 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 "net/quic/chromium/mock_quic_data.h" | 5 #include "net/quic/chromium/mock_quic_data.h" |
6 | 6 |
7 namespace net { | 7 namespace net { |
8 namespace test { | 8 namespace test { |
9 | 9 |
10 MockQuicData::MockQuicData() : sequence_number_(0) {} | 10 MockQuicData::MockQuicData() : sequence_number_(0) {} |
(...skipping 20 matching lines...) Expand all Loading... |
31 void MockQuicData::AddRead(IoMode mode, int rv) { | 31 void MockQuicData::AddRead(IoMode mode, int rv) { |
32 reads_.push_back(MockRead(mode, rv, sequence_number_++)); | 32 reads_.push_back(MockRead(mode, rv, sequence_number_++)); |
33 } | 33 } |
34 | 34 |
35 void MockQuicData::AddWrite(std::unique_ptr<QuicEncryptedPacket> packet) { | 35 void MockQuicData::AddWrite(std::unique_ptr<QuicEncryptedPacket> packet) { |
36 writes_.push_back(MockWrite(SYNCHRONOUS, packet->data(), packet->length(), | 36 writes_.push_back(MockWrite(SYNCHRONOUS, packet->data(), packet->length(), |
37 sequence_number_++)); | 37 sequence_number_++)); |
38 packets_.push_back(std::move(packet)); | 38 packets_.push_back(std::move(packet)); |
39 } | 39 } |
40 | 40 |
| 41 void MockQuicData::AddAsyncWrite(std::unique_ptr<QuicEncryptedPacket> packet) { |
| 42 writes_.push_back( |
| 43 MockWrite(ASYNC, packet->data(), packet->length(), sequence_number_++)); |
| 44 packets_.push_back(std::move(packet)); |
| 45 } |
| 46 |
41 void MockQuicData::AddWrite(IoMode mode, int rv) { | 47 void MockQuicData::AddWrite(IoMode mode, int rv) { |
42 writes_.push_back(MockWrite(mode, rv, sequence_number_++)); | 48 writes_.push_back(MockWrite(mode, rv, sequence_number_++)); |
43 } | 49 } |
44 | 50 |
45 void MockQuicData::AddSocketDataToFactory(MockClientSocketFactory* factory) { | 51 void MockQuicData::AddSocketDataToFactory(MockClientSocketFactory* factory) { |
46 MockRead* reads = reads_.empty() ? nullptr : &reads_[0]; | 52 MockRead* reads = reads_.empty() ? nullptr : &reads_[0]; |
47 MockWrite* writes = writes_.empty() ? nullptr : &writes_[0]; | 53 MockWrite* writes = writes_.empty() ? nullptr : &writes_[0]; |
48 socket_data_.reset( | 54 socket_data_.reset( |
49 new SequencedSocketData(reads, reads_.size(), writes, writes_.size())); | 55 new SequencedSocketData(reads, reads_.size(), writes, writes_.size())); |
50 if (connect_ != nullptr) | 56 if (connect_ != nullptr) |
51 socket_data_->set_connect_data(*connect_); | 57 socket_data_->set_connect_data(*connect_); |
52 factory->AddSocketDataProvider(socket_data_.get()); | 58 factory->AddSocketDataProvider(socket_data_.get()); |
53 } | 59 } |
54 | 60 |
55 bool MockQuicData::AllReadDataConsumed() { | 61 bool MockQuicData::AllReadDataConsumed() { |
56 return socket_data_->AllReadDataConsumed(); | 62 return socket_data_->AllReadDataConsumed(); |
57 } | 63 } |
58 | 64 |
59 bool MockQuicData::AllWriteDataConsumed() { | 65 bool MockQuicData::AllWriteDataConsumed() { |
60 return socket_data_->AllWriteDataConsumed(); | 66 return socket_data_->AllWriteDataConsumed(); |
61 } | 67 } |
62 | 68 |
63 void MockQuicData::Resume() { | 69 void MockQuicData::Resume() { |
64 socket_data_->Resume(); | 70 socket_data_->Resume(); |
65 } | 71 } |
66 | 72 |
67 } // namespace test | 73 } // namespace test |
68 } // namespace net | 74 } // namespace net |
OLD | NEW |