1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
|
// Copyright (c) 2010 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
//
// HttpAlternateProtocols is an in-memory data structure used for keeping track
// of which HTTP HostPortPairs have an alternate protocol that can be used
// instead of HTTP on a different port.
#include "net/http/http_alternate_protocols.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace net {
namespace {
TEST(HttpAlternateProtocols, Basic) {
HttpAlternateProtocols alternate_protocols;
HostPortPair test_host_port_pair("foo", 80);
EXPECT_FALSE(
alternate_protocols.HasAlternateProtocolFor(test_host_port_pair));
alternate_protocols.SetAlternateProtocolFor(
test_host_port_pair, 443, HttpAlternateProtocols::NPN_SPDY_1);
ASSERT_TRUE(alternate_protocols.HasAlternateProtocolFor(test_host_port_pair));
const HttpAlternateProtocols::PortProtocolPair alternate =
alternate_protocols.GetAlternateProtocolFor(test_host_port_pair);
EXPECT_EQ(443, alternate.port);
EXPECT_EQ(HttpAlternateProtocols::NPN_SPDY_1, alternate.protocol);
}
TEST(HttpAlternateProtocols, SetBroken) {
HttpAlternateProtocols alternate_protocols;
HostPortPair test_host_port_pair("foo", 80);
alternate_protocols.MarkBrokenAlternateProtocolFor(test_host_port_pair);
ASSERT_TRUE(alternate_protocols.HasAlternateProtocolFor(test_host_port_pair));
HttpAlternateProtocols::PortProtocolPair alternate =
alternate_protocols.GetAlternateProtocolFor(test_host_port_pair);
EXPECT_EQ(HttpAlternateProtocols::BROKEN, alternate.protocol);
alternate_protocols.SetAlternateProtocolFor(
test_host_port_pair,
1234,
HttpAlternateProtocols::NPN_SPDY_1);
alternate = alternate_protocols.GetAlternateProtocolFor(test_host_port_pair);
EXPECT_EQ(HttpAlternateProtocols::BROKEN, alternate.protocol)
<< "Second attempt should be ignored.";
}
TEST(HttpAlternateProtocols, Forced) {
// Test forced alternate protocols.
HttpAlternateProtocols::PortProtocolPair default_protocol;
default_protocol.port = 1234;
default_protocol.protocol = HttpAlternateProtocols::NPN_SPDY_2;
HttpAlternateProtocols::ForceAlternateProtocol(default_protocol);
HttpAlternateProtocols alternate_protocols;
// Verify the forced protocol.
HostPortPair test_host_port_pair("foo", 80);
EXPECT_TRUE(
alternate_protocols.HasAlternateProtocolFor(test_host_port_pair));
HttpAlternateProtocols::PortProtocolPair alternate =
alternate_protocols.GetAlternateProtocolFor(test_host_port_pair);
EXPECT_EQ(default_protocol.port, alternate.port);
EXPECT_EQ(default_protocol.protocol, alternate.protocol);
// Verify the real protocol overrides the forced protocol.
alternate_protocols.SetAlternateProtocolFor(
test_host_port_pair, 443, HttpAlternateProtocols::NPN_SPDY_1);
ASSERT_TRUE(alternate_protocols.HasAlternateProtocolFor(test_host_port_pair));
alternate = alternate_protocols.GetAlternateProtocolFor(test_host_port_pair);
EXPECT_EQ(443, alternate.port);
EXPECT_EQ(HttpAlternateProtocols::NPN_SPDY_1, alternate.protocol);
// Turn off the static, forced alternate protocol so that tests don't
// have this state.
HttpAlternateProtocols::DisableForcedAlternateProtocol();
// Verify the forced protocol is off.
HostPortPair test_host_port_pair2("bar", 80);
EXPECT_FALSE(
alternate_protocols.HasAlternateProtocolFor(test_host_port_pair2));
}
} // namespace
} // namespace net
|
