net/quic/quic_connection_logger_unittest.cc

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "net/quic/quic_connection_logger.h"
   6
   7 #include "net/quic/quic_protocol.h"
   8 #include "net/quic/test_tools/quic_test_utils.h"
   9 #include "testing/gtest/include/gtest/gtest.h"
  10
  11 namespace net {
  12 namespace test {
  13
  14 class QuicConnectionLoggerPeer {
  15  public:
  16   static size_t num_truncated_acks_sent(const QuicConnectionLogger& logger) {
  17     return logger.num_truncated_acks_sent_;
  18   }
  19 };
  20
  21 class QuicConnectionLoggerTest : public ::testing::Test {
  22  protected:
  23   QuicConnectionLoggerTest()
  24       : session_(new MockConnection(Perspective::IS_CLIENT)),
  25         logger_(&session_, "CONNECTION_UNKNOWN", net_log_) {}
  26
  27   BoundNetLog net_log_;
  28   MockSession session_;
  29   QuicConnectionLogger logger_;
  30 };
  31
  32 TEST_F(QuicConnectionLoggerTest, TruncatedAcksSentNotChanged) {
  33   QuicAckFrame frame;
  34   logger_.OnFrameAddedToPacket(QuicFrame(&frame));
  35   EXPECT_EQ(0u, QuicConnectionLoggerPeer::num_truncated_acks_sent(logger_));
  36
  37   for (QuicPacketSequenceNumber i = 0; i < 256; ++i) {
  38     frame.missing_packets.insert(i);
  39   }
  40   logger_.OnFrameAddedToPacket(QuicFrame(&frame));
  41   EXPECT_EQ(0u, QuicConnectionLoggerPeer::num_truncated_acks_sent(logger_));
  42 }
  43
  44 TEST_F(QuicConnectionLoggerTest, TruncatedAcksSent) {
  45   QuicAckFrame frame;
  46   for (QuicPacketSequenceNumber i = 0; i < 512; i += 2) {
  47     frame.missing_packets.insert(i);
  48   }
  49   logger_.OnFrameAddedToPacket(QuicFrame(&frame));
  50   EXPECT_EQ(1u, QuicConnectionLoggerPeer::num_truncated_acks_sent(logger_));
  51 }
  52
  53 }  // namespace test
  54 }  // namespace net