| Index: net/tools/quic/quic_time_wait_list_manager_test.cc
|
| diff --git a/net/tools/quic/quic_time_wait_list_manager_test.cc b/net/tools/quic/quic_time_wait_list_manager_test.cc
|
| index 1030182432e4102a6790318a9a29a4068f8b2e5a..104054b4be1fcebc1748a7fbbd4c4f369463f6df 100644
|
| --- a/net/tools/quic/quic_time_wait_list_manager_test.cc
|
| +++ b/net/tools/quic/quic_time_wait_list_manager_test.cc
|
| @@ -229,9 +229,9 @@ TEST_F(QuicTimeWaitListManagerTest, SendConnectionClose) {
|
| /*connection_rejected_statelessly=*/false,
|
| &termination_packets);
|
| const int kRandomSequenceNumber = 1;
|
| - EXPECT_CALL(writer_,
|
| - WritePacket(_, kConnectionCloseLength,
|
| - server_address_.address().bytes(), client_address_))
|
| + EXPECT_CALL(writer_, WritePacket(_, kConnectionCloseLength,
|
| + server_address_.address().bytes(),
|
| + client_address_, _))
|
| .WillOnce(Return(WriteResult(WRITE_STATUS_OK, 1)));
|
|
|
| ProcessPacket(connection_id_, kRandomSequenceNumber);
|
| @@ -249,9 +249,9 @@ TEST_F(QuicTimeWaitListManagerTest, SendTwoConnectionCloses) {
|
| /*connection_rejected_statelessly=*/false,
|
| &termination_packets);
|
| const int kRandomSequenceNumber = 1;
|
| - EXPECT_CALL(writer_,
|
| - WritePacket(_, kConnectionCloseLength,
|
| - server_address_.address().bytes(), client_address_))
|
| + EXPECT_CALL(writer_, WritePacket(_, kConnectionCloseLength,
|
| + server_address_.address().bytes(),
|
| + client_address_, _))
|
| .Times(2)
|
| .WillRepeatedly(Return(WriteResult(WRITE_STATUS_OK, 1)));
|
|
|
| @@ -263,7 +263,7 @@ TEST_F(QuicTimeWaitListManagerTest, SendPublicReset) {
|
| AddConnectionId(connection_id_);
|
| const int kRandomSequenceNumber = 1;
|
| EXPECT_CALL(writer_, WritePacket(_, _, server_address_.address().bytes(),
|
| - client_address_))
|
| + client_address_, _))
|
| .With(Args<0, 1>(
|
| PublicResetPacketEq(connection_id_, kRandomSequenceNumber)))
|
| .WillOnce(Return(WriteResult(WRITE_STATUS_OK, 0)));
|
| @@ -277,7 +277,7 @@ TEST_F(QuicTimeWaitListManagerTest, SendPublicResetWithExponentialBackOff) {
|
| EXPECT_EQ(1u, time_wait_list_manager_.num_connections());
|
| for (int packet_number = 1; packet_number < 101; ++packet_number) {
|
| if ((packet_number & (packet_number - 1)) == 0) {
|
| - EXPECT_CALL(writer_, WritePacket(_, _, _, _))
|
| + EXPECT_CALL(writer_, WritePacket(_, _, _, _, _))
|
| .WillOnce(Return(WriteResult(WRITE_STATUS_OK, 1)));
|
| }
|
| ProcessPacket(connection_id_, packet_number);
|
| @@ -298,7 +298,7 @@ TEST_F(QuicTimeWaitListManagerTest, NoPublicResetForStatelessConnections) {
|
| const int kRandomSequenceNumber = 1;
|
|
|
| EXPECT_CALL(writer_, WritePacket(_, _, server_address_.address().bytes(),
|
| - client_address_))
|
| + client_address_, _))
|
| .WillOnce(Return(WriteResult(WRITE_STATUS_OK, 1)));
|
|
|
| ProcessPacket(connection_id_, kRandomSequenceNumber);
|
| @@ -365,14 +365,14 @@ TEST_F(QuicTimeWaitListManagerTest, SendQueuedPackets) {
|
| ConstructEncryptedPacket(connection_id, packet_number));
|
| // Let first write through.
|
| EXPECT_CALL(writer_, WritePacket(_, _, server_address_.address().bytes(),
|
| - client_address_))
|
| + client_address_, _))
|
| .With(Args<0, 1>(PublicResetPacketEq(connection_id, packet_number)))
|
| .WillOnce(Return(WriteResult(WRITE_STATUS_OK, packet->length())));
|
| ProcessPacket(connection_id, packet_number);
|
|
|
| // write block for the next packet.
|
| EXPECT_CALL(writer_, WritePacket(_, _, server_address_.address().bytes(),
|
| - client_address_))
|
| + client_address_, _))
|
| .With(Args<0, 1>(PublicResetPacketEq(connection_id, packet_number)))
|
| .WillOnce(DoAll(Assign(&writer_is_blocked_, true),
|
| Return(WriteResult(WRITE_STATUS_BLOCKED, EAGAIN))));
|
| @@ -389,7 +389,7 @@ TEST_F(QuicTimeWaitListManagerTest, SendQueuedPackets) {
|
| QuicPacketNumber other_packet_number = 23423;
|
| scoped_ptr<QuicEncryptedPacket> other_packet(
|
| ConstructEncryptedPacket(other_connection_id, other_packet_number));
|
| - EXPECT_CALL(writer_, WritePacket(_, _, _, _)).Times(0);
|
| + EXPECT_CALL(writer_, WritePacket(_, _, _, _, _)).Times(0);
|
| EXPECT_CALL(visitor_, OnWriteBlocked(&time_wait_list_manager_));
|
| ProcessPacket(other_connection_id, other_packet_number);
|
| EXPECT_EQ(2u, time_wait_list_manager_.num_connections());
|
| @@ -397,11 +397,11 @@ TEST_F(QuicTimeWaitListManagerTest, SendQueuedPackets) {
|
| // Now expect all the write blocked public reset packets to be sent again.
|
| writer_is_blocked_ = false;
|
| EXPECT_CALL(writer_, WritePacket(_, _, server_address_.address().bytes(),
|
| - client_address_))
|
| + client_address_, _))
|
| .With(Args<0, 1>(PublicResetPacketEq(connection_id, packet_number)))
|
| .WillOnce(Return(WriteResult(WRITE_STATUS_OK, packet->length())));
|
| EXPECT_CALL(writer_, WritePacket(_, _, server_address_.address().bytes(),
|
| - client_address_))
|
| + client_address_, _))
|
| .With(Args<0, 1>(
|
| PublicResetPacketEq(other_connection_id, other_packet_number)))
|
| .WillOnce(Return(WriteResult(WRITE_STATUS_OK, other_packet->length())));
|
| @@ -450,9 +450,9 @@ TEST_F(QuicTimeWaitListManagerTest, AddConnectionIdTwice) {
|
| EXPECT_TRUE(IsConnectionIdInTimeWait(connection_id_));
|
| EXPECT_EQ(1u, time_wait_list_manager_.num_connections());
|
|
|
| - EXPECT_CALL(writer_,
|
| - WritePacket(_, kConnectionCloseLength,
|
| - server_address_.address().bytes(), client_address_))
|
| + EXPECT_CALL(writer_, WritePacket(_, kConnectionCloseLength,
|
| + server_address_.address().bytes(),
|
| + client_address_, _))
|
| .WillOnce(Return(WriteResult(WRITE_STATUS_OK, 1)));
|
|
|
| const int kRandomSequenceNumber = 1;
|
|
|