| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2016 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/spdy/core/spdy_header_indexing.h" | |
| 6 | |
| 7 #include "base/memory/ptr_util.h" | |
| 8 #include "net/spdy/platform/api/spdy_ptr_util.h" | |
| 9 #include "testing/gmock/include/gmock/gmock.h" | |
| 10 #include "testing/platform_test.h" | |
| 11 | |
| 12 namespace net { | |
| 13 | |
| 14 namespace test { | |
| 15 | |
| 16 class HeaderIndexingPeer { | |
| 17 public: | |
| 18 HeaderIndexingPeer() : hi_() {} | |
| 19 | |
| 20 void CreateTestInit() { | |
| 21 SpdyString input[] = {"key1", "key2", "key3"}; | |
| 22 hi_.indexing_set_ = | |
| 23 HeaderIndexing::HeaderSet(input, input + arraysize(input)); | |
| 24 hi_.tracking_set_ = | |
| 25 HeaderIndexing::HeaderSet(input, input + arraysize(input)); | |
| 26 } | |
| 27 | |
| 28 bool ShouldIndex(SpdyStringPiece header) { | |
| 29 return hi_.ShouldIndex(header, ""); | |
| 30 } | |
| 31 | |
| 32 void CreateInitIndexingHeaders() { hi_.CreateInitIndexingHeaders(); } | |
| 33 | |
| 34 void TryInsert(SpdyString&& header) { | |
| 35 hi_.TryInsertHeader(std::move(header), &(hi_.indexing_set_), | |
| 36 hi_.indexing_set_bound_); | |
| 37 } | |
| 38 | |
| 39 bool InTrackingSet(const SpdyString& str) { | |
| 40 return hi_.tracking_set_.find(str) != hi_.tracking_set_.end(); | |
| 41 } | |
| 42 | |
| 43 size_t indexing_set_size() const { return hi_.indexing_set_.size(); } | |
| 44 | |
| 45 size_t tracking_set_size() const { return hi_.tracking_set_.size(); } | |
| 46 | |
| 47 HeaderIndexing::HeaderSet* indexing_set() { return &(hi_.indexing_set_); } | |
| 48 | |
| 49 HeaderIndexing::HeaderSet* tracking_set() { return &(hi_.tracking_set_); } | |
| 50 | |
| 51 private: | |
| 52 HeaderIndexing hi_; | |
| 53 }; | |
| 54 | |
| 55 class SpdyHeaderIndexingTest : public ::testing::Test { | |
| 56 protected: | |
| 57 SpdyHeaderIndexingTest() { | |
| 58 FLAGS_gfe_spdy_indexing_set_bound = 3; | |
| 59 FLAGS_gfe_spdy_tracking_set_bound = 4; | |
| 60 hi_ = SpdyMakeUnique<HeaderIndexingPeer>(); | |
| 61 hi_->CreateTestInit(); | |
| 62 } | |
| 63 void SetUp() override { | |
| 64 EXPECT_EQ(3u, hi_->indexing_set_size()); | |
| 65 EXPECT_EQ(3u, hi_->tracking_set_size()); | |
| 66 } | |
| 67 std::unique_ptr<HeaderIndexingPeer> hi_; | |
| 68 }; | |
| 69 | |
| 70 TEST_F(SpdyHeaderIndexingTest, TestTryInsertHeader) { | |
| 71 SpdyString key("key4"); | |
| 72 hi_->TryInsert(std::move(key)); | |
| 73 EXPECT_EQ(3u, hi_->indexing_set_size()); | |
| 74 EXPECT_TRUE(hi_->ShouldIndex("key4")); | |
| 75 } | |
| 76 | |
| 77 TEST_F(SpdyHeaderIndexingTest, TestShouldIndex) { | |
| 78 SpdyString key3 = "key3"; | |
| 79 SpdyString key4 = "key4"; | |
| 80 SpdyString key5 = "key5"; | |
| 81 // Cache hit. | |
| 82 EXPECT_TRUE(hi_->ShouldIndex(key3)); | |
| 83 EXPECT_EQ(3u, hi_->indexing_set_size()); | |
| 84 EXPECT_EQ(3u, hi_->tracking_set_size()); | |
| 85 | |
| 86 // Cache miss. Add to tracking set. | |
| 87 EXPECT_FALSE(hi_->ShouldIndex(key4)); | |
| 88 EXPECT_EQ(3u, hi_->indexing_set_size()); | |
| 89 EXPECT_EQ(4u, hi_->tracking_set_size()); | |
| 90 EXPECT_TRUE(hi_->InTrackingSet(key4)); | |
| 91 // Cache miss. Add to indexing set by kicking one entry out. | |
| 92 EXPECT_FALSE(hi_->ShouldIndex(key4)); | |
| 93 EXPECT_EQ(3u, hi_->indexing_set_size()); | |
| 94 EXPECT_EQ(4u, hi_->tracking_set_size()); | |
| 95 EXPECT_TRUE(hi_->InTrackingSet(key4)); | |
| 96 // Cache hit. | |
| 97 EXPECT_TRUE(hi_->ShouldIndex(key4)); | |
| 98 | |
| 99 // Cache miss. Add to tracking set by kicking one entry out. | |
| 100 EXPECT_FALSE(hi_->ShouldIndex(key5)); | |
| 101 EXPECT_EQ(3u, hi_->indexing_set_size()); | |
| 102 EXPECT_EQ(4u, hi_->tracking_set_size()); | |
| 103 EXPECT_TRUE(hi_->ShouldIndex(key4)); | |
| 104 EXPECT_TRUE(hi_->InTrackingSet(key5)); | |
| 105 // Cache miss. Add to indexing set by kicking one entry out. | |
| 106 EXPECT_FALSE(hi_->ShouldIndex(key5)); | |
| 107 EXPECT_EQ(3u, hi_->indexing_set_size()); | |
| 108 EXPECT_EQ(4u, hi_->tracking_set_size()); | |
| 109 EXPECT_TRUE(hi_->ShouldIndex(key5)); | |
| 110 EXPECT_TRUE(hi_->InTrackingSet(key5)); | |
| 111 } | |
| 112 | |
| 113 TEST_F(SpdyHeaderIndexingTest, TestSetInit) { | |
| 114 hi_->CreateInitIndexingHeaders(); | |
| 115 EXPECT_EQ(100u, hi_->indexing_set_size()); | |
| 116 EXPECT_EQ(100u, hi_->tracking_set_size()); | |
| 117 EXPECT_TRUE(hi_->ShouldIndex(":status")); | |
| 118 EXPECT_TRUE(hi_->InTrackingSet(":status")); | |
| 119 EXPECT_FALSE(hi_->InTrackingSet("NotValid")); | |
| 120 EXPECT_FALSE(hi_->ShouldIndex("NotValid")); | |
| 121 } | |
| 122 | |
| 123 } // namespace test | |
| 124 | |
| 125 } // namespace net | |
| OLD | NEW |