77 lines
2.2 KiB
C++
77 lines
2.2 KiB
C++
// Copyright 2015 The Chromium Authors
|
|
// Use of this source code is governed by a BSD-style license that can be
|
|
// found in the LICENSE file.
|
|
|
|
#include "net/quic/mock_quic_data.h"
|
|
#include "net/base/hex_utils.h"
|
|
|
|
namespace net::test {
|
|
|
|
MockQuicData::MockQuicData(quic::ParsedQuicVersion version)
|
|
: printer_(version) {}
|
|
|
|
MockQuicData::~MockQuicData() = default;
|
|
|
|
void MockQuicData::AddConnect(IoMode mode, int rv) {
|
|
connect_ = std::make_unique<MockConnect>(mode, rv);
|
|
}
|
|
|
|
void MockQuicData::AddRead(IoMode mode,
|
|
std::unique_ptr<quic::QuicEncryptedPacket> packet) {
|
|
reads_.emplace_back(mode, packet->data(), packet->length(),
|
|
sequence_number_++);
|
|
packets_.push_back(std::move(packet));
|
|
}
|
|
void MockQuicData::AddRead(IoMode mode, int rv) {
|
|
reads_.emplace_back(mode, rv, sequence_number_++);
|
|
}
|
|
|
|
void MockQuicData::AddWrite(IoMode mode,
|
|
std::unique_ptr<quic::QuicEncryptedPacket> packet) {
|
|
writes_.emplace_back(mode, packet->data(), packet->length(),
|
|
sequence_number_++);
|
|
packets_.push_back(std::move(packet));
|
|
}
|
|
|
|
void MockQuicData::AddWrite(IoMode mode, int rv) {
|
|
writes_.emplace_back(mode, rv, sequence_number_++);
|
|
}
|
|
|
|
void MockQuicData::AddWrite(IoMode mode,
|
|
int rv,
|
|
std::unique_ptr<quic::QuicEncryptedPacket> packet) {
|
|
writes_.emplace_back(mode, rv, sequence_number_++);
|
|
packets_.push_back(std::move(packet));
|
|
}
|
|
|
|
void MockQuicData::AddSocketDataToFactory(MockClientSocketFactory* factory) {
|
|
factory->AddSocketDataProvider(InitializeAndGetSequencedSocketData());
|
|
}
|
|
|
|
bool MockQuicData::AllReadDataConsumed() {
|
|
return socket_data_->AllReadDataConsumed();
|
|
}
|
|
|
|
bool MockQuicData::AllWriteDataConsumed() {
|
|
return socket_data_->AllWriteDataConsumed();
|
|
}
|
|
|
|
void MockQuicData::Resume() {
|
|
socket_data_->Resume();
|
|
}
|
|
|
|
SequencedSocketData* MockQuicData::InitializeAndGetSequencedSocketData() {
|
|
socket_data_ = std::make_unique<SequencedSocketData>(reads_, writes_);
|
|
socket_data_->set_printer(&printer_);
|
|
if (connect_ != nullptr)
|
|
socket_data_->set_connect_data(*connect_);
|
|
|
|
return socket_data_.get();
|
|
}
|
|
|
|
SequencedSocketData* MockQuicData::GetSequencedSocketData() {
|
|
return socket_data_.get();
|
|
}
|
|
|
|
} // namespace net::test
|