Index: net/tools/quic/test_tools/quic_test_utils.cc |
diff --git a/net/tools/quic/test_tools/quic_test_utils.cc b/net/tools/quic/test_tools/quic_test_utils.cc |
index 6a5e93c8e593a2237ede843a94e270a5d6403f13..b2420b41f6d41636988e9e029116dc2cfbfe2eb2 100644 |
--- a/net/tools/quic/test_tools/quic_test_utils.cc |
+++ b/net/tools/quic/test_tools/quic_test_utils.cc |
@@ -14,6 +14,9 @@ using base::StringPiece; |
using net::test::MakeAckFrame; |
using net::test::MockHelper; |
using net::test::QuicConnectionPeer; |
+using testing::_; |
+using testing::AnyNumber; |
+using testing::Invoke; |
namespace net { |
namespace tools { |
@@ -117,6 +120,13 @@ MockTimeWaitListManager::MockTimeWaitListManager( |
QuicServerSessionVisitor* visitor, |
EpollServer* eps) |
: QuicTimeWaitListManager(writer, visitor, eps, QuicSupportedVersions()) { |
+ // Though AddConnectionIdToTimeWait is mocked, we want to retain its |
+ // functionality. |
+ EXPECT_CALL(*this, AddConnectionIdToTimeWait(_, _, _)).Times(AnyNumber()); |
+ ON_CALL(*this, AddConnectionIdToTimeWait(_, _, _)) |
+ .WillByDefault( |
+ Invoke(this, &MockTimeWaitListManager:: |
+ QuicTimeWaitListManager_AddConnectionIdToTimeWait)); |
} |
MockTimeWaitListManager::~MockTimeWaitListManager() { |