Add HeadlessTabSocket

headless/test/headless_browser_test.cc

 // Copyright 2016 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 "headless/test/headless_browser_test.h"
 
 #include "base/files/file_path.h"
 #include "base/macros.h"
 #include "base/memory/ptr_util.h"
 #include "base/message_loop/message_loop.h"
@@ skipping 206 matching lines @@
     int exit_code) {
   if (status == base::TERMINATION_STATUS_NORMAL_TERMINATION)
     return;
 
   FAIL() << "Abnormal renderer termination";
   FinishAsynchronousTest();
   render_process_exited_ = true;
 }
 
 void HeadlessAsyncDevTooledBrowserTest::RunTest() {
-  browser_context_ = browser()
-                         ->CreateBrowserContextBuilder()
-                         .SetProtocolHandlers(GetProtocolHandlers())
-                         .Build();
+  HeadlessBrowserContext::Builder builder =
+      browser()->CreateBrowserContextBuilder();
+  builder.SetProtocolHandlers(GetProtocolHandlers());
+  if (GetCreateTabSocket()) {
+    builder.EnableUnsafeNetworkAccessWithMojoBindings(true);
+    builder.AddTabSocketMojoBindings();
+  }
+  browser_context_ = builder.Build();
+
   browser()->SetDefaultBrowserContext(browser_context_);
 
-  web_contents_ = browser_context_->CreateWebContentsBuilder().Build();
+  web_contents_ = browser_context_->CreateWebContentsBuilder()
+                      .CreateTabSocket(GetCreateTabSocket())
+                      .Build();
   web_contents_->AddObserver(this);
 
   RunAsynchronousTest();
 
   if (!render_process_exited_)
     web_contents_->GetDevToolsTarget()->DetachClient(devtools_client_.get());
   web_contents_->RemoveObserver(this);
   web_contents_->Close();
   web_contents_ = nullptr;
   browser_context_->Close();
   browser_context_ = nullptr;
 }
 
 ProtocolHandlerMap HeadlessAsyncDevTooledBrowserTest::GetProtocolHandlers() {
   return ProtocolHandlerMap();
 }
 
+bool HeadlessAsyncDevTooledBrowserTest::GetCreateTabSocket() {
+  return false;
+}
+
 }  // namespace headless