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 { |