| Index: net/quic/test_tools/quic_test_utils.h
|
| diff --git a/net/quic/test_tools/quic_test_utils.h b/net/quic/test_tools/quic_test_utils.h
|
| index e6aa72361bb373c6764798b1bd0a2a8b0308b277..f7d1c5fbdc4a90ba201bd5e3bb746121a46ff2cd 100644
|
| --- a/net/quic/test_tools/quic_test_utils.h
|
| +++ b/net/quic/test_tools/quic_test_utils.h
|
| @@ -28,6 +28,7 @@
|
| #include "net/quic/platform/api/quic_string_piece.h"
|
| #include "net/quic/test_tools/mock_clock.h"
|
| #include "net/quic/test_tools/mock_random.h"
|
| +#include "net/quic/test_tools/simple_data_producer.h"
|
| #include "net/test/gtest_util.h"
|
| #include "net/tools/quic/quic_per_connection_packet_writer.h"
|
| #include "net/tools/quic/test_tools/mock_quic_session_visitor.h"
|
| @@ -295,8 +296,19 @@ class MockQuicConnectionVisitor : public QuicConnectionVisitorInterface {
|
| MOCK_METHOD0(OnConfigNegotiated, void());
|
| MOCK_METHOD0(PostProcessAfterData, void());
|
| MOCK_METHOD0(OnAckNeedsRetransmittableFrame, void());
|
| + void SaveStreamData(QuicStreamId id,
|
| + QuicIOVector iov,
|
| + size_t iov_offset,
|
| + QuicStreamOffset offset,
|
| + QuicByteCount data_length) override;
|
| + bool WriteStreamData(QuicStreamId id,
|
| + QuicStreamOffset offset,
|
| + QuicByteCount data_length,
|
| + QuicDataWriter* writer) override;
|
|
|
| private:
|
| + SimpleDataProducer producer_;
|
| +
|
| DISALLOW_COPY_AND_ASSIGN(MockQuicConnectionVisitor);
|
| };
|
|
|
| @@ -511,6 +523,15 @@ class MockQuicSession : public QuicSession {
|
| const QuicReferenceCountedPointer<QuicAckListenerInterface>&
|
| ack_listener);
|
|
|
| + QuicConsumedData ConsumeAndSaveAllData(
|
| + QuicStream* stream,
|
| + QuicStreamId id,
|
| + const QuicIOVector& data,
|
| + QuicStreamOffset offset,
|
| + StreamSendingState state,
|
| + const QuicReferenceCountedPointer<QuicAckListenerInterface>&
|
| + ack_listener);
|
| +
|
| private:
|
| std::unique_ptr<QuicCryptoStream> crypto_stream_;
|
|
|
| @@ -615,6 +636,15 @@ class MockQuicSpdySession : public QuicSpdySession {
|
| return QuicSpdySession::ShouldCreateOutgoingDynamicStream2();
|
| }
|
|
|
| + QuicConsumedData ConsumeAndSaveAllData(
|
| + QuicStream* stream,
|
| + QuicStreamId id,
|
| + const QuicIOVector& data,
|
| + QuicStreamOffset offset,
|
| + StreamSendingState state,
|
| + const QuicReferenceCountedPointer<QuicAckListenerInterface>&
|
| + ack_listener);
|
| +
|
| using QuicSession::ActivateStream;
|
|
|
| private:
|
| @@ -902,6 +932,33 @@ class MockConnectionCloseDelegate
|
| ConnectionCloseSource source));
|
| };
|
|
|
| +class MockPacketCreatorDelegate : public QuicPacketCreator::DelegateInterface {
|
| + public:
|
| + MockPacketCreatorDelegate();
|
| + ~MockPacketCreatorDelegate() override;
|
| +
|
| + MOCK_METHOD1(OnSerializedPacket, void(SerializedPacket* packet));
|
| + MOCK_METHOD3(OnUnrecoverableError,
|
| + void(QuicErrorCode,
|
| + const std::string&,
|
| + ConnectionCloseSource source));
|
| +
|
| + void SaveStreamData(QuicStreamId id,
|
| + QuicIOVector iov,
|
| + size_t iov_offset,
|
| + QuicStreamOffset offset,
|
| + QuicByteCount data_length) override;
|
| + bool WriteStreamData(QuicStreamId id,
|
| + QuicStreamOffset offset,
|
| + QuicByteCount data_length,
|
| + QuicDataWriter* writer) override;
|
| +
|
| + private:
|
| + SimpleDataProducer producer_;
|
| +
|
| + DISALLOW_COPY_AND_ASSIGN(MockPacketCreatorDelegate);
|
| +};
|
| +
|
| // Creates a client session for testing.
|
| //
|
| // server_id: The server id associated with this stream.
|
|
|