Index: net/tools/quic/quic_dispatcher_test.cc |
diff --git a/net/tools/quic/quic_dispatcher_test.cc b/net/tools/quic/quic_dispatcher_test.cc |
index 083d85a7435cca5e0cb1b0c62fce70cd60915084..c5898b771e11ca2f12d93bac7128e0cad2fa6a45 100644 |
--- a/net/tools/quic/quic_dispatcher_test.cc |
+++ b/net/tools/quic/quic_dispatcher_test.cc |
@@ -28,7 +28,6 @@ using net::test::ConstructEncryptedPacket; |
using net::test::MockConnection; |
using net::test::MockSession; |
using net::test::ValueRestore; |
-using std::make_pair; |
using std::string; |
using testing::DoAll; |
using testing::InSequence; |
@@ -97,22 +96,6 @@ QuicSession* CreateSession(QuicDispatcher* dispatcher, |
return *session; |
} |
-class MockTimeWaitListManager : public QuicTimeWaitListManager { |
- public: |
- MockTimeWaitListManager(QuicPacketWriter* writer, |
- QuicServerSessionVisitor* visitor, |
- EpollServer* eps) |
- : QuicTimeWaitListManager(writer, visitor, eps, QuicSupportedVersions()) { |
- } |
- |
- MOCK_METHOD5(ProcessPacket, |
- void(const IPEndPoint& server_address, |
- const IPEndPoint& client_address, |
- QuicConnectionId connection_id, |
- QuicPacketSequenceNumber sequence_number, |
- const QuicEncryptedPacket& packet)); |
-}; |
- |
class QuicDispatcherTest : public ::testing::Test { |
public: |
QuicDispatcherTest() |
@@ -257,6 +240,8 @@ TEST_F(QuicDispatcherTest, TimeWaitListManager) { |
// wait list manager. |
EXPECT_CALL(*time_wait_list_manager_, |
ProcessPacket(_, _, connection_id, _, _)).Times(1); |
+ EXPECT_CALL(*time_wait_list_manager_, |
+ AddConnectionIdToTimeWait(_, _, _)).Times(0); |
ProcessPacket(client_address, connection_id, true, "foo"); |
} |
@@ -270,6 +255,8 @@ TEST_F(QuicDispatcherTest, StrayPacketToTimeWaitListManager) { |
EXPECT_CALL(dispatcher_, CreateQuicSession(_, _, _)).Times(0); |
EXPECT_CALL(*time_wait_list_manager_, |
ProcessPacket(_, _, connection_id, _, _)).Times(1); |
+ EXPECT_CALL(*time_wait_list_manager_, |
+ AddConnectionIdToTimeWait(_, _, _)).Times(1); |
ProcessPacket(client_address, connection_id, false, "data"); |
} |
@@ -284,6 +271,8 @@ TEST_F(QuicDispatcherTest, ProcessPacketWithBogusPort) { |
// dispatcher_ should drop this packet. |
EXPECT_CALL(dispatcher_, CreateQuicSession(1, _, client_address)).Times(0); |
EXPECT_CALL(*time_wait_list_manager_, ProcessPacket(_, _, _, _, _)).Times(0); |
+ EXPECT_CALL(*time_wait_list_manager_, |
+ AddConnectionIdToTimeWait(_, _, _)).Times(0); |
ProcessPacket(client_address, 1, true, "foo"); |
EXPECT_EQ(client_address, dispatcher_.current_client_address()); |
EXPECT_EQ(server_address_, dispatcher_.current_server_address()); |