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 7e7957ba8cd9ff77404bfbf49e6855960c7f7bf8..d111fa0d3a029b38234a4679460cc3c453ffcd51 100644 |
--- a/net/quic/test_tools/quic_test_utils.h |
+++ b/net/quic/test_tools/quic_test_utils.h |
@@ -12,6 +12,7 @@ |
#include "base/strings/string_piece.h" |
#include "net/quic/congestion_control/send_algorithm_interface.h" |
+#include "net/quic/quic_ack_notifier.h" |
#include "net/quic/quic_connection.h" |
#include "net/quic/quic_framer.h" |
#include "net/quic/quic_session.h" |
@@ -137,6 +138,9 @@ class FramerVisitorCapturingFrames : public NoOpFramerVisitor { |
FramerVisitorCapturingFrames(); |
virtual ~FramerVisitorCapturingFrames(); |
+ // Reset the visitor to it's initial state. |
+ void Reset(); |
+ |
// NoOpFramerVisitor |
virtual void OnVersionNegotiationPacket( |
const QuicVersionNegotiationPacket& packet) OVERRIDE; |
@@ -294,11 +298,13 @@ class MockSession : public QuicSession { |
ReliableQuicStream*(QuicStreamId id)); |
MOCK_METHOD0(GetCryptoStream, QuicCryptoStream*()); |
MOCK_METHOD0(CreateOutgoingReliableStream, ReliableQuicStream*()); |
- MOCK_METHOD5(WritevData, QuicConsumedData(QuicStreamId id, |
- const struct iovec* iov, |
- int count, |
- QuicStreamOffset offset, |
- bool fin)); |
+ MOCK_METHOD6(WritevData, |
+ QuicConsumedData(QuicStreamId id, |
+ const struct iovec* iov, |
+ int count, |
+ QuicStreamOffset offset, |
+ bool fin, |
+ QuicAckNotifier::DelegateInterface*)); |
MOCK_METHOD0(IsHandshakeComplete, bool()); |
MOCK_METHOD0(IsCryptoHandshakeConfirmed, bool()); |