headless: Allow protocol handler customization

headless/public/headless_browser.cc

 // Copyright 2015 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 "content/public/common/user_agent.h"
 #include "headless/public/headless_browser.h"
 #include "net/url_request/url_request_context_getter.h"
 
 using Options = headless::HeadlessBrowser::Options;
 using Builder = headless::HeadlessBrowser::Options::Builder;
@@ skipping 47 matching lines @@
 Builder& Builder::SetHostResolverRules(const std::string& host_resolver_rules) {
   options_.host_resolver_rules = host_resolver_rules;
   return *this;
 }
 
 Builder& Builder::SetSingleProcessMode(bool single_process_mode) {
   options_.single_process_mode = single_process_mode;
   return *this;
 }
 
+Builder& Builder::SetProtocolHandlers(ProtocolHandlerMap protocol_handlers) {
+  options_.protocol_handlers = std::move(protocol_handlers);
+  return *this;
+}
+
 Options Builder::Build() {
   return std::move(options_);
 }
 
 }  // namespace headless