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 202edf3e152cb6b006391297705e7d684319ccda..2766209ae2ffb5e010f3d1b8cc864408d05d5508 100644 |
--- a/net/tools/quic/test_tools/quic_test_utils.h |
+++ b/net/tools/quic/test_tools/quic_test_utils.h |
@@ -78,17 +78,17 @@ class MockConnection : public QuicConnection { |
const IPEndPoint& peer_address, |
const QuicEncryptedPacket& packet)); |
MOCK_METHOD1(SendConnectionClose, void(QuicErrorCode error)); |
- MOCK_METHOD2(SendConnectionCloseWithDetails, void( |
- QuicErrorCode error, |
- const std::string& details)); |
- MOCK_METHOD2(SendConnectionClosePacket, void(QuicErrorCode error, |
- const std::string& details)); |
+ MOCK_METHOD2(SendConnectionCloseWithDetails, |
+ void(QuicErrorCode error, const std::string& details)); |
+ MOCK_METHOD2(SendConnectionClosePacket, |
+ void(QuicErrorCode error, const std::string& details)); |
MOCK_METHOD3(SendRstStream, void(QuicStreamId id, |
QuicRstStreamErrorCode error, |
QuicStreamOffset bytes_written)); |
- MOCK_METHOD3(SendGoAway, void(QuicErrorCode error, |
- QuicStreamId last_good_stream_id, |
- const std::string& reason)); |
+ MOCK_METHOD3(SendGoAway, |
+ void(QuicErrorCode error, |
+ QuicStreamId last_good_stream_id, |
+ const std::string& reason)); |
MOCK_METHOD1(SendBlocked, void(QuicStreamId id)); |
MOCK_METHOD2(SendWindowUpdate, void(QuicStreamId id, |
QuicStreamOffset byte_offset)); |
@@ -157,6 +157,10 @@ class MockQuicServerSessionVisitor : public QuicServerSessionVisitor { |
QuicErrorCode error)); |
MOCK_METHOD1(OnWriteBlocked, |
void(QuicBlockedWriterInterface* blocked_writer)); |
+ MOCK_METHOD1(OnConnectionAddedToTimeWaitList, |
+ void(QuicConnectionId connection_id)); |
+ MOCK_METHOD1(OnConnectionRemovedFromTimeWaitList, |
+ void(QuicConnectionId connection_id)); |
private: |
DISALLOW_COPY_AND_ASSIGN(MockQuicServerSessionVisitor); |