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 #ifndef HEADLESS_TEST_HEADLESS_BROWSER_TEST_H_ | 5 #ifndef HEADLESS_TEST_HEADLESS_BROWSER_TEST_H_ |
6 #define HEADLESS_TEST_HEADLESS_BROWSER_TEST_H_ | 6 #define HEADLESS_TEST_HEADLESS_BROWSER_TEST_H_ |
7 | 7 |
8 #include "content/public/test/browser_test_base.h" | 8 #include "content/public/test/browser_test_base.h" |
| 9 #include "headless/public/headless_browser.h" |
9 | 10 |
10 namespace headless { | 11 namespace headless { |
11 class HeadlessBrowser; | 12 class HeadlessWebContents; |
12 | 13 |
13 // Base class for tests which require a full instance of the headless browser. | 14 // Base class for tests which require a full instance of the headless browser. |
14 class HeadlessBrowserTest : public content::BrowserTestBase { | 15 class HeadlessBrowserTest : public content::BrowserTestBase { |
15 protected: | 16 protected: |
16 HeadlessBrowserTest(); | 17 HeadlessBrowserTest(); |
17 ~HeadlessBrowserTest() override; | 18 ~HeadlessBrowserTest() override; |
18 | 19 |
19 // BrowserTestBase: | 20 // BrowserTestBase: |
20 void RunTestOnMainThreadLoop() override; | 21 void RunTestOnMainThreadLoop() override; |
21 void SetUpOnMainThread() override; | 22 void SetUpOnMainThread() override; |
22 void TearDownOnMainThread() override; | 23 void TearDownOnMainThread() override; |
23 | 24 |
| 25 // Customize the options used in this test. Note that options which take |
| 26 // effect before the message loop has been started (e.g., custom message |
| 27 // pumps) cannot be set via this method. |
| 28 void SetBrowserOptions(const HeadlessBrowser::Options& options); |
| 29 |
| 30 // Navigate to |url| and wait for the document load to complete. |
| 31 bool NavigateAndWaitForLoad(HeadlessWebContents* web_contents, |
| 32 const GURL& url); |
| 33 |
24 protected: | 34 protected: |
25 // Returns the browser for the test. | 35 // Returns the browser for the test. |
26 HeadlessBrowser* browser() const; | 36 HeadlessBrowser* browser() const; |
27 | 37 |
28 private: | 38 private: |
29 DISALLOW_COPY_AND_ASSIGN(HeadlessBrowserTest); | 39 DISALLOW_COPY_AND_ASSIGN(HeadlessBrowserTest); |
30 }; | 40 }; |
31 | 41 |
32 } // namespace headless | 42 } // namespace headless |
33 | 43 |
34 #endif // HEADLESS_TEST_HEADLESS_BROWSER_TEST_H_ | 44 #endif // HEADLESS_TEST_HEADLESS_BROWSER_TEST_H_ |
OLD | NEW |