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/http/http_server_properties_impl.h"
6
7#include <string>
8#include <vector>
9
10#include "base/basictypes.h"
11#include "base/containers/hash_tables.h"
12#include "base/logging.h"
13#include "base/memory/scoped_ptr.h"
14#include "base/values.h"
15#include "net/base/host_port_pair.h"
16#include "testing/gtest/include/gtest/gtest.h"
17
18namespace base {
19class ListValue;
20}
21
22namespace net {
23
24const int kMaxSupportsSpdyServerHosts = 500;
25
26namespace {
27
28class HttpServerPropertiesImplTest : public testing::Test {
29 protected:
30  HttpServerPropertiesImpl impl_;
31};
32
33typedef HttpServerPropertiesImplTest SpdyServerPropertiesTest;
34
35TEST_F(SpdyServerPropertiesTest, Initialize) {
36  HostPortPair spdy_server_google("www.google.com", 443);
37  std::string spdy_server_g =
38      HttpServerPropertiesImpl::GetFlattenedSpdyServer(spdy_server_google);
39
40  HostPortPair spdy_server_docs("docs.google.com", 443);
41  std::string spdy_server_d =
42      HttpServerPropertiesImpl::GetFlattenedSpdyServer(spdy_server_docs);
43
44  // Check by initializing NULL spdy servers.
45  impl_.InitializeSpdyServers(NULL, true);
46  EXPECT_FALSE(impl_.SupportsSpdy(spdy_server_google));
47
48  // Check by initializing empty spdy servers.
49  std::vector<std::string> spdy_servers;
50  impl_.InitializeSpdyServers(&spdy_servers, true);
51  EXPECT_FALSE(impl_.SupportsSpdy(spdy_server_google));
52
53  // Check by initializing with www.google.com:443 spdy server.
54  std::vector<std::string> spdy_servers1;
55  spdy_servers1.push_back(spdy_server_g);
56  impl_.InitializeSpdyServers(&spdy_servers1, true);
57  EXPECT_TRUE(impl_.SupportsSpdy(spdy_server_google));
58
59  // Check by initializing with www.google.com:443 and docs.google.com:443 spdy
60  // servers.
61  std::vector<std::string> spdy_servers2;
62  spdy_servers2.push_back(spdy_server_g);
63  spdy_servers2.push_back(spdy_server_d);
64  impl_.InitializeSpdyServers(&spdy_servers2, true);
65
66  // Verify spdy_server_g and spdy_server_d are in the list in the same order.
67  base::ListValue spdy_server_list;
68  impl_.GetSpdyServerList(&spdy_server_list, kMaxSupportsSpdyServerHosts);
69  EXPECT_EQ(2U, spdy_server_list.GetSize());
70  std::string string_value_g;
71  ASSERT_TRUE(spdy_server_list.GetString(0, &string_value_g));
72  ASSERT_EQ(spdy_server_g, string_value_g);
73  std::string string_value_d;
74  ASSERT_TRUE(spdy_server_list.GetString(1, &string_value_d));
75  ASSERT_EQ(spdy_server_d, string_value_d);
76  EXPECT_TRUE(impl_.SupportsSpdy(spdy_server_google));
77  EXPECT_TRUE(impl_.SupportsSpdy(spdy_server_docs));
78}
79
80TEST_F(SpdyServerPropertiesTest, SupportsSpdyTest) {
81  HostPortPair spdy_server_empty(std::string(), 443);
82  EXPECT_FALSE(impl_.SupportsSpdy(spdy_server_empty));
83
84  // Add www.google.com:443 as supporting SPDY.
85  HostPortPair spdy_server_google("www.google.com", 443);
86  impl_.SetSupportsSpdy(spdy_server_google, true);
87  EXPECT_TRUE(impl_.SupportsSpdy(spdy_server_google));
88
89  // Add mail.google.com:443 as not supporting SPDY.
90  HostPortPair spdy_server_mail("mail.google.com", 443);
91  EXPECT_FALSE(impl_.SupportsSpdy(spdy_server_mail));
92
93  // Add docs.google.com:443 as supporting SPDY.
94  HostPortPair spdy_server_docs("docs.google.com", 443);
95  impl_.SetSupportsSpdy(spdy_server_docs, true);
96  EXPECT_TRUE(impl_.SupportsSpdy(spdy_server_docs));
97
98  // Verify all the entries are the same after additions.
99  EXPECT_TRUE(impl_.SupportsSpdy(spdy_server_google));
100  EXPECT_FALSE(impl_.SupportsSpdy(spdy_server_mail));
101  EXPECT_TRUE(impl_.SupportsSpdy(spdy_server_docs));
102}
103
104TEST_F(SpdyServerPropertiesTest, SetSupportsSpdy) {
105  HostPortPair spdy_server_empty(std::string(), 443);
106  impl_.SetSupportsSpdy(spdy_server_empty, true);
107  EXPECT_FALSE(impl_.SupportsSpdy(spdy_server_empty));
108
109  // Add www.google.com:443 as supporting SPDY.
110  HostPortPair spdy_server_google("www.google.com", 443);
111  EXPECT_FALSE(impl_.SupportsSpdy(spdy_server_google));
112  impl_.SetSupportsSpdy(spdy_server_google, true);
113  EXPECT_TRUE(impl_.SupportsSpdy(spdy_server_google));
114
115  // Make www.google.com:443 as not supporting SPDY.
116  impl_.SetSupportsSpdy(spdy_server_google, false);
117  EXPECT_FALSE(impl_.SupportsSpdy(spdy_server_google));
118
119  // Add mail.google.com:443 as supporting SPDY.
120  HostPortPair spdy_server_mail("mail.google.com", 443);
121  EXPECT_FALSE(impl_.SupportsSpdy(spdy_server_mail));
122  impl_.SetSupportsSpdy(spdy_server_mail, true);
123  EXPECT_TRUE(impl_.SupportsSpdy(spdy_server_mail));
124  EXPECT_FALSE(impl_.SupportsSpdy(spdy_server_google));
125}
126
127TEST_F(SpdyServerPropertiesTest, Clear) {
128  // Add www.google.com:443 and mail.google.com:443 as supporting SPDY.
129  HostPortPair spdy_server_google("www.google.com", 443);
130  impl_.SetSupportsSpdy(spdy_server_google, true);
131  HostPortPair spdy_server_mail("mail.google.com", 443);
132  impl_.SetSupportsSpdy(spdy_server_mail, true);
133
134  EXPECT_TRUE(impl_.SupportsSpdy(spdy_server_google));
135  EXPECT_TRUE(impl_.SupportsSpdy(spdy_server_mail));
136
137  impl_.Clear();
138  EXPECT_FALSE(impl_.SupportsSpdy(spdy_server_google));
139  EXPECT_FALSE(impl_.SupportsSpdy(spdy_server_mail));
140}
141
142TEST_F(SpdyServerPropertiesTest, GetSpdyServerList) {
143  base::ListValue spdy_server_list;
144
145  // Check there are no spdy_servers.
146  impl_.GetSpdyServerList(&spdy_server_list, kMaxSupportsSpdyServerHosts);
147  EXPECT_EQ(0U, spdy_server_list.GetSize());
148
149  // Check empty server is not added.
150  HostPortPair spdy_server_empty(std::string(), 443);
151  impl_.SetSupportsSpdy(spdy_server_empty, true);
152  impl_.GetSpdyServerList(&spdy_server_list, kMaxSupportsSpdyServerHosts);
153  EXPECT_EQ(0U, spdy_server_list.GetSize());
154
155  std::string string_value_g;
156  std::string string_value_m;
157  HostPortPair spdy_server_google("www.google.com", 443);
158  std::string spdy_server_g =
159      HttpServerPropertiesImpl::GetFlattenedSpdyServer(spdy_server_google);
160  HostPortPair spdy_server_mail("mail.google.com", 443);
161  std::string spdy_server_m =
162      HttpServerPropertiesImpl::GetFlattenedSpdyServer(spdy_server_mail);
163
164  // Add www.google.com:443 as not supporting SPDY.
165  impl_.SetSupportsSpdy(spdy_server_google, false);
166  impl_.GetSpdyServerList(&spdy_server_list, kMaxSupportsSpdyServerHosts);
167  EXPECT_EQ(0U, spdy_server_list.GetSize());
168
169  // Add www.google.com:443 as supporting SPDY.
170  impl_.SetSupportsSpdy(spdy_server_google, true);
171  impl_.GetSpdyServerList(&spdy_server_list, kMaxSupportsSpdyServerHosts);
172  ASSERT_EQ(1U, spdy_server_list.GetSize());
173  ASSERT_TRUE(spdy_server_list.GetString(0, &string_value_g));
174  ASSERT_EQ(spdy_server_g, string_value_g);
175
176  // Add mail.google.com:443 as not supporting SPDY.
177  impl_.SetSupportsSpdy(spdy_server_mail, false);
178  impl_.GetSpdyServerList(&spdy_server_list, kMaxSupportsSpdyServerHosts);
179  ASSERT_EQ(1U, spdy_server_list.GetSize());
180  ASSERT_TRUE(spdy_server_list.GetString(0, &string_value_g));
181  ASSERT_EQ(spdy_server_g, string_value_g);
182
183  // Add mail.google.com:443 as supporting SPDY.
184  impl_.SetSupportsSpdy(spdy_server_mail, true);
185  impl_.GetSpdyServerList(&spdy_server_list, kMaxSupportsSpdyServerHosts);
186  ASSERT_EQ(2U, spdy_server_list.GetSize());
187
188  // Verify www.google.com:443 and mail.google.com:443 are in the list.
189  ASSERT_TRUE(spdy_server_list.GetString(0, &string_value_m));
190  ASSERT_EQ(spdy_server_m, string_value_m);
191  ASSERT_TRUE(spdy_server_list.GetString(1, &string_value_g));
192  ASSERT_EQ(spdy_server_g, string_value_g);
193
194  // Request for only one server and verify that we get only one server.
195  impl_.GetSpdyServerList(&spdy_server_list, 1);
196  ASSERT_EQ(1U, spdy_server_list.GetSize());
197  ASSERT_TRUE(spdy_server_list.GetString(0, &string_value_m));
198  ASSERT_EQ(spdy_server_m, string_value_m);
199}
200
201TEST_F(SpdyServerPropertiesTest, MRUOfGetSpdyServerList) {
202  base::ListValue spdy_server_list;
203
204  std::string string_value_g;
205  std::string string_value_m;
206  HostPortPair spdy_server_google("www.google.com", 443);
207  std::string spdy_server_g =
208      HttpServerPropertiesImpl::GetFlattenedSpdyServer(spdy_server_google);
209  HostPortPair spdy_server_mail("mail.google.com", 443);
210  std::string spdy_server_m =
211      HttpServerPropertiesImpl::GetFlattenedSpdyServer(spdy_server_mail);
212
213  // Add www.google.com:443 as supporting SPDY.
214  impl_.SetSupportsSpdy(spdy_server_google, true);
215  impl_.GetSpdyServerList(&spdy_server_list, kMaxSupportsSpdyServerHosts);
216  ASSERT_EQ(1U, spdy_server_list.GetSize());
217  ASSERT_TRUE(spdy_server_list.GetString(0, &string_value_g));
218  ASSERT_EQ(spdy_server_g, string_value_g);
219
220  // Add mail.google.com:443 as supporting SPDY. Verify mail.google.com:443 and
221  // www.google.com:443 are in the list.
222  impl_.SetSupportsSpdy(spdy_server_mail, true);
223  impl_.GetSpdyServerList(&spdy_server_list, kMaxSupportsSpdyServerHosts);
224  ASSERT_EQ(2U, spdy_server_list.GetSize());
225  ASSERT_TRUE(spdy_server_list.GetString(0, &string_value_m));
226  ASSERT_EQ(spdy_server_m, string_value_m);
227  ASSERT_TRUE(spdy_server_list.GetString(1, &string_value_g));
228  ASSERT_EQ(spdy_server_g, string_value_g);
229
230  // Get www.google.com:443 should reorder SpdyServerHostPortMap. Verify that it
231  // is www.google.com:443 is the MRU server.
232  EXPECT_TRUE(impl_.SupportsSpdy(spdy_server_google));
233  impl_.GetSpdyServerList(&spdy_server_list, kMaxSupportsSpdyServerHosts);
234  ASSERT_EQ(2U, spdy_server_list.GetSize());
235  ASSERT_TRUE(spdy_server_list.GetString(0, &string_value_g));
236  ASSERT_EQ(spdy_server_g, string_value_g);
237  ASSERT_TRUE(spdy_server_list.GetString(1, &string_value_m));
238  ASSERT_EQ(spdy_server_m, string_value_m);
239}
240
241typedef HttpServerPropertiesImplTest AlternateProtocolServerPropertiesTest;
242
243TEST_F(AlternateProtocolServerPropertiesTest, Basic) {
244  HostPortPair test_host_port_pair("foo", 80);
245  EXPECT_FALSE(impl_.HasAlternateProtocol(test_host_port_pair));
246  impl_.SetAlternateProtocol(test_host_port_pair, 443, NPN_SPDY_3, 1);
247  ASSERT_TRUE(impl_.HasAlternateProtocol(test_host_port_pair));
248  const AlternateProtocolInfo alternate =
249      impl_.GetAlternateProtocol(test_host_port_pair);
250  EXPECT_EQ(443, alternate.port);
251  EXPECT_EQ(NPN_SPDY_3, alternate.protocol);
252
253  impl_.Clear();
254  EXPECT_FALSE(impl_.HasAlternateProtocol(test_host_port_pair));
255}
256
257TEST_F(AlternateProtocolServerPropertiesTest, DefaultProbabilityExcluded) {
258  HostPortPair test_host_port_pair("foo", 80);
259  impl_.SetAlternateProtocol(test_host_port_pair, 443, NPN_SPDY_3, .99);
260
261  EXPECT_FALSE(impl_.HasAlternateProtocol(test_host_port_pair));
262}
263
264TEST_F(AlternateProtocolServerPropertiesTest, Probability) {
265  impl_.SetAlternateProtocolProbabilityThreshold(.25);
266
267  HostPortPair test_host_port_pair("foo", 80);
268  impl_.SetAlternateProtocol(test_host_port_pair, 443, NPN_SPDY_3, .5);
269
270  ASSERT_TRUE(impl_.HasAlternateProtocol(test_host_port_pair));
271  const AlternateProtocolInfo alternate =
272      impl_.GetAlternateProtocol(test_host_port_pair);
273  EXPECT_EQ(443, alternate.port);
274  EXPECT_EQ(NPN_SPDY_3, alternate.protocol);
275  EXPECT_EQ(.5, alternate.probability);
276}
277
278TEST_F(AlternateProtocolServerPropertiesTest, ProbabilityExcluded) {
279  impl_.SetAlternateProtocolProbabilityThreshold(.75);
280
281  HostPortPair test_host_port_pair("foo", 80);
282
283  impl_.SetAlternateProtocol(test_host_port_pair, 443, NPN_SPDY_3, .5);
284  EXPECT_FALSE(impl_.HasAlternateProtocol(test_host_port_pair));
285}
286
287TEST_F(AlternateProtocolServerPropertiesTest, Initialize) {
288  HostPortPair test_host_port_pair1("foo1", 80);
289  impl_.SetBrokenAlternateProtocol(test_host_port_pair1);
290  HostPortPair test_host_port_pair2("foo2", 80);
291  impl_.SetAlternateProtocol(test_host_port_pair2, 443, NPN_SPDY_3, 1);
292
293  AlternateProtocolMap alternate_protocol_map(
294      AlternateProtocolMap::NO_AUTO_EVICT);
295  AlternateProtocolInfo port_alternate_protocol_pair(123, NPN_SPDY_3, 1);
296  alternate_protocol_map.Put(test_host_port_pair2,
297                             port_alternate_protocol_pair);
298  HostPortPair test_host_port_pair3("foo3", 80);
299  port_alternate_protocol_pair.port = 1234;
300  alternate_protocol_map.Put(test_host_port_pair3,
301                             port_alternate_protocol_pair);
302  impl_.InitializeAlternateProtocolServers(&alternate_protocol_map);
303
304  // Verify test_host_port_pair3 is the MRU server.
305  const net::AlternateProtocolMap& map = impl_.alternate_protocol_map();
306  net::AlternateProtocolMap::const_iterator it = map.begin();
307  it = map.begin();
308  EXPECT_TRUE(it->first.Equals(test_host_port_pair3));
309  EXPECT_EQ(1234, it->second.port);
310  EXPECT_EQ(NPN_SPDY_3, it->second.protocol);
311
312  ASSERT_TRUE(impl_.HasAlternateProtocol(test_host_port_pair1));
313  ASSERT_TRUE(impl_.HasAlternateProtocol(test_host_port_pair2));
314  port_alternate_protocol_pair =
315      impl_.GetAlternateProtocol(test_host_port_pair1);
316  EXPECT_EQ(ALTERNATE_PROTOCOL_BROKEN, port_alternate_protocol_pair.protocol);
317  port_alternate_protocol_pair =
318      impl_.GetAlternateProtocol(test_host_port_pair2);
319  EXPECT_EQ(123, port_alternate_protocol_pair.port);
320  EXPECT_EQ(NPN_SPDY_3, port_alternate_protocol_pair.protocol);
321}
322
323TEST_F(AlternateProtocolServerPropertiesTest, MRUOfHasAlternateProtocol) {
324  HostPortPair test_host_port_pair1("foo1", 80);
325  impl_.SetAlternateProtocol(test_host_port_pair1, 443, NPN_SPDY_3, 1);
326  HostPortPair test_host_port_pair2("foo2", 80);
327  impl_.SetAlternateProtocol(test_host_port_pair2, 1234, NPN_SPDY_3, 1);
328
329  const net::AlternateProtocolMap& map = impl_.alternate_protocol_map();
330  net::AlternateProtocolMap::const_iterator it = map.begin();
331  EXPECT_TRUE(it->first.Equals(test_host_port_pair2));
332  EXPECT_EQ(1234, it->second.port);
333  EXPECT_EQ(NPN_SPDY_3, it->second.protocol);
334
335  // HasAlternateProtocol should reoder the AlternateProtocol map.
336  ASSERT_TRUE(impl_.HasAlternateProtocol(test_host_port_pair1));
337  it = map.begin();
338  EXPECT_TRUE(it->first.Equals(test_host_port_pair1));
339  EXPECT_EQ(443, it->second.port);
340  EXPECT_EQ(NPN_SPDY_3, it->second.protocol);
341}
342
343TEST_F(AlternateProtocolServerPropertiesTest, MRUOfGetAlternateProtocol) {
344  HostPortPair test_host_port_pair1("foo1", 80);
345  impl_.SetAlternateProtocol(test_host_port_pair1, 443, NPN_SPDY_3, 1);
346  HostPortPair test_host_port_pair2("foo2", 80);
347  impl_.SetAlternateProtocol(test_host_port_pair2, 1234, NPN_SPDY_3, 1);
348
349  const net::AlternateProtocolMap& map = impl_.alternate_protocol_map();
350  net::AlternateProtocolMap::const_iterator it = map.begin();
351  EXPECT_TRUE(it->first.Equals(test_host_port_pair2));
352  EXPECT_EQ(1234, it->second.port);
353  EXPECT_EQ(NPN_SPDY_3, it->second.protocol);
354
355  // GetAlternateProtocol should reoder the AlternateProtocol map.
356  AlternateProtocolInfo alternate =
357      impl_.GetAlternateProtocol(test_host_port_pair1);
358  EXPECT_EQ(443, alternate.port);
359  EXPECT_EQ(NPN_SPDY_3, alternate.protocol);
360  it = map.begin();
361  EXPECT_TRUE(it->first.Equals(test_host_port_pair1));
362  EXPECT_EQ(443, it->second.port);
363  EXPECT_EQ(NPN_SPDY_3, it->second.protocol);
364}
365
366TEST_F(AlternateProtocolServerPropertiesTest, SetBroken) {
367  HostPortPair test_host_port_pair("foo", 80);
368  impl_.SetBrokenAlternateProtocol(test_host_port_pair);
369  ASSERT_TRUE(impl_.HasAlternateProtocol(test_host_port_pair));
370  AlternateProtocolInfo alternate =
371      impl_.GetAlternateProtocol(test_host_port_pair);
372  EXPECT_EQ(ALTERNATE_PROTOCOL_BROKEN, alternate.protocol);
373
374  impl_.SetAlternateProtocol(
375      test_host_port_pair,
376      1234,
377      NPN_SPDY_3,
378      1);
379  alternate = impl_.GetAlternateProtocol(test_host_port_pair);
380  EXPECT_EQ(ALTERNATE_PROTOCOL_BROKEN, alternate.protocol)
381      << "Second attempt should be ignored.";
382}
383
384TEST_F(AlternateProtocolServerPropertiesTest, ClearBroken) {
385  HostPortPair test_host_port_pair("foo", 80);
386  impl_.SetBrokenAlternateProtocol(test_host_port_pair);
387  ASSERT_TRUE(impl_.HasAlternateProtocol(test_host_port_pair));
388  AlternateProtocolInfo alternate =
389      impl_.GetAlternateProtocol(test_host_port_pair);
390  EXPECT_EQ(ALTERNATE_PROTOCOL_BROKEN, alternate.protocol);
391  impl_.ClearAlternateProtocol(test_host_port_pair);
392  EXPECT_FALSE(impl_.HasAlternateProtocol(test_host_port_pair));
393}
394
395TEST_F(AlternateProtocolServerPropertiesTest, Forced) {
396  // Test forced alternate protocols.
397
398  AlternateProtocolInfo default_protocol(1234, NPN_SPDY_3, 1);
399  HttpServerPropertiesImpl::ForceAlternateProtocol(default_protocol);
400
401  // Verify the forced protocol.
402  HostPortPair test_host_port_pair("foo", 80);
403  EXPECT_TRUE(impl_.HasAlternateProtocol(test_host_port_pair));
404  AlternateProtocolInfo alternate =
405      impl_.GetAlternateProtocol(test_host_port_pair);
406  EXPECT_EQ(default_protocol.port, alternate.port);
407  EXPECT_EQ(default_protocol.protocol, alternate.protocol);
408
409  // Verify the real protocol overrides the forced protocol.
410  impl_.SetAlternateProtocol(test_host_port_pair, 443, NPN_SPDY_3, 1);
411  ASSERT_TRUE(impl_.HasAlternateProtocol(test_host_port_pair));
412  alternate = impl_.GetAlternateProtocol(test_host_port_pair);
413  EXPECT_EQ(443, alternate.port);
414  EXPECT_EQ(NPN_SPDY_3, alternate.protocol);
415
416  // Turn off the static, forced alternate protocol so that tests don't
417  // have this state.
418  HttpServerPropertiesImpl::DisableForcedAlternateProtocol();
419
420  // Verify the forced protocol is off.
421  HostPortPair test_host_port_pair2("bar", 80);
422  EXPECT_FALSE(impl_.HasAlternateProtocol(test_host_port_pair2));
423}
424
425TEST_F(AlternateProtocolServerPropertiesTest, Canonical) {
426  HostPortPair test_host_port_pair("foo.c.youtube.com", 80);
427  EXPECT_FALSE(impl_.HasAlternateProtocol(test_host_port_pair));
428
429  HostPortPair canonical_port_pair("bar.c.youtube.com", 80);
430  EXPECT_FALSE(impl_.HasAlternateProtocol(canonical_port_pair));
431
432  AlternateProtocolInfo canonical_protocol(1234, QUIC, 1);
433
434  impl_.SetAlternateProtocol(canonical_port_pair,
435                             canonical_protocol.port,
436                             canonical_protocol.protocol,
437                             1);
438  // Verify the forced protocol.
439  ASSERT_TRUE(impl_.HasAlternateProtocol(test_host_port_pair));
440  AlternateProtocolInfo alternate =
441      impl_.GetAlternateProtocol(test_host_port_pair);
442  EXPECT_EQ(canonical_protocol.port, alternate.port);
443  EXPECT_EQ(canonical_protocol.protocol, alternate.protocol);
444
445  // Verify the canonical suffix.
446  EXPECT_EQ(".c.youtube.com", impl_.GetCanonicalSuffix(test_host_port_pair));
447  EXPECT_EQ(".c.youtube.com", impl_.GetCanonicalSuffix(canonical_port_pair));
448}
449
450TEST_F(AlternateProtocolServerPropertiesTest, ClearCanonical) {
451  HostPortPair test_host_port_pair("foo.c.youtube.com", 80);
452  HostPortPair canonical_port_pair("bar.c.youtube.com", 80);
453
454  AlternateProtocolInfo canonical_protocol(1234, QUIC, 1);
455
456  impl_.SetAlternateProtocol(canonical_port_pair,
457                             canonical_protocol.port,
458                             canonical_protocol.protocol,
459                             canonical_protocol.probability);
460
461  impl_.ClearAlternateProtocol(canonical_port_pair);
462  EXPECT_FALSE(impl_.HasAlternateProtocol(test_host_port_pair));
463}
464
465TEST_F(AlternateProtocolServerPropertiesTest, CanonicalBroken) {
466  HostPortPair test_host_port_pair("foo.c.youtube.com", 80);
467  HostPortPair canonical_port_pair("bar.c.youtube.com", 80);
468
469  AlternateProtocolInfo canonical_protocol(1234, QUIC, 1);
470
471  impl_.SetAlternateProtocol(canonical_port_pair,
472                             canonical_protocol.port,
473                             canonical_protocol.protocol,
474                             canonical_protocol.probability);
475
476  impl_.SetBrokenAlternateProtocol(canonical_port_pair);
477  EXPECT_FALSE(impl_.HasAlternateProtocol(test_host_port_pair));
478}
479
480TEST_F(AlternateProtocolServerPropertiesTest, ClearWithCanonical) {
481  HostPortPair test_host_port_pair("foo.c.youtube.com", 80);
482  HostPortPair canonical_port_pair("bar.c.youtube.com", 80);
483
484  AlternateProtocolInfo canonical_protocol(1234, QUIC, 1);
485
486  impl_.SetAlternateProtocol(canonical_port_pair,
487                             canonical_protocol.port,
488                             canonical_protocol.protocol,
489                             canonical_protocol.probability);
490
491  impl_.Clear();
492  EXPECT_FALSE(impl_.HasAlternateProtocol(test_host_port_pair));
493}
494
495typedef HttpServerPropertiesImplTest SpdySettingsServerPropertiesTest;
496
497TEST_F(SpdySettingsServerPropertiesTest, Initialize) {
498  HostPortPair spdy_server_google("www.google.com", 443);
499
500  // Check by initializing empty spdy settings.
501  SpdySettingsMap spdy_settings_map(SpdySettingsMap::NO_AUTO_EVICT);
502  impl_.InitializeSpdySettingsServers(&spdy_settings_map);
503  EXPECT_TRUE(impl_.GetSpdySettings(spdy_server_google).empty());
504
505  // Check by initializing with www.google.com:443 spdy server settings.
506  SettingsMap settings_map;
507  const SpdySettingsIds id = SETTINGS_UPLOAD_BANDWIDTH;
508  const SpdySettingsFlags flags = SETTINGS_FLAG_PERSISTED;
509  const uint32 value = 31337;
510  SettingsFlagsAndValue flags_and_value(flags, value);
511  settings_map[id] = flags_and_value;
512  spdy_settings_map.Put(spdy_server_google, settings_map);
513  impl_.InitializeSpdySettingsServers(&spdy_settings_map);
514
515  const SettingsMap& settings_map2 = impl_.GetSpdySettings(spdy_server_google);
516  ASSERT_EQ(1U, settings_map2.size());
517  SettingsMap::const_iterator it = settings_map2.find(id);
518  EXPECT_TRUE(it != settings_map2.end());
519  SettingsFlagsAndValue flags_and_value2 = it->second;
520  EXPECT_EQ(flags, flags_and_value2.first);
521  EXPECT_EQ(value, flags_and_value2.second);
522}
523
524TEST_F(SpdySettingsServerPropertiesTest, SetSpdySetting) {
525  HostPortPair spdy_server_empty(std::string(), 443);
526  const SettingsMap& settings_map0 = impl_.GetSpdySettings(spdy_server_empty);
527  EXPECT_EQ(0U, settings_map0.size());  // Returns kEmptySettingsMap.
528
529  // Add www.google.com:443 as persisting.
530  HostPortPair spdy_server_google("www.google.com", 443);
531  const SpdySettingsIds id1 = SETTINGS_UPLOAD_BANDWIDTH;
532  const SpdySettingsFlags flags1 = SETTINGS_FLAG_PLEASE_PERSIST;
533  const uint32 value1 = 31337;
534  EXPECT_TRUE(impl_.SetSpdySetting(spdy_server_google, id1, flags1, value1));
535  // Check the values.
536  const SettingsMap& settings_map1_ret =
537      impl_.GetSpdySettings(spdy_server_google);
538  ASSERT_EQ(1U, settings_map1_ret.size());
539  SettingsMap::const_iterator it1_ret = settings_map1_ret.find(id1);
540  EXPECT_TRUE(it1_ret != settings_map1_ret.end());
541  SettingsFlagsAndValue flags_and_value1_ret = it1_ret->second;
542  EXPECT_EQ(SETTINGS_FLAG_PERSISTED, flags_and_value1_ret.first);
543  EXPECT_EQ(value1, flags_and_value1_ret.second);
544
545  // Add mail.google.com:443 as not persisting.
546  HostPortPair spdy_server_mail("mail.google.com", 443);
547  const SpdySettingsIds id2 = SETTINGS_DOWNLOAD_BANDWIDTH;
548  const SpdySettingsFlags flags2 = SETTINGS_FLAG_NONE;
549  const uint32 value2 = 62667;
550  EXPECT_FALSE(impl_.SetSpdySetting(spdy_server_mail, id2, flags2, value2));
551  const SettingsMap& settings_map2_ret =
552      impl_.GetSpdySettings(spdy_server_mail);
553  EXPECT_EQ(0U, settings_map2_ret.size());  // Returns kEmptySettingsMap.
554
555  // Add docs.google.com:443 as persisting
556  HostPortPair spdy_server_docs("docs.google.com", 443);
557  const SpdySettingsIds id3 = SETTINGS_ROUND_TRIP_TIME;
558  const SpdySettingsFlags flags3 = SETTINGS_FLAG_PLEASE_PERSIST;
559  const uint32 value3 = 93997;
560  SettingsFlagsAndValue flags_and_value3(flags3, value3);
561  EXPECT_TRUE(impl_.SetSpdySetting(spdy_server_docs, id3, flags3, value3));
562  // Check the values.
563  const SettingsMap& settings_map3_ret =
564      impl_.GetSpdySettings(spdy_server_docs);
565  ASSERT_EQ(1U, settings_map3_ret.size());
566  SettingsMap::const_iterator it3_ret = settings_map3_ret.find(id3);
567  EXPECT_TRUE(it3_ret != settings_map3_ret.end());
568  SettingsFlagsAndValue flags_and_value3_ret = it3_ret->second;
569  EXPECT_EQ(SETTINGS_FLAG_PERSISTED, flags_and_value3_ret.first);
570  EXPECT_EQ(value3, flags_and_value3_ret.second);
571
572  // Check data for www.google.com:443 (id1).
573  const SettingsMap& settings_map4_ret =
574      impl_.GetSpdySettings(spdy_server_google);
575  ASSERT_EQ(1U, settings_map4_ret.size());
576  SettingsMap::const_iterator it4_ret = settings_map4_ret.find(id1);
577  EXPECT_TRUE(it4_ret != settings_map4_ret.end());
578  SettingsFlagsAndValue flags_and_value4_ret = it4_ret->second;
579  EXPECT_EQ(SETTINGS_FLAG_PERSISTED, flags_and_value4_ret.first);
580  EXPECT_EQ(value1, flags_and_value1_ret.second);
581
582  // Clear www.google.com:443 as persisting.
583  impl_.ClearSpdySettings(spdy_server_google);
584  // Check the values.
585  const SettingsMap& settings_map5_ret =
586      impl_.GetSpdySettings(spdy_server_google);
587  ASSERT_EQ(0U, settings_map5_ret.size());
588
589  // Clear all settings.
590  ASSERT_GT(impl_.spdy_settings_map().size(), 0U);
591  impl_.ClearAllSpdySettings();
592  ASSERT_EQ(0U, impl_.spdy_settings_map().size());
593}
594
595TEST_F(SpdySettingsServerPropertiesTest, Clear) {
596  // Add www.google.com:443 as persisting.
597  HostPortPair spdy_server_google("www.google.com", 443);
598  const SpdySettingsIds id1 = SETTINGS_UPLOAD_BANDWIDTH;
599  const SpdySettingsFlags flags1 = SETTINGS_FLAG_PLEASE_PERSIST;
600  const uint32 value1 = 31337;
601  EXPECT_TRUE(impl_.SetSpdySetting(spdy_server_google, id1, flags1, value1));
602  // Check the values.
603  const SettingsMap& settings_map1_ret =
604      impl_.GetSpdySettings(spdy_server_google);
605  ASSERT_EQ(1U, settings_map1_ret.size());
606  SettingsMap::const_iterator it1_ret = settings_map1_ret.find(id1);
607  EXPECT_TRUE(it1_ret != settings_map1_ret.end());
608  SettingsFlagsAndValue flags_and_value1_ret = it1_ret->second;
609  EXPECT_EQ(SETTINGS_FLAG_PERSISTED, flags_and_value1_ret.first);
610  EXPECT_EQ(value1, flags_and_value1_ret.second);
611
612  // Add docs.google.com:443 as persisting
613  HostPortPair spdy_server_docs("docs.google.com", 443);
614  const SpdySettingsIds id3 = SETTINGS_ROUND_TRIP_TIME;
615  const SpdySettingsFlags flags3 = SETTINGS_FLAG_PLEASE_PERSIST;
616  const uint32 value3 = 93997;
617  EXPECT_TRUE(impl_.SetSpdySetting(spdy_server_docs, id3, flags3, value3));
618  // Check the values.
619  const SettingsMap& settings_map3_ret =
620      impl_.GetSpdySettings(spdy_server_docs);
621  ASSERT_EQ(1U, settings_map3_ret.size());
622  SettingsMap::const_iterator it3_ret = settings_map3_ret.find(id3);
623  EXPECT_TRUE(it3_ret != settings_map3_ret.end());
624  SettingsFlagsAndValue flags_and_value3_ret = it3_ret->second;
625  EXPECT_EQ(SETTINGS_FLAG_PERSISTED, flags_and_value3_ret.first);
626  EXPECT_EQ(value3, flags_and_value3_ret.second);
627
628  impl_.Clear();
629  EXPECT_EQ(0U, impl_.GetSpdySettings(spdy_server_google).size());
630  EXPECT_EQ(0U, impl_.GetSpdySettings(spdy_server_docs).size());
631}
632
633TEST_F(SpdySettingsServerPropertiesTest, MRUOfGetSpdySettings) {
634  // Add www.google.com:443 as persisting.
635  HostPortPair spdy_server_google("www.google.com", 443);
636  const SpdySettingsIds id1 = SETTINGS_UPLOAD_BANDWIDTH;
637  const SpdySettingsFlags flags1 = SETTINGS_FLAG_PLEASE_PERSIST;
638  const uint32 value1 = 31337;
639  EXPECT_TRUE(impl_.SetSpdySetting(spdy_server_google, id1, flags1, value1));
640
641  // Add docs.google.com:443 as persisting
642  HostPortPair spdy_server_docs("docs.google.com", 443);
643  const SpdySettingsIds id2 = SETTINGS_ROUND_TRIP_TIME;
644  const SpdySettingsFlags flags2 = SETTINGS_FLAG_PLEASE_PERSIST;
645  const uint32 value2 = 93997;
646  EXPECT_TRUE(impl_.SetSpdySetting(spdy_server_docs, id2, flags2, value2));
647
648  // Verify the first element is docs.google.com:443.
649  const net::SpdySettingsMap& map = impl_.spdy_settings_map();
650  net::SpdySettingsMap::const_iterator it = map.begin();
651  EXPECT_TRUE(it->first.Equals(spdy_server_docs));
652  const SettingsMap& settings_map2_ret = it->second;
653  ASSERT_EQ(1U, settings_map2_ret.size());
654  SettingsMap::const_iterator it2_ret = settings_map2_ret.find(id2);
655  EXPECT_TRUE(it2_ret != settings_map2_ret.end());
656  SettingsFlagsAndValue flags_and_value2_ret = it2_ret->second;
657  EXPECT_EQ(SETTINGS_FLAG_PERSISTED, flags_and_value2_ret.first);
658  EXPECT_EQ(value2, flags_and_value2_ret.second);
659
660  // GetSpdySettings should reoder the SpdySettingsMap.
661  const SettingsMap& settings_map1_ret =
662      impl_.GetSpdySettings(spdy_server_google);
663  ASSERT_EQ(1U, settings_map1_ret.size());
664  SettingsMap::const_iterator it1_ret = settings_map1_ret.find(id1);
665  EXPECT_TRUE(it1_ret != settings_map1_ret.end());
666  SettingsFlagsAndValue flags_and_value1_ret = it1_ret->second;
667  EXPECT_EQ(SETTINGS_FLAG_PERSISTED, flags_and_value1_ret.first);
668  EXPECT_EQ(value1, flags_and_value1_ret.second);
669
670  // Check the first entry is spdy_server_google by accessing it via iterator.
671  it = map.begin();
672  EXPECT_TRUE(it->first.Equals(spdy_server_google));
673  const SettingsMap& settings_map1_it_ret = it->second;
674  ASSERT_EQ(1U, settings_map1_it_ret.size());
675  it1_ret = settings_map1_it_ret.find(id1);
676  EXPECT_TRUE(it1_ret != settings_map1_it_ret.end());
677  flags_and_value1_ret = it1_ret->second;
678  EXPECT_EQ(SETTINGS_FLAG_PERSISTED, flags_and_value1_ret.first);
679  EXPECT_EQ(value1, flags_and_value1_ret.second);
680}
681
682typedef HttpServerPropertiesImplTest SupportsQuicServerPropertiesTest;
683
684TEST_F(SupportsQuicServerPropertiesTest, Initialize) {
685  HostPortPair quic_server_google("www.google.com", 443);
686
687  // Check by initializing empty SupportsQuic.
688  SupportsQuicMap supports_quic_map;
689  impl_.InitializeSupportsQuic(&supports_quic_map);
690  SupportsQuic supports_quic = impl_.GetSupportsQuic(quic_server_google);
691  EXPECT_FALSE(supports_quic.used_quic);
692  EXPECT_EQ("", supports_quic.address);
693
694  // Check by initializing with www.google.com:443.
695  SupportsQuic supports_quic1(true, "foo");
696  supports_quic_map.insert(std::make_pair(quic_server_google, supports_quic1));
697  impl_.InitializeSupportsQuic(&supports_quic_map);
698
699  SupportsQuic supports_quic2 = impl_.GetSupportsQuic(quic_server_google);
700  EXPECT_TRUE(supports_quic2.used_quic);
701  EXPECT_EQ("foo", supports_quic2.address);
702}
703
704TEST_F(SupportsQuicServerPropertiesTest, SetSupportsQuic) {
705  HostPortPair test_host_port_pair("foo", 80);
706  SupportsQuic supports_quic = impl_.GetSupportsQuic(test_host_port_pair);
707  EXPECT_FALSE(supports_quic.used_quic);
708  EXPECT_EQ("", supports_quic.address);
709  impl_.SetSupportsQuic(test_host_port_pair, true, "foo");
710  SupportsQuic supports_quic1 = impl_.GetSupportsQuic(test_host_port_pair);
711  EXPECT_TRUE(supports_quic1.used_quic);
712  EXPECT_EQ("foo", supports_quic1.address);
713
714  impl_.Clear();
715  SupportsQuic supports_quic2 = impl_.GetSupportsQuic(test_host_port_pair);
716  EXPECT_FALSE(supports_quic2.used_quic);
717  EXPECT_EQ("", supports_quic2.address);
718}
719}  // namespace
720
721}  // namespace net
722