Add option to simulate alternate protocol always being present....

net/http/http_alternate_protocols_unittest.cc

 // 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"
@@ skipping 26 matching lines @@
 
   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