net/quic/quic_sent_entropy_manager_test.cc

   1 // Copyright 2013 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_sent_entropy_manager.h"
   6
   7 #include <algorithm>
   8 #include <vector>
   9
  10 #include "testing/gmock/include/gmock/gmock.h"
  11 #include "testing/gtest/include/gtest/gtest.h"
  12
  13 using std::make_pair;
  14 using std::pair;
  15 using std::vector;
  16
  17 namespace net {
  18 namespace test {
  19 namespace {
  20
  21 class QuicSentEntropyManagerTest : public ::testing::Test {
  22  protected:
  23   QuicSentEntropyManager entropy_manager_;
  24 };
  25
  26 TEST_F(QuicSentEntropyManagerTest, SentEntropyHash) {
  27   EXPECT_EQ(0, entropy_manager_.EntropyHash(0));
  28
  29   vector<pair<QuicPacketSequenceNumber, QuicPacketEntropyHash> > entropies;
  30   entropies.push_back(make_pair(1, 12));
  31   entropies.push_back(make_pair(2, 1));
  32   entropies.push_back(make_pair(3, 33));
  33   entropies.push_back(make_pair(4, 3));
  34
  35   for (size_t i = 0; i < entropies.size(); ++i) {
  36     entropy_manager_.RecordPacketEntropyHash(entropies[i].first,
  37                                              entropies[i].second);
  38   }
  39
  40   QuicPacketEntropyHash hash = 0;
  41   for (size_t i = 0; i < entropies.size(); ++i) {
  42     hash ^= entropies[i].second;
  43     EXPECT_EQ(hash, entropy_manager_.EntropyHash(i + 1));
  44   }
  45 }
  46
  47 TEST_F(QuicSentEntropyManagerTest, IsValidEntropy) {
  48   QuicPacketEntropyHash entropies[10] =
  49       {12, 1, 33, 3, 32, 100, 28, 42, 22, 255};
  50   for (size_t i = 0; i < 10; ++i) {
  51     entropy_manager_.RecordPacketEntropyHash(i + 1, entropies[i]);
  52   }
  53
  54   SequenceNumberSet missing_packets;
  55   missing_packets.insert(1);
  56   missing_packets.insert(4);
  57   missing_packets.insert(7);
  58   missing_packets.insert(8);
  59
  60   QuicPacketEntropyHash entropy_hash = 0;
  61   for (size_t i = 0; i < 10; ++i) {
  62     if (missing_packets.find(i + 1) == missing_packets.end()) {
  63       entropy_hash ^= entropies[i];
  64     }
  65   }
  66
  67   EXPECT_TRUE(entropy_manager_.IsValidEntropy(10, missing_packets,
  68                                               entropy_hash));
  69 }
  70
  71 }  // namespace
  72 }  // namespace test
  73 }  // namespace net