Index: net/tools/quic/test_tools/quic_test_utils.h |
diff --git a/net/tools/quic/test_tools/quic_test_utils.h b/net/tools/quic/test_tools/quic_test_utils.h |
index cbd7fbee23cdaab60e0792326b5c1962f8dcf166..9db9fb596a6d014e18b2fd01727bddc54f96ca80 100644 |
--- a/net/tools/quic/test_tools/quic_test_utils.h |
+++ b/net/tools/quic/test_tools/quic_test_utils.h |
@@ -39,11 +39,25 @@ static const uint32 kInitialSessionFlowControlWindowForTest = |
QuicAckFrame MakeAckFrameWithNackRanges(size_t num_nack_ranges, |
QuicPacketSequenceNumber least_unacked); |
+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); |
@@ -83,10 +97,12 @@ class MockConnection : public QuicConnection { |
void ReallyProcessUdpPacket(const IPEndPoint& self_address, |
const IPEndPoint& peer_address, |
const QuicEncryptedPacket& packet) { |
- return QuicConnection::ProcessUdpPacket(self_address, peer_address, packet); |
+ QuicConnection::ProcessUdpPacket(self_address, peer_address, packet); |
} |
- virtual bool OnProtocolVersionMismatch(QuicVersion version) { return false; } |
+ virtual bool OnProtocolVersionMismatch(QuicVersion version) override { |
+ return false; |
+ } |
private: |
scoped_ptr<QuicConnectionHelperInterface> helper_; |