Add HeadlessTabSocket

headless/test/headless_browser_test.cc

Index: headless/test/headless_browser_test.cc
diff --git a/headless/test/headless_browser_test.cc b/headless/test/headless_browser_test.cc
index a5191a3ccae26e9834cff332d7c04ec4be4806a3..7bc6c70171cc8b57583cb4e2c2036217f2444a20 100644
--- a/headless/test/headless_browser_test.cc
+++ b/headless/test/headless_browser_test.cc
@@ -224,13 +224,20 @@ void HeadlessAsyncDevTooledBrowserTest::RenderProcessExited(
 }
 
 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();
@@ -248,4 +255,8 @@ ProtocolHandlerMap HeadlessAsyncDevTooledBrowserTest::GetProtocolHandlers() {
   return ProtocolHandlerMap();
 }
 
+bool HeadlessAsyncDevTooledBrowserTest::GetCreateTabSocket() {
+  return false;
+}
+
 }  // namespace headless