| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2012 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/spdy_header_block.h" | |
| 6 | |
| 7 #include <memory> | |
| 8 #include <utility> | |
| 9 | |
| 10 #include "base/values.h" | |
| 11 #include "net/log/net_log_capture_mode.h" | |
| 12 #include "net/spdy/spdy_test_utils.h" | |
| 13 #include "testing/gmock/include/gmock/gmock.h" | |
| 14 #include "testing/gtest/include/gtest/gtest.h" | |
| 15 | |
| 16 using ::testing::ElementsAre; | |
| 17 | |
| 18 namespace net { | |
| 19 namespace test { | |
| 20 | |
| 21 class ValueProxyPeer { | |
| 22 public: | |
| 23 static SpdyStringPiece key(SpdyHeaderBlock::ValueProxy* p) { return p->key_; } | |
| 24 }; | |
| 25 | |
| 26 std::pair<SpdyStringPiece, SpdyStringPiece> Pair(SpdyStringPiece k, | |
| 27 SpdyStringPiece v) { | |
| 28 return std::make_pair(k, v); | |
| 29 } | |
| 30 | |
| 31 // This test verifies that SpdyHeaderBlock behaves correctly when empty. | |
| 32 TEST(SpdyHeaderBlockTest, EmptyBlock) { | |
| 33 SpdyHeaderBlock block; | |
| 34 EXPECT_TRUE(block.empty()); | |
| 35 EXPECT_EQ(0u, block.size()); | |
| 36 EXPECT_EQ(block.end(), block.find("foo")); | |
| 37 EXPECT_TRUE(block.end() == block.begin()); | |
| 38 | |
| 39 // Should have no effect. | |
| 40 block.erase("bar"); | |
| 41 } | |
| 42 | |
| 43 TEST(SpdyHeaderBlockTest, KeyMemoryReclaimedOnLookup) { | |
| 44 SpdyHeaderBlock block; | |
| 45 SpdyStringPiece copied_key1; | |
| 46 { | |
| 47 auto proxy1 = block["some key name"]; | |
| 48 copied_key1 = ValueProxyPeer::key(&proxy1); | |
| 49 } | |
| 50 SpdyStringPiece copied_key2; | |
| 51 { | |
| 52 auto proxy2 = block["some other key name"]; | |
| 53 copied_key2 = ValueProxyPeer::key(&proxy2); | |
| 54 } | |
| 55 // Because proxy1 was never used to modify the block, the memory used for the | |
| 56 // key could be reclaimed and used for the second call to operator[]. | |
| 57 // Therefore, we expect the pointers of the two SpdyStringPieces to be equal. | |
| 58 EXPECT_EQ(copied_key1.data(), copied_key2.data()); | |
| 59 | |
| 60 { | |
| 61 auto proxy1 = block["some key name"]; | |
| 62 block["some other key name"] = "some value"; | |
| 63 } | |
| 64 // Nothing should blow up when proxy1 is destructed, and we should be able to | |
| 65 // modify and access the SpdyHeaderBlock. | |
| 66 block["key"] = "value"; | |
| 67 EXPECT_EQ("value", block["key"]); | |
| 68 EXPECT_EQ("some value", block["some other key name"]); | |
| 69 EXPECT_TRUE(block.find("some key name") == block.end()); | |
| 70 } | |
| 71 | |
| 72 // This test verifies that headers can be set in a variety of ways. | |
| 73 TEST(SpdyHeaderBlockTest, AddHeaders) { | |
| 74 SpdyHeaderBlock block; | |
| 75 block["foo"] = SpdyString(300, 'x'); | |
| 76 block["bar"] = "baz"; | |
| 77 block["qux"] = "qux1"; | |
| 78 block["qux"] = "qux2"; | |
| 79 block.insert(std::make_pair("key", "value")); | |
| 80 | |
| 81 EXPECT_EQ(Pair("foo", SpdyString(300, 'x')), *block.find("foo")); | |
| 82 EXPECT_EQ("baz", block["bar"]); | |
| 83 SpdyString qux("qux"); | |
| 84 EXPECT_EQ("qux2", block[qux]); | |
| 85 ASSERT_NE(block.end(), block.find("key")); | |
| 86 EXPECT_EQ(Pair("key", "value"), *block.find("key")); | |
| 87 | |
| 88 block.erase("key"); | |
| 89 EXPECT_EQ(block.end(), block.find("key")); | |
| 90 } | |
| 91 | |
| 92 // This test verifies that SpdyHeaderBlock can be copied using Clone(). | |
| 93 TEST(SpdyHeaderBlockTest, CopyBlocks) { | |
| 94 SpdyHeaderBlock block1; | |
| 95 block1["foo"] = SpdyString(300, 'x'); | |
| 96 block1["bar"] = "baz"; | |
| 97 block1.insert(std::make_pair("qux", "qux1")); | |
| 98 | |
| 99 SpdyHeaderBlock block2 = block1.Clone(); | |
| 100 SpdyHeaderBlock block3(block1.Clone()); | |
| 101 | |
| 102 EXPECT_EQ(block1, block2); | |
| 103 EXPECT_EQ(block1, block3); | |
| 104 } | |
| 105 | |
| 106 TEST(SpdyHeaderBlockTest, ToNetLogParamAndBackAgain) { | |
| 107 SpdyHeaderBlock headers; | |
| 108 headers["A"] = "a"; | |
| 109 headers["B"] = "b"; | |
| 110 | |
| 111 std::unique_ptr<base::Value> event_param(SpdyHeaderBlockNetLogCallback( | |
| 112 &headers, NetLogCaptureMode::IncludeCookiesAndCredentials())); | |
| 113 | |
| 114 SpdyHeaderBlock headers2; | |
| 115 ASSERT_TRUE(SpdyHeaderBlockFromNetLogParam(event_param.get(), &headers2)); | |
| 116 EXPECT_EQ(headers, headers2); | |
| 117 } | |
| 118 | |
| 119 TEST(SpdyHeaderBlockTest, Equality) { | |
| 120 // Test equality and inequality operators. | |
| 121 SpdyHeaderBlock block1; | |
| 122 block1["foo"] = "bar"; | |
| 123 | |
| 124 SpdyHeaderBlock block2; | |
| 125 block2["foo"] = "bar"; | |
| 126 | |
| 127 SpdyHeaderBlock block3; | |
| 128 block3["baz"] = "qux"; | |
| 129 | |
| 130 EXPECT_EQ(block1, block2); | |
| 131 EXPECT_NE(block1, block3); | |
| 132 | |
| 133 block2["baz"] = "qux"; | |
| 134 EXPECT_NE(block1, block2); | |
| 135 } | |
| 136 | |
| 137 // Test that certain methods do not crash on moved-from instances. | |
| 138 TEST(SpdyHeaderBlockTest, MovedFromIsValid) { | |
| 139 SpdyHeaderBlock block1; | |
| 140 block1["foo"] = "bar"; | |
| 141 | |
| 142 SpdyHeaderBlock block2(std::move(block1)); | |
| 143 EXPECT_THAT(block2, ElementsAre(Pair("foo", "bar"))); | |
| 144 | |
| 145 block1["baz"] = "qux"; // NOLINT testing post-move behavior | |
| 146 | |
| 147 SpdyHeaderBlock block3(std::move(block1)); | |
| 148 | |
| 149 block1["foo"] = "bar"; // NOLINT testing post-move behavior | |
| 150 | |
| 151 SpdyHeaderBlock block4(std::move(block1)); | |
| 152 | |
| 153 block1.clear(); // NOLINT testing post-move behavior | |
| 154 EXPECT_TRUE(block1.empty()); | |
| 155 | |
| 156 block1["foo"] = "bar"; | |
| 157 EXPECT_THAT(block1, ElementsAre(Pair("foo", "bar"))); | |
| 158 } | |
| 159 | |
| 160 // This test verifies that headers can be appended to no matter how they were | |
| 161 // added originally. | |
| 162 TEST(SpdyHeaderBlockTest, AppendHeaders) { | |
| 163 SpdyHeaderBlock block; | |
| 164 block["foo"] = "foo"; | |
| 165 block.AppendValueOrAddHeader("foo", "bar"); | |
| 166 EXPECT_EQ(Pair("foo", SpdyString("foo\0bar", 7)), *block.find("foo")); | |
| 167 | |
| 168 block.insert(std::make_pair("foo", "baz")); | |
| 169 EXPECT_EQ("baz", block["foo"]); | |
| 170 EXPECT_EQ(Pair("foo", "baz"), *block.find("foo")); | |
| 171 | |
| 172 // Try all four methods of adding an entry. | |
| 173 block["cookie"] = "key1=value1"; | |
| 174 block.AppendValueOrAddHeader("h1", "h1v1"); | |
| 175 block.insert(std::make_pair("h2", "h2v1")); | |
| 176 | |
| 177 block.AppendValueOrAddHeader("h3", "h3v2"); | |
| 178 block.AppendValueOrAddHeader("h2", "h2v2"); | |
| 179 block.AppendValueOrAddHeader("h1", "h1v2"); | |
| 180 block.AppendValueOrAddHeader("cookie", "key2=value2"); | |
| 181 | |
| 182 block.AppendValueOrAddHeader("cookie", "key3=value3"); | |
| 183 block.AppendValueOrAddHeader("h1", "h1v3"); | |
| 184 block.AppendValueOrAddHeader("h2", "h2v3"); | |
| 185 block.AppendValueOrAddHeader("h3", "h3v3"); | |
| 186 block.AppendValueOrAddHeader("h4", "singleton"); | |
| 187 | |
| 188 EXPECT_EQ("key1=value1; key2=value2; key3=value3", block["cookie"]); | |
| 189 EXPECT_EQ("baz", block["foo"]); | |
| 190 EXPECT_EQ(SpdyString("h1v1\0h1v2\0h1v3", 14), block["h1"]); | |
| 191 EXPECT_EQ(SpdyString("h2v1\0h2v2\0h2v3", 14), block["h2"]); | |
| 192 EXPECT_EQ(SpdyString("h3v2\0h3v3", 9), block["h3"]); | |
| 193 EXPECT_EQ("singleton", block["h4"]); | |
| 194 } | |
| 195 | |
| 196 TEST(JoinTest, JoinEmpty) { | |
| 197 std::vector<SpdyStringPiece> empty; | |
| 198 SpdyStringPiece separator = ", "; | |
| 199 char buf[10] = ""; | |
| 200 size_t written = Join(buf, empty, separator); | |
| 201 EXPECT_EQ(0u, written); | |
| 202 } | |
| 203 | |
| 204 TEST(JoinTest, JoinOne) { | |
| 205 std::vector<SpdyStringPiece> v = {"one"}; | |
| 206 SpdyStringPiece separator = ", "; | |
| 207 char buf[15]; | |
| 208 size_t written = Join(buf, v, separator); | |
| 209 EXPECT_EQ(3u, written); | |
| 210 EXPECT_EQ("one", SpdyStringPiece(buf, written)); | |
| 211 } | |
| 212 | |
| 213 TEST(JoinTest, JoinMultiple) { | |
| 214 std::vector<SpdyStringPiece> v = {"one", "two", "three"}; | |
| 215 SpdyStringPiece separator = ", "; | |
| 216 char buf[15]; | |
| 217 size_t written = Join(buf, v, separator); | |
| 218 EXPECT_EQ(15u, written); | |
| 219 EXPECT_EQ("one, two, three", SpdyStringPiece(buf, written)); | |
| 220 } | |
| 221 | |
| 222 } // namespace test | |
| 223 } // namespace net | |
| OLD | NEW |