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..d404b5a4b24b95dfaa8ed7c7e61e213d3b63d49c 100644 |
--- a/net/tools/quic/quic_dispatcher_test.cc |
+++ b/net/tools/quic/quic_dispatcher_test.cc |
@@ -97,22 +97,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() |
@@ -193,10 +177,9 @@ TEST_F(QuicDispatcherTest, ProcessPackets) { |
EXPECT_EQ(client_address, dispatcher_.current_client_address()); |
EXPECT_EQ(server_address_, dispatcher_.current_server_address()); |
- |
EXPECT_CALL(dispatcher_, CreateQuicSession(2, _, client_address)) |
- .WillOnce(testing::Return(CreateSession( |
- &dispatcher_, 2, client_address, &session2_))); |
+ .WillOnce(testing::Return( |
+ CreateSession(&dispatcher_, 2, client_address, &session2_))); |
ProcessPacket(client_address, 2, true, "bar"); |
EXPECT_CALL(*reinterpret_cast<MockConnection*>(session1_->connection()), |
@@ -210,8 +193,8 @@ TEST_F(QuicDispatcherTest, Shutdown) { |
IPEndPoint client_address(net::test::Loopback4(), 1); |
EXPECT_CALL(dispatcher_, CreateQuicSession(_, _, client_address)) |
- .WillOnce(testing::Return(CreateSession( |
- &dispatcher_, 1, client_address, &session1_))); |
+ .WillOnce(testing::Return( |
+ CreateSession(&dispatcher_, 1, client_address, &session1_))); |
ProcessPacket(client_address, 1, true, "foo"); |
@@ -228,8 +211,8 @@ TEST_F(QuicDispatcherTest, TimeWaitListManager) { |
IPEndPoint client_address(net::test::Loopback4(), 1); |
QuicConnectionId connection_id = 1; |
EXPECT_CALL(dispatcher_, CreateQuicSession(connection_id, _, client_address)) |
- .WillOnce(testing::Return(CreateSession( |
- &dispatcher_, connection_id, client_address, &session1_))); |
+ .WillOnce(testing::Return(CreateSession(&dispatcher_, connection_id, |
+ client_address, &session1_))); |
ProcessPacket(client_address, connection_id, true, "foo"); |
// Close the connection by sending public reset packet. |
@@ -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()); |
@@ -321,13 +310,13 @@ class QuicDispatcherWriteBlockedListTest : public QuicDispatcherTest { |
IPEndPoint client_address(net::test::Loopback4(), 1); |
EXPECT_CALL(dispatcher_, CreateQuicSession(_, _, client_address)) |
- .WillOnce(testing::Return(CreateSession( |
- &dispatcher_, 1, client_address, &session1_))); |
+ .WillOnce(testing::Return( |
+ CreateSession(&dispatcher_, 1, client_address, &session1_))); |
ProcessPacket(client_address, 1, true, "foo"); |
EXPECT_CALL(dispatcher_, CreateQuicSession(_, _, client_address)) |
- .WillOnce(testing::Return(CreateSession( |
- &dispatcher_, 2, client_address, &session2_))); |
+ .WillOnce(testing::Return( |
+ CreateSession(&dispatcher_, 2, client_address, &session2_))); |
ProcessPacket(client_address, 2, true, "bar"); |
blocked_list_ = QuicDispatcherPeer::GetWriteBlockedList(&dispatcher_); |