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 <memory> | 5 #include <memory> |
| 6 #include <string> |
| 7 #include <vector> |
6 | 8 |
7 #include "content/public/test/browser_test.h" | 9 #include "content/public/test/browser_test.h" |
| 10 #include "headless/public/domains/page.h" |
8 #include "headless/public/headless_browser.h" | 11 #include "headless/public/headless_browser.h" |
| 12 #include "headless/public/headless_devtools_client.h" |
9 #include "headless/public/headless_web_contents.h" | 13 #include "headless/public/headless_web_contents.h" |
10 #include "headless/test/headless_browser_test.h" | 14 #include "headless/test/headless_browser_test.h" |
11 #include "testing/gtest/include/gtest/gtest.h" | 15 #include "testing/gtest/include/gtest/gtest.h" |
12 #include "ui/gfx/geometry/size.h" | 16 #include "ui/gfx/geometry/size.h" |
13 #include "url/gurl.h" | 17 #include "url/gurl.h" |
14 | 18 |
15 namespace headless { | 19 namespace headless { |
16 | 20 |
17 class HeadlessWebContentsTest : public HeadlessBrowserTest {}; | 21 class HeadlessWebContentsTest : public HeadlessBrowserTest {}; |
18 | 22 |
19 IN_PROC_BROWSER_TEST_F(HeadlessWebContentsTest, Navigation) { | 23 IN_PROC_BROWSER_TEST_F(HeadlessWebContentsTest, Navigation) { |
20 EXPECT_TRUE(embedded_test_server()->Start()); | 24 EXPECT_TRUE(embedded_test_server()->Start()); |
| 25 |
21 HeadlessWebContents* web_contents = browser()->CreateWebContents( | 26 HeadlessWebContents* web_contents = browser()->CreateWebContents( |
22 embedded_test_server()->GetURL("/hello.html"), gfx::Size(800, 600)); | 27 embedded_test_server()->GetURL("/hello.html"), gfx::Size(800, 600)); |
23 EXPECT_TRUE(WaitForLoad(web_contents)); | 28 EXPECT_TRUE(WaitForLoad(web_contents)); |
24 | 29 |
25 std::vector<HeadlessWebContents*> all_web_contents = | 30 std::vector<HeadlessWebContents*> all_web_contents = |
26 browser()->GetAllWebContents(); | 31 browser()->GetAllWebContents(); |
27 | 32 |
28 EXPECT_EQ(static_cast<size_t>(1), all_web_contents.size()); | 33 EXPECT_EQ(static_cast<size_t>(1), all_web_contents.size()); |
29 EXPECT_EQ(web_contents, all_web_contents[0]); | 34 EXPECT_EQ(web_contents, all_web_contents[0]); |
30 } | 35 } |
31 | 36 |
32 IN_PROC_BROWSER_TEST_F(HeadlessWebContentsTest, WindowOpen) { | 37 IN_PROC_BROWSER_TEST_F(HeadlessWebContentsTest, WindowOpen) { |
33 EXPECT_TRUE(embedded_test_server()->Start()); | 38 EXPECT_TRUE(embedded_test_server()->Start()); |
34 | 39 |
35 HeadlessWebContents* web_contents = browser()->CreateWebContents( | 40 HeadlessWebContents* web_contents = browser()->CreateWebContents( |
36 embedded_test_server()->GetURL("/window_open.html"), gfx::Size(800, 600)); | 41 embedded_test_server()->GetURL("/window_open.html"), gfx::Size(800, 600)); |
37 EXPECT_TRUE(WaitForLoad(web_contents)); | 42 EXPECT_TRUE(WaitForLoad(web_contents)); |
38 | 43 |
39 std::vector<HeadlessWebContents*> all_web_contents = | 44 std::vector<HeadlessWebContents*> all_web_contents = |
40 browser()->GetAllWebContents(); | 45 browser()->GetAllWebContents(); |
41 | 46 |
42 EXPECT_EQ(static_cast<size_t>(2), all_web_contents.size()); | 47 EXPECT_EQ(static_cast<size_t>(2), all_web_contents.size()); |
43 } | 48 } |
44 | 49 |
| 50 class HeadlessWebContentsScreenshotTest |
| 51 : public HeadlessAsyncDevTooledBrowserTest { |
| 52 public: |
| 53 void RunDevTooledTest() override { |
| 54 devtools_client_->GetPage()->GetExperimental()->CaptureScreenshot( |
| 55 page::CaptureScreenshotParams::Builder().Build(), |
| 56 base::Bind(&HeadlessWebContentsScreenshotTest::OnScreenshotCaptured, |
| 57 base::Unretained(this))); |
| 58 } |
| 59 |
| 60 void OnScreenshotCaptured( |
| 61 std::unique_ptr<page::CaptureScreenshotResult> result) { |
| 62 EXPECT_LT(0U, result->GetData().length()); |
| 63 FinishAsynchronousTest(); |
| 64 } |
| 65 }; |
| 66 |
| 67 HEADLESS_ASYNC_DEVTOOLED_TEST_F(HeadlessWebContentsScreenshotTest); |
| 68 |
45 } // namespace headless | 69 } // namespace headless |
OLD | NEW |