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 ad6fa20156ccd11b7f90488eae15dffaed25a439..3099c4c6053cedc30d2ea29b3d33ad09405eff07 100644 |
--- a/net/tools/quic/test_tools/quic_test_utils.h |
+++ b/net/tools/quic/test_tools/quic_test_utils.h |
@@ -153,8 +153,23 @@ class MockTimeWaitListManager : public QuicTimeWaitListManager { |
MockTimeWaitListManager(QuicPacketWriter* writer, |
QuicServerSessionVisitor* visitor, |
EpollServer* eps); |
+ |
~MockTimeWaitListManager() override; |
+ MOCK_METHOD3(AddConnectionIdToTimeWait, |
+ void(QuicConnectionId connection_id, |
+ QuicVersion version, |
+ QuicEncryptedPacket* close_packet)); |
+ |
+ void QuicTimeWaitListManager_AddConnectionIdToTimeWait( |
+ QuicConnectionId connection_id, |
+ QuicVersion version, |
+ QuicEncryptedPacket* close_packet) { |
+ QuicTimeWaitListManager::AddConnectionIdToTimeWait(connection_id, |
+ version, |
+ close_packet); |
+ } |
+ |
MOCK_METHOD5(ProcessPacket, |
void(const IPEndPoint& server_address, |
const IPEndPoint& client_address, |