Set trusted SPDY proxy dynamically on per-profile basis

net/spdy/spdy_test_util_common.h

 // 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.
 
 #ifndef NET_SPDY_SPDY_TEST_UTIL_COMMON_H_
 #define NET_SPDY_SPDY_TEST_UTIL_COMMON_H_
 
 #include <stddef.h>
 #include <stdint.h>
 
 #include <string>
 #include <vector>
 
 #include "base/macros.h"
 #include "base/memory/ref_counted.h"
 #include "base/memory/scoped_ptr.h"
 #include "crypto/ec_private_key.h"
 #include "crypto/ec_signature_creator.h"
 #include "net/base/completion_callback.h"
+#include "net/base/proxy_delegate.h"
 #include "net/base/request_priority.h"
 #include "net/base/test_completion_callback.h"
 #include "net/cert/cert_verifier.h"
 #include "net/dns/mock_host_resolver.h"
 #include "net/http/http_auth_handler_factory.h"
 #include "net/http/http_network_session.h"
 #include "net/http/http_response_info.h"
 #include "net/http/http_server_properties_impl.h"
 #include "net/http/transport_security_state.h"
 #include "net/proxy/proxy_service.h"
@@ skipping 167 matching lines @@
   bool enable_ip_pooling;
   bool enable_compression;
   bool enable_ping;
   bool enable_user_alternate_protocol_ports;
   bool enable_npn;
   NextProto protocol;
   size_t session_max_recv_window_size;
   size_t stream_max_recv_window_size;
   SpdySession::TimeFunc time_func;
   NextProtoVector next_protos;
-  std::string trusted_spdy_proxy;
+  scoped_ptr<ProxyDelegate> proxy_delegate;
   bool use_alternative_services;
   NetLog* net_log;
 };
 
 class SpdyURLRequestContext : public URLRequestContext {
  public:
   explicit SpdyURLRequestContext(NextProto protocol);
   ~SpdyURLRequestContext() override;
 
   MockClientSocketFactory& socket_factory() { return socket_factory_; }
@@ skipping 358 matching lines @@
 
   const NextProto protocol_;
   const SpdyMajorVersion spdy_version_;
   GURL default_url_;
   bool dependency_priorities_;
 
   // Track a FIFO list of the stream_id of all created requests by priority.
   std::map<int, std::vector<int>> priority_to_stream_id_list_;
 };
 
+class TestProxyDelegate : public ProxyDelegate {
+ public:
+  TestProxyDelegate();
+  ~TestProxyDelegate() override;
+
+  bool on_before_tunnel_request_called() const {
+    return on_before_tunnel_request_called_;
+  }
+
+  bool on_tunnel_request_completed_called() const {
+    return on_tunnel_request_completed_called_;
+  }
+
+  bool on_tunnel_headers_received_called() const {
+    return on_tunnel_headers_received_called_;
+  }
+
+  void set_trusted_spdy_proxy(const net::ProxyServer& proxy_server) {
+    trusted_spdy_proxy_ = proxy_server;
+  }
+
+  void VerifyOnTunnelRequestCompleted(const std::string& endpoint,
+                                      const std::string& proxy_server) const;
+
+  void VerifyOnTunnelHeadersReceived(const std::string& origin,
+                                     const std::string& proxy_server,
+                                     const std::string& status_line) const;
+
+  // ProxyDelegate implementation:
+  void OnResolveProxy(const GURL& url,
+                      int load_flags,
+                      const ProxyService& proxy_service,
+                      ProxyInfo* result) override;

[bengr, 2016/01/26 23:23:27]

Forward declare ProxyInfo.

[tbansal1, 2016/01/27 00:13:34]

Done but is it necessary since this class is overriding a base class. The base
class must have included/forward declared ProxyInfo. Similarly, if the base
class stops using ProxyInfo, then this class would also need to stop using it.

+  void OnTunnelConnectCompleted(const HostPortPair& endpoint,
+                                const HostPortPair& proxy_server,
+                                int net_error) override;
+  void OnFallback(const ProxyServer& bad_proxy, int net_error) override;
+  void OnBeforeSendHeaders(URLRequest* request,

[bengr, 2016/01/26 23:23:27]

Forward declare URLRequest.

[tbansal1, 2016/01/27 00:13:34]

Done.

+                           const ProxyInfo& proxy_info,
+                           HttpRequestHeaders* headers) override;

[bengr, 2016/01/26 23:23:26]

Forward declare HttpRequestHeaders.

[tbansal1, 2016/01/27 00:13:34]

Done.

+  void OnBeforeTunnelRequest(const HostPortPair& proxy_server,
+                             HttpRequestHeaders* extra_headers) override;
+  void OnTunnelHeadersReceived(
+      const HostPortPair& origin,
+      const HostPortPair& proxy_server,
+      const HttpResponseHeaders& response_headers) override;

[bengr, 2016/01/26 23:23:26]

Forward declare HttpResponseHeaders.

[tbansal1, 2016/01/27 00:13:34]

Done.

+  bool IsTrustedSpdyProxy(const net::ProxyServer& proxy_server) const override;
+
+ private:
+  bool on_before_tunnel_request_called_;
+  bool on_tunnel_request_completed_called_;
+  bool on_tunnel_headers_received_called_;
+  net::ProxyServer trusted_spdy_proxy_;

[bengr, 2016/01/26 23:23:27]

#include "net/proxy/proxy_server.h"

[tbansal1, 2016/01/27 00:13:34]

Done.

+  HostPortPair on_tunnel_request_completed_endpoint_;

[bengr, 2016/01/26 23:23:26]

#include "net/base/host_port_pair.h"

[tbansal1, 2016/01/27 00:13:34]

Done.

+  HostPortPair on_tunnel_request_completed_proxy_server_;
+  HostPortPair on_tunnel_headers_received_origin_;
+  HostPortPair on_tunnel_headers_received_proxy_server_;
+  std::string on_tunnel_headers_received_status_line_;
+};
+
 }  // namespace net
 
 #endif  // NET_SPDY_SPDY_TEST_UTIL_COMMON_H_