| 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> | 6 #include <string> |
| 7 #include <vector> | 7 #include <vector> |
| 8 | 8 |
| 9 #include "base/base64.h" | |
| 10 #include "content/public/test/browser_test.h" | 9 #include "content/public/test/browser_test.h" |
| 11 #include "headless/public/devtools/domains/page.h" | 10 #include "headless/public/devtools/domains/page.h" |
| 12 #include "headless/public/devtools/domains/runtime.h" | |
| 13 #include "headless/public/devtools/domains/security.h" | 11 #include "headless/public/devtools/domains/security.h" |
| 14 #include "headless/public/headless_browser.h" | 12 #include "headless/public/headless_browser.h" |
| 15 #include "headless/public/headless_devtools_client.h" | 13 #include "headless/public/headless_devtools_client.h" |
| 16 #include "headless/public/headless_web_contents.h" | 14 #include "headless/public/headless_web_contents.h" |
| 17 #include "headless/test/headless_browser_test.h" | 15 #include "headless/test/headless_browser_test.h" |
| 18 #include "testing/gmock/include/gmock/gmock.h" | 16 #include "testing/gmock/include/gmock/gmock.h" |
| 19 #include "testing/gtest/include/gtest/gtest.h" | 17 #include "testing/gtest/include/gtest/gtest.h" |
| 20 #include "third_party/skia/include/core/SkBitmap.h" | |
| 21 #include "third_party/skia/include/core/SkColor.h" | |
| 22 #include "ui/gfx/codec/png_codec.h" | |
| 23 #include "ui/gfx/geometry/size.h" | 18 #include "ui/gfx/geometry/size.h" |
| 24 #include "url/gurl.h" | 19 #include "url/gurl.h" |
| 25 | 20 |
| 26 using testing::UnorderedElementsAre; | 21 using testing::UnorderedElementsAre; |
| 27 | 22 |
| 28 namespace headless { | 23 namespace headless { |
| 29 | 24 |
| 30 class HeadlessWebContentsTest : public HeadlessBrowserTest {}; | 25 class HeadlessWebContentsTest : public HeadlessBrowserTest {}; |
| 31 | 26 |
| 32 IN_PROC_BROWSER_TEST_F(HeadlessWebContentsTest, Navigation) { | 27 IN_PROC_BROWSER_TEST_F(HeadlessWebContentsTest, Navigation) { |
| (...skipping 21 matching lines...) Expand all Loading... |
| 54 HeadlessWebContents* web_contents = | 49 HeadlessWebContents* web_contents = |
| 55 browser_context->CreateWebContentsBuilder() | 50 browser_context->CreateWebContentsBuilder() |
| 56 .SetInitialURL(embedded_test_server()->GetURL("/window_open.html")) | 51 .SetInitialURL(embedded_test_server()->GetURL("/window_open.html")) |
| 57 .Build(); | 52 .Build(); |
| 58 EXPECT_TRUE(WaitForLoad(web_contents)); | 53 EXPECT_TRUE(WaitForLoad(web_contents)); |
| 59 | 54 |
| 60 EXPECT_EQ(static_cast<size_t>(2), | 55 EXPECT_EQ(static_cast<size_t>(2), |
| 61 browser_context->GetAllWebContents().size()); | 56 browser_context->GetAllWebContents().size()); |
| 62 } | 57 } |
| 63 | 58 |
| 64 namespace { | |
| 65 bool DecodePNG(std::string base64_data, SkBitmap* bitmap) { | |
| 66 std::string png_data; | |
| 67 if (!base::Base64Decode(base64_data, &png_data)) | |
| 68 return false; | |
| 69 return gfx::PNGCodec::Decode( | |
| 70 reinterpret_cast<unsigned const char*>(png_data.data()), png_data.size(), | |
| 71 bitmap); | |
| 72 } | |
| 73 } // namespace | |
| 74 | |
| 75 // Parameter specifies whether --disable-gpu should be used. | |
| 76 class HeadlessWebContentsScreenshotTest | 59 class HeadlessWebContentsScreenshotTest |
| 77 : public HeadlessAsyncDevTooledBrowserTest, | 60 : public HeadlessAsyncDevTooledBrowserTest { |
| 78 public ::testing::WithParamInterface<bool> { | |
| 79 public: | 61 public: |
| 80 void SetUp() override { | |
| 81 EnablePixelOutput(); | |
| 82 if (GetParam()) | |
| 83 UseSoftwareCompositing(); | |
| 84 HeadlessAsyncDevTooledBrowserTest::SetUp(); | |
| 85 } | |
| 86 | |
| 87 void RunDevTooledTest() override { | 62 void RunDevTooledTest() override { |
| 88 std::unique_ptr<runtime::EvaluateParams> params = | |
| 89 runtime::EvaluateParams::Builder() | |
| 90 .SetExpression("document.body.style.background = '#0000ff'") | |
| 91 .Build(); | |
| 92 devtools_client_->GetRuntime()->Evaluate( | |
| 93 std::move(params), | |
| 94 base::Bind(&HeadlessWebContentsScreenshotTest::OnPageSetupCompleted, | |
| 95 base::Unretained(this))); | |
| 96 } | |
| 97 | |
| 98 void OnPageSetupCompleted(std::unique_ptr<runtime::EvaluateResult> result) { | |
| 99 devtools_client_->GetPage()->GetExperimental()->CaptureScreenshot( | 63 devtools_client_->GetPage()->GetExperimental()->CaptureScreenshot( |
| 100 page::CaptureScreenshotParams::Builder().Build(), | 64 page::CaptureScreenshotParams::Builder().Build(), |
| 101 base::Bind(&HeadlessWebContentsScreenshotTest::OnScreenshotCaptured, | 65 base::Bind(&HeadlessWebContentsScreenshotTest::OnScreenshotCaptured, |
| 102 base::Unretained(this))); | 66 base::Unretained(this))); |
| 103 } | 67 } |
| 104 | 68 |
| 105 void OnScreenshotCaptured( | 69 void OnScreenshotCaptured( |
| 106 std::unique_ptr<page::CaptureScreenshotResult> result) { | 70 std::unique_ptr<page::CaptureScreenshotResult> result) { |
| 107 std::string base64 = result->GetData(); | 71 EXPECT_LT(0U, result->GetData().length()); |
| 108 EXPECT_LT(0U, base64.length()); | |
| 109 SkBitmap result_bitmap; | |
| 110 EXPECT_TRUE(DecodePNG(base64, &result_bitmap)); | |
| 111 | |
| 112 EXPECT_EQ(800, result_bitmap.width()); | |
| 113 EXPECT_EQ(600, result_bitmap.height()); | |
| 114 SkColor actual_color = result_bitmap.getColor(400, 300); | |
| 115 SkColor expected_color = SkColorSetRGB(0x00, 0x00, 0xff); | |
| 116 EXPECT_EQ(expected_color, actual_color); | |
| 117 FinishAsynchronousTest(); | 72 FinishAsynchronousTest(); |
| 118 } | 73 } |
| 119 }; | 74 }; |
| 120 | 75 |
| 121 HEADLESS_ASYNC_DEVTOOLED_TEST_P(HeadlessWebContentsScreenshotTest); | 76 HEADLESS_ASYNC_DEVTOOLED_TEST_F(HeadlessWebContentsScreenshotTest); |
| 122 | |
| 123 // Instantiate test case for both software and gpu compositing modes. | |
| 124 INSTANTIATE_TEST_CASE_P(HeadlessWebContentsScreenshotTests, | |
| 125 HeadlessWebContentsScreenshotTest, | |
| 126 ::testing::Bool()); | |
| 127 | 77 |
| 128 class HeadlessWebContentsSecurityTest | 78 class HeadlessWebContentsSecurityTest |
| 129 : public HeadlessAsyncDevTooledBrowserTest, | 79 : public HeadlessAsyncDevTooledBrowserTest, |
| 130 public security::ExperimentalObserver { | 80 public security::ExperimentalObserver { |
| 131 public: | 81 public: |
| 132 void RunDevTooledTest() override { | 82 void RunDevTooledTest() override { |
| 133 devtools_client_->GetSecurity()->GetExperimental()->AddObserver(this); | 83 devtools_client_->GetSecurity()->GetExperimental()->AddObserver(this); |
| 134 devtools_client_->GetSecurity()->GetExperimental()->Enable( | 84 devtools_client_->GetSecurity()->GetExperimental()->Enable( |
| 135 security::EnableParams::Builder().Build()); | 85 security::EnableParams::Builder().Build()); |
| 136 } | 86 } |
| 137 | 87 |
| 138 void OnSecurityStateChanged( | 88 void OnSecurityStateChanged( |
| 139 const security::SecurityStateChangedParams& params) override { | 89 const security::SecurityStateChangedParams& params) override { |
| 140 EXPECT_EQ(security::SecurityState::NEUTRAL, params.GetSecurityState()); | 90 EXPECT_EQ(security::SecurityState::NEUTRAL, params.GetSecurityState()); |
| 141 | 91 |
| 142 FinishAsynchronousTest(); | 92 FinishAsynchronousTest(); |
| 143 } | 93 } |
| 144 }; | 94 }; |
| 145 | 95 |
| 146 HEADLESS_ASYNC_DEVTOOLED_TEST_F(HeadlessWebContentsSecurityTest); | 96 HEADLESS_ASYNC_DEVTOOLED_TEST_F(HeadlessWebContentsSecurityTest); |
| 147 | 97 |
| 148 } // namespace headless | 98 } // namespace headless |
| OLD | NEW |