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