Upstream version 7.36.149.0

[platform/framework/web/crosswalk.git] / src / net / tools / quic / quic_time_wait_list_manager_test.cc

diff --git a/src/net/tools/quic/quic_time_wait_list_manager_test.cc b/src/net/tools/quic/quic_time_wait_list_manager_test.cc
index 66333d1..d58b4e1 100644 (file)
--- a/src/net/tools/quic/quic_time_wait_list_manager_test.cc
+++ b/src/net/tools/quic/quic_time_wait_list_manager_test.cc
@@ -21,7 +21,9 @@
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
-using net::test::FramerVisitorCapturingPublicReset;
+using net::test::NoOpFramerVisitor;
+using net::test::QuicVersionMax;
+using net::test::QuicVersionMin;
 using testing::_;
 using testing::Args;
 using testing::Assign;
@@ -39,6 +41,24 @@ namespace net {
 namespace tools {
 namespace test {
 
+class FramerVisitorCapturingPublicReset : public NoOpFramerVisitor {
+ public:
+  FramerVisitorCapturingPublicReset() {}
+  virtual ~FramerVisitorCapturingPublicReset() {}
+
+  virtual void OnPublicResetPacket(
+      const QuicPublicResetPacket& public_reset) OVERRIDE {
+    public_reset_packet_ = public_reset;
+  }
+
+  const QuicPublicResetPacket public_reset_packet() {
+    return public_reset_packet_;
+  }
+
+ private:
+  QuicPublicResetPacket public_reset_packet_;
+};
+
 class QuicTimeWaitListManagerPeer {
  public:
   static bool ShouldSendResponse(QuicTimeWaitListManager* manager,
@@ -85,7 +105,7 @@ class QuicTimeWaitListManagerTest : public ::testing::Test {
   }
 
   void AddConnectionId(QuicConnectionId connection_id) {
-    AddConnectionId(connection_id, net::test::QuicVersionMax(), NULL);
+    AddConnectionId(connection_id, QuicVersionMax(), NULL);
   }
 
   void AddConnectionId(QuicConnectionId connection_id,
@@ -101,10 +121,12 @@ class QuicTimeWaitListManagerTest : public ::testing::Test {
 
   void ProcessPacket(QuicConnectionId connection_id,
                      QuicPacketSequenceNumber sequence_number) {
+    QuicEncryptedPacket packet(NULL, 0);
     time_wait_list_manager_.ProcessPacket(server_address_,
                                           client_address_,
                                           connection_id,
-                                          sequence_number);
+                                          sequence_number,
+                                          packet);
   }
 
   QuicEncryptedPacket* ConstructEncryptedPacket(
@@ -202,7 +224,7 @@ TEST_F(QuicTimeWaitListManagerTest, SendConnectionClose) {
   size_t kConnectionCloseLength = 100;
   AddConnectionId(
       connection_id_,
-      net::test::QuicVersionMax(),
+      QuicVersionMax(),
       new QuicEncryptedPacket(
           new char[kConnectionCloseLength], kConnectionCloseLength, true));
   const int kRandomSequenceNumber = 1;
@@ -354,17 +376,17 @@ TEST_F(QuicTimeWaitListManagerTest, GetQuicVersionFromMap) {
   const int kConnectionId2 = 456;
   const int kConnectionId3 = 789;
 
-  AddConnectionId(kConnectionId1, net::test::QuicVersionMin(), NULL);
-  AddConnectionId(kConnectionId2, net::test::QuicVersionMax(), NULL);
-  AddConnectionId(kConnectionId3, net::test::QuicVersionMax(), NULL);
+  AddConnectionId(kConnectionId1, QuicVersionMin(), NULL);
+  AddConnectionId(kConnectionId2, QuicVersionMax(), NULL);
+  AddConnectionId(kConnectionId3, QuicVersionMax(), NULL);
 
-  EXPECT_EQ(net::test::QuicVersionMin(),
+  EXPECT_EQ(QuicVersionMin(),
             QuicTimeWaitListManagerPeer::GetQuicVersionFromConnectionId(
                 &time_wait_list_manager_, kConnectionId1));
-  EXPECT_EQ(net::test::QuicVersionMax(),
+  EXPECT_EQ(QuicVersionMax(),
             QuicTimeWaitListManagerPeer::GetQuicVersionFromConnectionId(
                 &time_wait_list_manager_, kConnectionId2));
-  EXPECT_EQ(net::test::QuicVersionMax(),
+  EXPECT_EQ(QuicVersionMax(),
             QuicTimeWaitListManagerPeer::GetQuicVersionFromConnectionId(
                 &time_wait_list_manager_, kConnectionId3));
 }
@@ -377,7 +399,7 @@ TEST_F(QuicTimeWaitListManagerTest, AddConnectionIdTwice) {
   size_t kConnectionCloseLength = 100;
   AddConnectionId(
       connection_id_,
-      net::test::QuicVersionMax(),
+      QuicVersionMax(),
       new QuicEncryptedPacket(
           new char[kConnectionCloseLength], kConnectionCloseLength, true));
   EXPECT_TRUE(IsConnectionIdInTimeWait(connection_id_));