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 |