Index: headless/lib/headless_devtools_client_browsertest.cc |
diff --git a/headless/lib/headless_devtools_client_browsertest.cc b/headless/lib/headless_devtools_client_browsertest.cc |
index 0bb4bf1be6ae3d26991615743860a6899822f886..d81df8611c70ea0686d813441b6f549e169239c9 100644 |
--- a/headless/lib/headless_devtools_client_browsertest.cc |
+++ b/headless/lib/headless_devtools_client_browsertest.cc |
@@ -433,11 +433,19 @@ class TargetDomainDisposeContextFailsIfInUse |
HEADLESS_ASYNC_DEVTOOLED_TEST_F(TargetDomainDisposeContextFailsIfInUse); |
class TargetDomainCreateTwoContexts : public HeadlessAsyncDevTooledBrowserTest, |
- public target::ExperimentalObserver { |
+ public target::ExperimentalObserver, |
+ public page::Observer { |
public: |
void RunDevTooledTest() override { |
EXPECT_TRUE(embedded_test_server()->Start()); |
+ base::RunLoop run_loop; |
+ devtools_client_->GetPage()->AddObserver(this); |
+ devtools_client_->GetPage()->Enable(run_loop.QuitClosure()); |
+ base::MessageLoop::ScopedNestableTaskAllower nest_loop( |
+ base::MessageLoop::current()); |
+ run_loop.Run(); |
+ |
devtools_client_->GetTarget()->GetExperimental()->AddObserver(this); |
devtools_client_->GetTarget()->GetExperimental()->CreateBrowserContext( |
target::CreateBrowserContextParams::Builder().Build(), |