| Index: net/quic/core/quic_packet_generator_test.cc
|
| diff --git a/net/quic/core/quic_packet_generator_test.cc b/net/quic/core/quic_packet_generator_test.cc
|
| index d72e191c4cd79ecf6880b01b1efe4581eede8f32..254098ca536d972fafa682a0fdf0661564dcb96a 100644
|
| --- a/net/quic/core/quic_packet_generator_test.cc
|
| +++ b/net/quic/core/quic_packet_generator_test.cc
|
| @@ -22,6 +22,7 @@
|
| #include "net/quic/test_tools/quic_packet_creator_peer.h"
|
| #include "net/quic/test_tools/quic_packet_generator_peer.h"
|
| #include "net/quic/test_tools/quic_test_utils.h"
|
| +#include "net/quic/test_tools/simple_data_producer.h"
|
| #include "net/quic/test_tools/simple_quic_framer.h"
|
|
|
| using std::string;
|
| @@ -69,7 +70,23 @@ class MockDelegate : public QuicPacketGenerator::DelegateInterface {
|
| .WillRepeatedly(Return(true));
|
| }
|
|
|
| + void SaveStreamData(QuicStreamId id,
|
| + QuicIOVector iov,
|
| + size_t iov_offset,
|
| + QuicStreamOffset offset,
|
| + QuicByteCount data_length) override {
|
| + producer_.SaveStreamData(id, iov, iov_offset, offset, data_length);
|
| + }
|
| + bool WriteStreamData(QuicStreamId id,
|
| + QuicStreamOffset offset,
|
| + QuicByteCount data_length,
|
| + QuicDataWriter* writer) override {
|
| + return producer_.WriteStreamData(id, offset, data_length, writer);
|
| + }
|
| +
|
| private:
|
| + SimpleDataProducer producer_;
|
| +
|
| DISALLOW_COPY_AND_ASSIGN(MockDelegate);
|
| };
|
|
|
| @@ -116,6 +133,8 @@ class QuicPacketGeneratorTest : public QuicTest {
|
| creator_->SetEncrypter(ENCRYPTION_FORWARD_SECURE,
|
| new NullEncrypter(Perspective::IS_CLIENT));
|
| creator_->set_encryption_level(ENCRYPTION_FORWARD_SECURE);
|
| + generator_.SetDelegateSavesData(
|
| + FLAGS_quic_reloadable_flag_quic_stream_owns_data);
|
| }
|
|
|
| ~QuicPacketGeneratorTest() override {
|
| @@ -228,6 +247,7 @@ class QuicPacketGeneratorTest : public QuicTest {
|
| private:
|
| std::unique_ptr<char[]> data_array_;
|
| struct iovec iov_;
|
| + SimpleDataProducer producer_;
|
| };
|
|
|
| class MockDebugDelegate : public QuicPacketCreator::DebugDelegate {
|
|
|