| // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "quic/test_tools/mock_quic_time_wait_list_manager.h" |
| |
| using testing::_; |
| using testing::Invoke; |
| |
| namespace quic { |
| namespace test { |
| |
| MockTimeWaitListManager::MockTimeWaitListManager( |
| QuicPacketWriter* writer, |
| Visitor* visitor, |
| const QuicClock* clock, |
| QuicAlarmFactory* alarm_factory) |
| : QuicTimeWaitListManager(writer, visitor, clock, alarm_factory) { |
| // Though AddConnectionIdToTimeWait is mocked, we want to retain its |
| // functionality. |
| EXPECT_CALL(*this, AddConnectionIdToTimeWait(_, _)) |
| .Times(testing::AnyNumber()); |
| ON_CALL(*this, AddConnectionIdToTimeWait(_, _)) |
| .WillByDefault( |
| Invoke(this, &MockTimeWaitListManager:: |
| QuicTimeWaitListManager_AddConnectionIdToTimeWait)); |
| } |
| |
| MockTimeWaitListManager::~MockTimeWaitListManager() = default; |
| |
| } // namespace test |
| } // namespace quic |