Index: headless/test/headless_browser_test.cc |
diff --git a/headless/test/headless_browser_test.cc b/headless/test/headless_browser_test.cc |
index 1ad15d8077653c8302b77df83b336575ae76fb8b..f8ff677eb73c4e3066ea2c8ebef45a1217e5d47c 100644 |
--- a/headless/test/headless_browser_test.cc |
+++ b/headless/test/headless_browser_test.cc |
@@ -220,7 +220,7 @@ void HeadlessAsyncDevTooledBrowserTest::RunTest() { |
HeadlessBrowserContext::Builder builder = |
browser()->CreateBrowserContextBuilder(); |
builder.SetProtocolHandlers(GetProtocolHandlers()); |
- if (GetCreateTabSocket()) { |
+ if (GetTabSocketType() != HeadlessWebContents::Builder::TabSocketType::NONE) { |
builder.EnableUnsafeNetworkAccessWithMojoBindings(true); |
builder.AddTabSocketMojoBindings(); |
} |
@@ -229,7 +229,7 @@ void HeadlessAsyncDevTooledBrowserTest::RunTest() { |
browser()->SetDefaultBrowserContext(browser_context_); |
web_contents_ = browser_context_->CreateWebContentsBuilder() |
- .CreateTabSocket(GetCreateTabSocket()) |
+ .SetTabSocketType(GetTabSocketType()) |
.Build(); |
web_contents_->AddObserver(this); |
@@ -248,7 +248,13 @@ ProtocolHandlerMap HeadlessAsyncDevTooledBrowserTest::GetProtocolHandlers() { |
return ProtocolHandlerMap(); |
} |
-bool HeadlessAsyncDevTooledBrowserTest::GetCreateTabSocket() { |
+HeadlessWebContents::Builder::TabSocketType |
+HeadlessAsyncDevTooledBrowserTest::GetTabSocketType() { |
+ return HeadlessWebContents::Builder::TabSocketType::NONE; |
+} |
+ |
+bool HeadlessAsyncDevTooledBrowserTest:: |
+ GetCreateTabSocketOnlyForIsolatedWorld() { |
return false; |
} |