Set trusted SPDY proxy dynamically on per-profile basis

net/spdy/spdy_test_util_common.cc

 // Copyright (c) 2013 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.
 
 #include "net/spdy/spdy_test_util_common.h"
 
 #include <stdint.h>
 #include <cstddef>
 #include <utility>
 
 #include "base/compiler_specific.h"
 #include "base/logging.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/strings/string_split.h"
 #include "net/cert/mock_cert_verifier.h"
 #include "net/http/http_cache.h"
 #include "net/http/http_network_session.h"
 #include "net/http/http_network_transaction.h"
+#include "net/http/http_request_headers.h"
+#include "net/http/http_response_headers.h"
 #include "net/http/http_server_properties_impl.h"
 #include "net/socket/socket_test_util.h"
 #include "net/socket/ssl_client_socket.h"
 #include "net/socket/transport_client_socket_pool.h"
 #include "net/spdy/buffered_spdy_framer.h"
 #include "net/spdy/spdy_framer.h"
 #include "net/spdy/spdy_http_utils.h"
 #include "net/spdy/spdy_session.h"
 #include "net/spdy/spdy_session_pool.h"
 #include "net/spdy/spdy_stream.h"
@@ skipping 404 matching lines @@
   params.enable_user_alternate_protocol_ports =
       session_deps->enable_user_alternate_protocol_ports;
   params.enable_npn = session_deps->enable_npn;
   params.spdy_default_protocol = session_deps->protocol;
   params.spdy_session_max_recv_window_size =
       session_deps->session_max_recv_window_size;
   params.spdy_stream_max_recv_window_size =
       session_deps->stream_max_recv_window_size;
   params.time_func = session_deps->time_func;
   params.next_protos = session_deps->next_protos;
-  params.trusted_spdy_proxy = session_deps->trusted_spdy_proxy;
+  params.proxy_delegate = session_deps->proxy_delegate.get();
   params.parse_alternative_services = session_deps->parse_alternative_services;
   params.enable_alternative_service_with_different_host =
       session_deps->enable_alternative_service_with_different_host;
   params.net_log = session_deps->net_log;
   return params;
 }
 
 SpdyURLRequestContext::SpdyURLRequestContext(NextProto protocol)
     : storage_(this) {
   DCHECK(next_proto_is_spdy(protocol)) << "Invalid protocol: " << protocol;
@@ skipping 889 matching lines @@
     (*block)[GetVersionKey()] = "HTTP/1.1";
   }
 }
 
 void SpdyTestUtil::SetPriority(RequestPriority priority,
                                SpdySynStreamIR* ir) const {
   ir->set_priority(ConvertRequestPriorityToSpdyPriority(
       priority, spdy_version()));
 }
 
+TestProxyDelegate::TestProxyDelegate()
+    : on_before_tunnel_request_called_(false),
+      on_tunnel_request_completed_called_(false),
+      on_tunnel_headers_received_called_(false) {}
+
+TestProxyDelegate::~TestProxyDelegate() {}
+
+void TestProxyDelegate::VerifyOnTunnelRequestCompleted(
+    const std::string& endpoint,
+    const std::string& proxy_server) const {
+  EXPECT_TRUE(on_tunnel_request_completed_called_);
+  EXPECT_TRUE(HostPortPair::FromString(endpoint).Equals(
+      on_tunnel_request_completed_endpoint_));
+  EXPECT_TRUE(HostPortPair::FromString(proxy_server)
+                  .Equals(on_tunnel_request_completed_proxy_server_));
+}
+
+void TestProxyDelegate::VerifyOnTunnelHeadersReceived(
+    const std::string& origin,
+    const std::string& proxy_server,
+    const std::string& status_line) const {
+  EXPECT_TRUE(on_tunnel_headers_received_called_);
+  EXPECT_TRUE(HostPortPair::FromString(origin).Equals(
+      on_tunnel_headers_received_origin_));
+  EXPECT_TRUE(HostPortPair::FromString(proxy_server)
+                  .Equals(on_tunnel_headers_received_proxy_server_));
+  EXPECT_EQ(status_line, on_tunnel_headers_received_status_line_);
+}
+
+void TestProxyDelegate::OnResolveProxy(const GURL& url,
+                                       int load_flags,
+                                       const ProxyService& proxy_service,
+                                       ProxyInfo* result) {}
+
+void TestProxyDelegate::OnTunnelConnectCompleted(
+    const HostPortPair& endpoint,
+    const HostPortPair& proxy_server,
+    int net_error) {
+  on_tunnel_request_completed_called_ = true;
+  on_tunnel_request_completed_endpoint_ = endpoint;
+  on_tunnel_request_completed_proxy_server_ = proxy_server;
+}
+
+void TestProxyDelegate::OnFallback(const ProxyServer& bad_proxy,
+                                   int net_error) {}
+
+void TestProxyDelegate::OnBeforeSendHeaders(URLRequest* request,
+                                            const ProxyInfo& proxy_info,
+                                            HttpRequestHeaders* headers) {}
+
+void TestProxyDelegate::OnBeforeTunnelRequest(
+    const HostPortPair& proxy_server,
+    HttpRequestHeaders* extra_headers) {
+  on_before_tunnel_request_called_ = true;
+  if (extra_headers)
+    extra_headers->SetHeader("Foo", proxy_server.ToString());
+}
+
+void TestProxyDelegate::OnTunnelHeadersReceived(
+    const HostPortPair& origin,
+    const HostPortPair& proxy_server,
+    const HttpResponseHeaders& response_headers) {
+  on_tunnel_headers_received_called_ = true;
+  on_tunnel_headers_received_origin_ = origin;
+  on_tunnel_headers_received_proxy_server_ = proxy_server;
+  on_tunnel_headers_received_status_line_ = response_headers.GetStatusLine();
+}
+
+bool TestProxyDelegate::IsTrustedSpdyProxy(
+    const net::ProxyServer& proxy_server) {
+  return proxy_server.is_valid() && trusted_spdy_proxy_ == proxy_server;
+}
+
 }  // namespace net