Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(771)

Unified Diff: net/tools/quic/quic_dispatcher_test.cc

Issue 1029463003: Improve tests. Correct things clang_tidy complains about. Make (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Created 5 years, 9 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « net/quic/quic_time_wait_list_manager.cc ('k') | net/tools/quic/quic_time_wait_list_manager.h » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
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_);
« no previous file with comments | « net/quic/quic_time_wait_list_manager.cc ('k') | net/tools/quic/quic_time_wait_list_manager.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698