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 bae805074d5d3e2a21f5f89538ed7173b7966693..631d59f54a533e0a05d18a17a8e198185e02d19d 100644 |
--- a/net/quic/test_tools/quic_test_utils.h |
+++ b/net/quic/test_tools/quic_test_utils.h |
@@ -17,6 +17,7 @@ |
#include "net/quic/quic_client_session_base.h" |
#include "net/quic/quic_connection.h" |
#include "net/quic/quic_framer.h" |
+#include "net/quic/quic_sent_packet_manager.h" |
#include "net/quic/quic_session.h" |
#include "net/quic/test_tools/mock_clock.h" |
#include "net/quic/test_tools/mock_random.h" |
@@ -464,6 +465,7 @@ class MockSendAlgorithm : public SendAlgorithmInterface { |
MOCK_CONST_METHOD0(GetCongestionWindow, QuicByteCount()); |
MOCK_CONST_METHOD0(InSlowStart, bool()); |
MOCK_CONST_METHOD0(GetSlowStartThreshold, QuicByteCount()); |
+ MOCK_CONST_METHOD0(GetCongestionControlType, CongestionControlType()); |
private: |
DISALLOW_COPY_AND_ASSIGN(MockSendAlgorithm); |
@@ -530,6 +532,18 @@ class MockAckNotifierDelegate : public QuicAckNotifier::DelegateInterface { |
DISALLOW_COPY_AND_ASSIGN(MockAckNotifierDelegate); |
}; |
+class MockNetworkChangeVisitor : |
+ public QuicSentPacketManager::NetworkChangeVisitor { |
+ public: |
+ MockNetworkChangeVisitor(); |
+ virtual ~MockNetworkChangeVisitor(); |
+ |
+ MOCK_METHOD1(OnCongestionWindowChange, void(QuicByteCount)); |
+ |
+ private: |
+ DISALLOW_COPY_AND_ASSIGN(MockNetworkChangeVisitor); |
+}; |
+ |
} // namespace test |
} // namespace net |