| // Copyright 2016 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #ifndef HEADLESS_TEST_HEADLESS_BROWSER_TEST_H_ |
| #define HEADLESS_TEST_HEADLESS_BROWSER_TEST_H_ |
| |
| #include <memory> |
| #include "content/public/test/browser_test_base.h" |
| #include "headless/public/headless_browser.h" |
| |
| namespace base { |
| class RunLoop; |
| } |
| |
| namespace headless { |
| class HeadlessWebContents; |
| |
| // Base class for tests which require a full instance of the headless browser. |
| class HeadlessBrowserTest : public content::BrowserTestBase { |
| public: |
| // Notify that an asynchronous test is now complete and the test runner should |
| // exit. |
| void FinishAsynchronousTest(); |
| |
| protected: |
| HeadlessBrowserTest(); |
| ~HeadlessBrowserTest() override; |
| |
| // BrowserTestBase: |
| void RunTestOnMainThreadLoop() override; |
| void SetUpOnMainThread() override; |
| void TearDownOnMainThread() override; |
| |
| // Customize the options used in this test. Note that options which take |
| // effect before the message loop has been started (e.g., custom message |
| // pumps) cannot be set via this method. |
| void SetBrowserOptions(const HeadlessBrowser::Options& options); |
| |
| // Run an asynchronous test in a nested run loop. The caller should call |
| // FinishAsynchronousTest() to notify that the test should finish. |
| void RunAsynchronousTest(); |
| |
| // Synchronously waits for a tab to finish loading. |
| bool WaitForLoad(HeadlessWebContents* web_contents); |
| |
| protected: |
| // Returns the browser for the test. |
| HeadlessBrowser* browser() const; |
| |
| private: |
| std::unique_ptr<base::RunLoop> run_loop_; |
| |
| DISALLOW_COPY_AND_ASSIGN(HeadlessBrowserTest); |
| }; |
| |
| } // namespace headless |
| |
| #endif // HEADLESS_TEST_HEADLESS_BROWSER_TEST_H_ |