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 ea5e5efa64a0e28806349278f6fd89417bf23467..fa987de6ea4f6e0aaf5c545fbd508c61c3b2d49a 100644 |
--- a/net/quic/test_tools/quic_test_utils.h |
+++ b/net/quic/test_tools/quic_test_utils.h |
@@ -265,11 +265,25 @@ class MockHelper : public QuicConnectionHelperInterface { |
DISALLOW_COPY_AND_ASSIGN(MockHelper); |
}; |
+class NiceMockPacketWriterFactory : public QuicConnection::PacketWriterFactory { |
+ public: |
+ NiceMockPacketWriterFactory() {} |
+ ~NiceMockPacketWriterFactory() override {} |
+ |
+ QuicPacketWriter* Create(QuicConnection* /*connection*/) const override; |
+ |
+ private: |
+ DISALLOW_COPY_AND_ASSIGN(NiceMockPacketWriterFactory); |
+}; |
+ |
class MockConnection : public QuicConnection { |
public: |
// Uses a MockHelper, ConnectionId of 42, and 127.0.0.1:123. |
explicit MockConnection(bool is_server); |
+ // Uses a MockHelper, ConnectionId of 42, and 127.0.0.1:123. |
+ MockConnection(bool is_server, bool is_secure); |
+ |
// Uses a MockHelper, ConnectionId of 42. |
MockConnection(IPEndPoint address, bool is_server); |
@@ -437,7 +451,9 @@ class MockSendAlgorithm : public SendAlgorithmInterface { |
MockSendAlgorithm(); |
virtual ~MockSendAlgorithm(); |
- MOCK_METHOD2(SetFromConfig, void(const QuicConfig& config, bool is_server)); |
+ MOCK_METHOD3(SetFromConfig, void(const QuicConfig& config, |
+ bool is_server, |
+ bool using_pacing)); |
MOCK_METHOD1(SetNumEmulatedConnections, void(int num_connections)); |
MOCK_METHOD1(SetMaxPacketSize, void(QuicByteCount max_packet_size)); |
MOCK_METHOD2(OnIncomingQuicCongestionFeedbackFrame, |