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 8db06789c2df8f9365718f83ea192b7b1dc13a00..74cab69e6a79b2168349ab20e9f85dd2bc744b3f 100644 |
--- a/net/tools/quic/quic_dispatcher_test.cc |
+++ b/net/tools/quic/quic_dispatcher_test.cc |
@@ -58,11 +58,8 @@ class TestDispatcher : public QuicDispatcher { |
class MockServerConnection : public MockConnection { |
public: |
MockServerConnection(QuicGuid guid, |
- IPEndPoint address, |
- int fd, |
- EpollServer* eps, |
QuicDispatcher* dispatcher) |
- : MockConnection(guid, address, fd, eps, true), |
+ : MockConnection(guid, true), |
dispatcher_(dispatcher) { |
} |
void UnregisterOnConnectionClosed() { |
@@ -76,11 +73,9 @@ class MockServerConnection : public MockConnection { |
QuicSession* CreateSession(QuicDispatcher* dispatcher, |
QuicGuid guid, |
const IPEndPoint& addr, |
- MockSession** session, |
- EpollServer* eps) { |
- MockServerConnection* connection = |
- new MockServerConnection(guid, addr, 0, eps, dispatcher); |
- *session = new MockSession(connection, true); |
+ MockSession** session) { |
+ MockServerConnection* connection = new MockServerConnection(guid, dispatcher); |
+ *session = new MockSession(connection); |
ON_CALL(*connection, SendConnectionClose(_)).WillByDefault( |
WithoutArgs(Invoke( |
connection, &MockServerConnection::UnregisterOnConnectionClosed))); |
@@ -123,12 +118,6 @@ class QuicDispatcherTest : public ::testing::Test { |
EXPECT_TRUE(packet.AsStringPiece().find(data_) != StringPiece::npos); |
} |
- IPAddressNumber Loopback4() { |
- net::IPAddressNumber addr; |
- CHECK(net::ParseIPLiteralToNumber("127.0.0.1", &addr)); |
- return addr; |
- } |
- |
EpollServer eps_; |
QuicConfig config_; |
QuicCryptoServerConfig crypto_config_; |
@@ -139,16 +128,16 @@ class QuicDispatcherTest : public ::testing::Test { |
}; |
TEST_F(QuicDispatcherTest, ProcessPackets) { |
- IPEndPoint addr(Loopback4(), 1); |
+ IPEndPoint addr(net::test::Loopback4(), 1); |
EXPECT_CALL(dispatcher_, CreateQuicSession(1, _, addr)) |
.WillOnce(testing::Return(CreateSession( |
- &dispatcher_, 1, addr, &session1_, &eps_))); |
+ &dispatcher_, 1, addr, &session1_))); |
ProcessPacket(addr, 1, true, "foo"); |
EXPECT_CALL(dispatcher_, CreateQuicSession(2, _, addr)) |
.WillOnce(testing::Return(CreateSession( |
- &dispatcher_, 2, addr, &session2_, &eps_))); |
+ &dispatcher_, 2, addr, &session2_))); |
ProcessPacket(addr, 2, true, "bar"); |
data_ = "eep"; |
@@ -160,11 +149,11 @@ TEST_F(QuicDispatcherTest, ProcessPackets) { |
} |
TEST_F(QuicDispatcherTest, Shutdown) { |
- IPEndPoint addr(Loopback4(), 1); |
+ IPEndPoint addr(net::test::Loopback4(), 1); |
EXPECT_CALL(dispatcher_, CreateQuicSession(_, _, addr)) |
.WillOnce(testing::Return(CreateSession( |
- &dispatcher_, 1, addr, &session1_, &eps_))); |
+ &dispatcher_, 1, addr, &session1_))); |
ProcessPacket(addr, 1, true, "foo"); |
@@ -195,11 +184,11 @@ TEST_F(QuicDispatcherTest, TimeWaitListManager) { |
QuicDispatcherPeer::SetTimeWaitListManager(&dispatcher_, |
time_wait_list_manager); |
// Create a new session. |
- IPEndPoint addr(Loopback4(), 1); |
+ IPEndPoint addr(net::test::Loopback4(), 1); |
QuicGuid guid = 1; |
EXPECT_CALL(dispatcher_, CreateQuicSession(guid, _, addr)) |
.WillOnce(testing::Return(CreateSession( |
- &dispatcher_, guid, addr, &session1_, &eps_))); |
+ &dispatcher_, guid, addr, &session1_))); |
ProcessPacket(addr, guid, true, "foo"); |
// Close the connection by sending public reset packet. |
@@ -237,7 +226,7 @@ TEST_F(QuicDispatcherTest, StrayPacketToTimeWaitListManager) { |
QuicDispatcherPeer::SetTimeWaitListManager(&dispatcher_, |
time_wait_list_manager); |
- IPEndPoint addr(Loopback4(), 1); |
+ IPEndPoint addr(net::test::Loopback4(), 1); |
QuicGuid guid = 1; |
// Dispatcher forwards all packets for this guid to the time wait list |
// manager. |
@@ -250,16 +239,16 @@ TEST_F(QuicDispatcherTest, StrayPacketToTimeWaitListManager) { |
class QuicWriteBlockedListTest : public QuicDispatcherTest { |
public: |
virtual void SetUp() { |
- IPEndPoint addr(Loopback4(), 1); |
+ IPEndPoint addr(net::test::Loopback4(), 1); |
EXPECT_CALL(dispatcher_, CreateQuicSession(_, _, addr)) |
.WillOnce(testing::Return(CreateSession( |
- &dispatcher_, 1, addr, &session1_, &eps_))); |
+ &dispatcher_, 1, addr, &session1_))); |
ProcessPacket(addr, 1, true, "foo"); |
EXPECT_CALL(dispatcher_, CreateQuicSession(_, _, addr)) |
.WillOnce(testing::Return(CreateSession( |
- &dispatcher_, 2, addr, &session2_, &eps_))); |
+ &dispatcher_, 2, addr, &session2_))); |
ProcessPacket(addr, 2, true, "bar"); |
blocked_list_ = dispatcher_.write_blocked_list(); |