| OLD | NEW |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "headless/test/headless_browser_test.h" | 5 #include "headless/test/headless_browser_test.h" |
| 6 | 6 |
| 7 #include "base/files/file_path.h" | 7 #include "base/files/file_path.h" |
| 8 #include "base/memory/ptr_util.h" | 8 #include "base/memory/ptr_util.h" |
| 9 #include "base/message_loop/message_loop.h" | 9 #include "base/message_loop/message_loop.h" |
| 10 #include "base/run_loop.h" | 10 #include "base/run_loop.h" |
| (...skipping 129 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 140 RunTestOnMainThread(); | 140 RunTestOnMainThread(); |
| 141 TearDownOnMainThread(); | 141 TearDownOnMainThread(); |
| 142 | 142 |
| 143 for (content::RenderProcessHost::iterator i( | 143 for (content::RenderProcessHost::iterator i( |
| 144 content::RenderProcessHost::AllHostsIterator()); | 144 content::RenderProcessHost::AllHostsIterator()); |
| 145 !i.IsAtEnd(); i.Advance()) { | 145 !i.IsAtEnd(); i.Advance()) { |
| 146 i.GetCurrentValue()->FastShutdownIfPossible(); | 146 i.GetCurrentValue()->FastShutdownIfPossible(); |
| 147 } | 147 } |
| 148 } | 148 } |
| 149 | 149 |
| 150 void HeadlessBrowserTest::SetBrowserOptions(HeadlessBrowser::Options options) { | |
| 151 HeadlessContentMainDelegate::GetInstance()->browser()->SetOptionsForTesting( | |
| 152 std::move(options)); | |
| 153 } | |
| 154 | |
| 155 HeadlessBrowser* HeadlessBrowserTest::browser() const { | 150 HeadlessBrowser* HeadlessBrowserTest::browser() const { |
| 156 return HeadlessContentMainDelegate::GetInstance()->browser(); | 151 return HeadlessContentMainDelegate::GetInstance()->browser(); |
| 157 } | 152 } |
| 158 | 153 |
| 159 bool HeadlessBrowserTest::WaitForLoad(HeadlessWebContents* web_contents) { | 154 bool HeadlessBrowserTest::WaitForLoad(HeadlessWebContents* web_contents) { |
| 160 SynchronousLoadObserver load_observer(this, web_contents); | 155 SynchronousLoadObserver load_observer(this, web_contents); |
| 161 RunAsynchronousTest(); | 156 RunAsynchronousTest(); |
| 162 return load_observer.navigation_succeeded(); | 157 return load_observer.navigation_succeeded(); |
| 163 } | 158 } |
| 164 | 159 |
| (...skipping 24 matching lines...) Expand all Loading... |
| 189 | 184 |
| 190 HeadlessAsyncDevTooledBrowserTest::~HeadlessAsyncDevTooledBrowserTest() {} | 185 HeadlessAsyncDevTooledBrowserTest::~HeadlessAsyncDevTooledBrowserTest() {} |
| 191 | 186 |
| 192 void HeadlessAsyncDevTooledBrowserTest::DevToolsTargetReady() { | 187 void HeadlessAsyncDevTooledBrowserTest::DevToolsTargetReady() { |
| 193 EXPECT_TRUE(web_contents_->GetDevToolsTarget()); | 188 EXPECT_TRUE(web_contents_->GetDevToolsTarget()); |
| 194 web_contents_->GetDevToolsTarget()->AttachClient(devtools_client_.get()); | 189 web_contents_->GetDevToolsTarget()->AttachClient(devtools_client_.get()); |
| 195 RunDevTooledTest(); | 190 RunDevTooledTest(); |
| 196 } | 191 } |
| 197 | 192 |
| 198 void HeadlessAsyncDevTooledBrowserTest::RunTest() { | 193 void HeadlessAsyncDevTooledBrowserTest::RunTest() { |
| 199 web_contents_ = browser()->CreateWebContentsBuilder().Build(); | 194 browser_context_ = browser()->CreateBrowserContextBuilder().Build(); |
| 195 |
| 196 web_contents_ = browser_context_->CreateWebContentsBuilder().Build(); |
| 200 web_contents_->AddObserver(this); | 197 web_contents_->AddObserver(this); |
| 201 | 198 |
| 202 RunAsynchronousTest(); | 199 RunAsynchronousTest(); |
| 203 | 200 |
| 204 web_contents_->GetDevToolsTarget()->DetachClient(devtools_client_.get()); | 201 web_contents_->GetDevToolsTarget()->DetachClient(devtools_client_.get()); |
| 205 web_contents_->RemoveObserver(this); | 202 web_contents_->RemoveObserver(this); |
| 206 web_contents_->Close(); | 203 web_contents_->Close(); |
| 207 web_contents_ = nullptr; | 204 web_contents_ = nullptr; |
| 205 browser_context_.reset(); |
| 208 } | 206 } |
| 209 | 207 |
| 210 } // namespace headless | 208 } // namespace headless |
| OLD | NEW |