| 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_);
|
|
|